package main

import (
	"github.com/gdamore/tcell/v2"
	"github.com/rivo/tview"
	"strconv"
)

type UICycle struct {
	widget []tview.Primitive
	index  int
	len    int
}

func NewUICycle(widgets ...tview.Primitive) *UICycle {
	var w []tview.Primitive

	for _, widget := range widgets {
		w = append(w, widget)
	}

	return &UICycle{
		widget: w,
		index:  0,
		len:    len(w),
	}
}

func (ui *UICycle) Increase() {
	ui.index = (ui.index + 1) % ui.len
}

func (ui *UICycle) GetCurrentUI() tview.Primitive {
	return ui.widget[ui.index]
}

func updateBackgroundColor() {
	// box
	srcBox.SetBackgroundColor(window.src.backgroundColor)
	srcBox.SetTextStyle(tcell.StyleDefault.
		Background(window.src.backgroundColor).
		Foreground(window.src.foregroundColor))
	dstBox.SetBackgroundColor(window.dst.backgroundColor)

	// dropdown
	srcLangDropDown.SetBackgroundColor(window.src.backgroundColor)
	srcLangDropDown.SetListStyles(tcell.StyleDefault.
		Background(window.src.backgroundColor).
		Foreground(window.src.foregroundColor),
		tcell.StyleDefault.
			Background(window.src.selectedColor).
			Foreground(window.src.prefixColor))
	dstLangDropDown.SetBackgroundColor(window.dst.backgroundColor)
	dstLangDropDown.SetListStyles(tcell.StyleDefault.
		Background(window.src.backgroundColor).
		Foreground(window.src.foregroundColor),
		tcell.StyleDefault.
			Background(window.src.selectedColor).
			Foreground(window.src.prefixColor))
	themeDropDown.SetBackgroundColor(window.src.backgroundColor)
	themeDropDown.SetListStyles(tcell.StyleDefault.
		Background(window.src.backgroundColor).
		Foreground(window.src.foregroundColor),
		tcell.StyleDefault.
			Background(window.src.selectedColor).
			Foreground(window.src.prefixColor))
	transparentDropDown.SetBackgroundColor(window.src.backgroundColor)
	transparentDropDown.SetListStyles(tcell.StyleDefault.
		Background(window.src.backgroundColor).
		Foreground(window.src.foregroundColor),
		tcell.StyleDefault.
			Background(window.src.selectedColor).
			Foreground(window.src.prefixColor))
	srcBorderDropDown.SetBackgroundColor(window.src.backgroundColor)
	srcBorderDropDown.SetListStyles(tcell.StyleDefault.
		Background(window.src.backgroundColor).
		Foreground(window.src.foregroundColor),
		tcell.StyleDefault.
			Background(window.src.selectedColor).
			Foreground(window.src.prefixColor))
	dstBorderDropDown.SetBackgroundColor(window.dst.backgroundColor)
	dstBorderDropDown.SetListStyles(tcell.StyleDefault.
		Background(window.src.backgroundColor).
		Foreground(window.src.foregroundColor),
		tcell.StyleDefault.
			Background(window.src.selectedColor).
			Foreground(window.src.prefixColor))
}

func updateBorderColor() {
	// box
	srcBox.SetBorderColor(window.src.borderColor).
		SetTitleColor(window.src.borderColor)
	dstBox.SetBorderColor(window.dst.borderColor).
		SetTitleColor(window.dst.borderColor)

	// dropdown
	srcLangDropDown.SetBorderColor(window.src.borderColor).
		SetTitleColor(window.src.borderColor)
	dstLangDropDown.SetBorderColor(window.dst.borderColor).
		SetTitleColor(window.dst.borderColor)
	srcBorderDropDown.SetBorderColor(window.src.borderColor).
		SetTitleColor(window.src.borderColor)
	dstBorderDropDown.SetBorderColor(window.dst.borderColor).
		SetTitleColor(window.dst.borderColor)
}

func updateNonConfigColor() {
	// box
	srcBox.SetSelectedStyle(tcell.StyleDefault.
		Background(window.src.selectedColor).
		Foreground(window.src.foregroundColor))
	dstBox.SetTextColor(window.dst.foregroundColor)

	// dropdown
	srcLangDropDown.SetFieldBackgroundColor(window.src.selectedColor).
		SetFieldTextColor(window.src.foregroundColor).
		SetPrefixTextColor(window.src.prefixColor)
	dstLangDropDown.SetFieldBackgroundColor(window.dst.selectedColor).
		SetFieldTextColor(window.dst.foregroundColor).
		SetPrefixTextColor(window.dst.prefixColor)
	themeDropDown.SetLabelColor(window.src.labelColor)
	themeDropDown.SetFieldBackgroundColor(window.src.selectedColor).
		SetFieldTextColor(window.src.foregroundColor).
		SetPrefixTextColor(window.src.prefixColor)
	transparentDropDown.SetLabelColor(window.src.labelColor)
	transparentDropDown.SetFieldBackgroundColor(window.src.selectedColor).
		SetFieldTextColor(window.src.foregroundColor).
		SetPrefixTextColor(window.src.prefixColor)
	srcBorderDropDown.SetLabelColor(window.src.labelColor)
	srcBorderDropDown.SetFieldBackgroundColor(window.src.selectedColor).
		SetFieldTextColor(window.src.foregroundColor).
		SetPrefixTextColor(window.src.prefixColor)
	dstBorderDropDown.SetLabelColor(window.src.labelColor)
	dstBorderDropDown.SetFieldBackgroundColor(window.src.selectedColor).
		SetFieldTextColor(window.src.foregroundColor).
		SetPrefixTextColor(window.src.prefixColor)

	// button
	langButton.SetLabelColor(window.src.foregroundColor).
		SetBackgroundColorActivated(window.src.pressColor).
		SetLabelColorActivated(window.src.foregroundColor).
		SetBackgroundColor(window.src.selectedColor)
	styleButton.SetLabelColor(window.src.foregroundColor).
		SetBackgroundColorActivated(window.src.pressColor).
		SetLabelColorActivated(window.src.foregroundColor).
		SetBackgroundColor(window.src.selectedColor)
	menuButton.SetLabelColor(window.src.foregroundColor).
		SetBackgroundColorActivated(window.src.pressColor).
		SetLabelColorActivated(window.src.foregroundColor).
		SetBackgroundColor(window.src.selectedColor)
}

func updateAllColor() {
	updateBackgroundColor()
	updateBorderColor()
	updateNonConfigColor()
}

// Update title and option
func updateTitle() {
	srcBox.SetTitle(translator.srcLang)
	dstBox.SetTitle(translator.dstLang)
	srcLangDropDown.SetCurrentOption(IndexOf(translator.srcLang, Lang))
	srcLangDropDown.SetTitle(translator.srcLang)
	dstLangDropDown.SetCurrentOption(IndexOf(translator.dstLang, Lang))
	dstLangDropDown.SetTitle(translator.dstLang)
}

func attachButton() *tview.Flex {
	return tview.NewFlex().SetDirection(tview.FlexColumn).
		AddItem(nil, 0, 1, false).
		AddItem(langButton, 11, 1, true).
		AddItem(nil, 18, 1, false).
		AddItem(styleButton, 8, 1, true).
		AddItem(nil, 18, 1, false).
		AddItem(menuButton, 9, 1, true).
		AddItem(nil, 0, 1, false)
}

func uiInit() {
	// box
	srcBox.SetBorder(true)
	dstBox.SetBorder(true)

	// dropdown
	srcLangDropDown.SetBorder(true)
	srcLangDropDown.SetOptions(Lang, nil)
	dstLangDropDown.SetBorder(true)
	dstLangDropDown.SetOptions(Lang, nil)
	themeDropDown.SetLabel("Theme: ").
		SetOptions(ThemesName, nil).
		SetCurrentOption(IndexOf(theme, ThemesName))
	transparentDropDown.SetLabel("Transparent: ").
		SetOptions([]string{"true", "false"}, nil).
		SetCurrentOption(
			IndexOf(strconv.FormatBool(transparent),
				[]string{"true", "false"}))
	srcBorderDropDown.SetOptions(Palette, nil)
	srcBorderDropDown.SetBorder(true).
		SetTitle("Source")
	dstBorderDropDown.SetOptions(Palette, nil)
	dstBorderDropDown.SetBorder(true).
		SetTitle("Destination")

	// window
	translateWindow.SetDirection(tview.FlexColumn).
		AddItem(srcBox, 0, 1, true).
		AddItem(dstBox, 0, 1, false)
	langWindow.SetDirection(tview.FlexRow).
		AddItem(nil, 0, 1, false).
		AddItem(tview.NewFlex().SetDirection(tview.FlexColumn).
			AddItem(nil, 0, 1, false).
			AddItem(srcLangDropDown, 32, 1, true).
			AddItem(dstLangDropDown, 32, 1, false).
			AddItem(nil, 0, 1, false), 20, 1, true).
		AddItem(attachButton(), 1, 1, false).
		AddItem(nil, 0, 1, false)
	styleWindow.SetDirection(tview.FlexRow).
		AddItem(nil, 0, 1, false).
		AddItem(tview.NewFlex().SetDirection(tview.FlexColumn).
			AddItem(nil, 0, 1, false).
			AddItem(tview.NewFlex().SetDirection(tview.FlexRow).
				AddItem(themeDropDown, 1, 1, true).
				AddItem(transparentDropDown, 1, 1, false).
				AddItem(tview.NewFlex().SetDirection(tview.FlexColumn).
					AddItem(nil, 0, 1, false).
					AddItem(srcBorderDropDown, 32, 1, false).
					AddItem(dstBorderDropDown, 32, 1, false).
					AddItem(nil, 0, 1, false), 0, 1, false),
				20, 1, true).
			AddItem(nil, 0, 1, false), 20, 1, true).
		AddItem(attachButton(), 1, 1, false).
		AddItem(nil, 0, 1, false)

	updateAllColor()
	updateTitle()

	// handler
	mainPage.SetInputCapture(pagesHandler)
	langWindow.SetInputCapture(langWindowHandler)
	styleWindow.SetInputCapture(styleWindowHandler)
	translateWindow.SetInputCapture(translatePageHandler)
	srcLangDropDown.SetDoneFunc(langDropDownHandler).
		SetSelectedFunc(srcLangSelected)
	dstLangDropDown.SetDoneFunc(langDropDownHandler).
		SetSelectedFunc(dstLangSelected)
	themeDropDown.SetDoneFunc(styleDropDownHandler).
		SetSelectedFunc(themeSelected)
	transparentDropDown.SetDoneFunc(styleDropDownHandler).
		SetSelectedFunc(transparentSelected)
	srcBorderDropDown.SetDoneFunc(styleDropDownHandler)
	dstBorderDropDown.SetDoneFunc(styleDropDownHandler)
	langButton.SetSelectedFunc(func() {
		mainPage.HidePage("stylePage")
		mainPage.ShowPage("langPage")
	})
	styleButton.SetSelectedFunc(func() {
		mainPage.HidePage("langPage")
		mainPage.ShowPage("stylePage")
	})
}

func pagesHandler(event *tcell.EventKey) *tcell.EventKey {
	key := event.Key()

	switch key {
	case tcell.KeyCtrlT:
		if transparent {
			window.src.backgroundColor = Themes[theme]["bg"]
			window.dst.backgroundColor = Themes[theme]["bg"]
		} else {
			window.src.backgroundColor = Transparent
			window.dst.backgroundColor = Transparent
		}
		updateBackgroundColor()
		transparent = !transparent
		transparentDropDown.SetCurrentOption(
			IndexOf(strconv.FormatBool(transparent),
				[]string{"true", "false"}))
	}

	return event
}

func langWindowHandler(event *tcell.EventKey) *tcell.EventKey {
	ch := event.Rune()

	switch ch {
	case '2':
		mainPage.HidePage("langPage")
		mainPage.ShowPage("stylePage")
	}

	return event
}

func styleWindowHandler(event *tcell.EventKey) *tcell.EventKey {
	ch := event.Rune()

	switch ch {
	case '1':
		mainPage.HidePage("stylePage")
		mainPage.ShowPage("langPage")
	}

	return event
}

func translatePageHandler(event *tcell.EventKey) *tcell.EventKey {
	key := event.Key()

	switch key {
	case tcell.KeyEsc:
		mainPage.ShowPage("langPage")
	case tcell.KeyCtrlJ:
		message := srcBox.GetText()
		if len(message) > 0 {
			result, err := translator.Translate(message)
			if err != nil {
				dstBox.SetText(err.Error())
			} else {
				dstBox.SetText(result)
			}
		}
	case tcell.KeyCtrlQ:
		srcBox.SetText("", true)
	case tcell.KeyCtrlS:
		translator.srcLang, translator.dstLang = translator.dstLang, translator.srcLang
		updateTitle()
		src_text := srcBox.GetText()
		dst_text := dstBox.GetText(false)
		if len(dst_text) > 0 {
			// GetText of Box contains "\n" if it has words
			srcBox.SetText(dst_text[:len(dst_text)-1], true)
		} else {
			srcBox.SetText(dst_text, true)
		}
		dstBox.SetText(src_text)
	case tcell.KeyCtrlO:
		// Play source sound
		if translator.soundLock.Available() {
			message := srcBox.GetText()
			if len(message) > 0 {
				translator.soundLock.Acquire()
				go func() {
					err := translator.PlaySound(translator.srcLang, message)
					if err != nil {
						srcBox.SetText(err.Error(), true)
					}
				}()
			}

		}
	case tcell.KeyCtrlP:
		// Play destination sound
		if translator.soundLock.Available() {
			message := dstBox.GetText(false)
			if len(message) > 0 {
				translator.soundLock.Acquire()
				go func() {
					err := translator.PlaySound(translator.dstLang, message)
					if err != nil {
						dstBox.SetText(err.Error())
					}
				}()
			}
		}
	case tcell.KeyCtrlX:
		// Stop play sound
		translator.soundLock.stop = true
	}

	return event
}

func langDropDownHandler(key tcell.Key) {
	switch key {
	case tcell.KeyTAB:
		langCycle.Increase()
		app.SetFocus(langCycle.GetCurrentUI())
	case tcell.KeyEsc:
		mainPage.HidePage("langPage")
	}
}

func srcLangSelected(text string, index int) {
	translator.srcLang = text
	srcBox.SetTitle(text)
	srcLangDropDown.SetTitle(text)
}

func dstLangSelected(text string, index int) {
	translator.dstLang = text
	dstBox.SetTitle(text)
	dstLangDropDown.SetTitle(text)
}

func styleDropDownHandler(key tcell.Key) {
	switch key {
	case tcell.KeyTAB:
		styleCycle.Increase()
		app.SetFocus(styleCycle.GetCurrentUI())
	case tcell.KeyEsc:
		mainPage.HidePage("stylePage")
	}
}

func themeSelected(text string, index int) {
	theme = text
	window.colorInit()
	updateAllColor()
}

func transparentSelected(text string, index int) {
	transparent, _ = strconv.ParseBool(text)
	if transparent {
		window.src.backgroundColor = Transparent
		window.dst.backgroundColor = Transparent
	} else {
		window.src.backgroundColor = Themes[theme]["bg"]
		window.dst.backgroundColor = Themes[theme]["bg"]
	}
	updateBackgroundColor()
}