package confirmation

import (
	"fmt"
	"git.shadowhosting.xyz/Eggactyl/tui"
	"strings"

	"golang.org/x/term"
)

type InputData struct {
	Notice   string
	Question string
}

func New(data InputData) (*bool, error) {

	if data.Notice != "" {
		fmt.Printf(
			tui.Format(tui.FmtBold, tui.FgColorGrey) +
				"[" + tui.FgColorGold + "!" + tui.FgColorGrey + "]" + tui.FmtBoldReset + " " +
				tui.FmtItalic + data.Notice + tui.FmtReset + "\n",
		)
		//fmt.Printf("\033[1m\033[38;5;247m[\033[38;5;214m!\033[38;5;247m]\033[22m \033[3m%s\033[0m\n", data.Notice)
	}

	if data.Question != "" {
		fmt.Printf(
			tui.Format(tui.FmtBold, tui.FgColorGrey) +
				"[" + tui.FgColorGold + "!" + tui.FgColorGrey + "]" + tui.FmtReset + " " +
				data.Question +
				"(" + tui.FgColorGreen + "y" + tui.FmtReset + "/" + tui.FgColorRed + "n" + tui.FmtReset + "\n " +
				tui.FgColorGrey + ">>" + tui.Format(tui.FmtItalic, tui.FgColorGold) + "\n",
		)
		//fmt.Printf("\033[1m\033[38;5;247m[\033[38;5;214m?\033[38;5;247m]\033[0m %s (\033[38;5;34my\033[0m/\033[38;5;167mn\033[0m) \033[38;5;247m>>\033[3m\033[38;5;214m\n", data.Question)
	}

	var input string
	var chosen bool

inputLoop:
	for {
		if _, err := fmt.Scanln(&input); err != nil {
			return nil, err
		}

		width, _, err := term.GetSize(0)
		if err != nil {
			return nil, err
		}

		switch strings.ToLower(input) {
		case "y", "yes":
			chosen = true
			break inputLoop
		case "n", "no":
			chosen = false
			break inputLoop
		default:
			var lineNum int

			if width == 0 {
				if data.Notice != "" {
					lineNum++
				}
				lineNum += 2
			} else {
				if data.Notice == "" {
					lineNum = ((len(data.Question) + 5 + width - 1) / width) + 1
				} else {
					lineNum = ((len(data.Notice) + 5 + width) / width) + ((len(data.Question) + 5 + width - 1) / width) + 1
				}

			}

			for i := 0; i < lineNum; i++ {
				fmt.Printf("\033[A\033[K\033[0G")
			}
			fmt.Printf(
				tui.Format(tui.FmtBold, tui.FgColorGrey) +
					"[" + tui.FgColorGold + "!" + tui.FgColorGrey + "]" + tui.FmtReset + " " +
					"Invalid input, please try again!\n",
			)
			//fmt.Printf("\033[1m\033[38;5;247m[\033[38;5;167m!\033[38;5;247m]\033[0m Invalid input, please try again!\n")

			if data.Notice != "" {
				fmt.Printf(
					tui.Format(tui.FmtBold, tui.FgColorGrey) +
						"[" + tui.FgColorGold + "!" + tui.FgColorGrey + "]" + tui.FmtBoldReset + " " +
						tui.FmtItalic + data.Notice + tui.FmtReset + "\n",
				)
				//fmt.Printf("\033[1m\033[38;5;247m[\033[38;5;214m!\033[38;5;247m]\033[22m \033[3m%s\033[0m\n", data.Notice)
			}

			if data.Question != "" {
				fmt.Printf(
					tui.Format(tui.FmtBold, tui.FgColorGrey) +
						"[" + tui.FgColorGold + "?" + tui.FgColorGrey + "]" + tui.FmtReset + " " + data.Notice +
						"(" + tui.FgColorGreen + "y" + tui.FmtReset + "/" + tui.FgColorRed + "n" + tui.FmtReset + ")" +
						tui.FgColorGrey + ">>" + tui.Format(tui.FmtItalic, tui.FgColorGold) + "\n",
				)
				//fmt.Printf("\033[1m\033[38;5;247m[\033[38;5;214m?\033[38;5;247m]\033[0m %s (\033[38;5;34my\033[0m/\033[38;5;167mn\033[0m) \033[38;5;247m>>\033[3m\033[38;5;214m\n", data.Question)
			}
			continue inputLoop
		}
	}

	fmt.Println("\033[0m")

	return &chosen, nil

}