color in string instead of tcell.Color

2025-05-19 01:00:20 -07:00 · 2022-10-24 16:37:20 +08:00 · 2022-10-24 16:37:20 +08:00 · e3f67f4454
commit e3f67f4454
parent cc3db4bb69
4 changed files with 153 additions and 130 deletions
--- a/color.go
+++ b/color.go
@ -34,38 +34,73 @@ var (
 	}
 )

-type Colors struct {
-	backgroundColor tcell.Color
-	foregroundColor tcell.Color
-	borderColor     tcell.Color
-	textColor       tcell.Color
-	selectedColor   tcell.Color
-	prefixColor     tcell.Color
-	labelColor      tcell.Color
-	pressColor      tcell.Color
+type WindowStyle struct {
+	borderColor string
 }

-type Window struct {
-	src Colors
-	dst Colors
+type Style struct {
+	src             WindowStyle
+	dst             WindowStyle
+	backgroundColor string
+	foregroundColor string
+	selectedColor   string
+	prefixColor     string
+	labelColor      string
+	pressColor      string
 }

-func (w *Window) colorInit() {
-	if transparent {
-		w.src.backgroundColor = Transparent
-		w.dst.backgroundColor = Transparent
-	} else {
-		w.src.backgroundColor = Themes[theme]["bg"]
-		w.dst.backgroundColor = Themes[theme]["bg"]
+func NewStyle() *Style {
+	return &Style{
+		backgroundColor: "bg",
+		foregroundColor: "fg",
+		selectedColor:   "gray",
+		prefixColor:     "cyan",
+		labelColor:      "yellow",
+		pressColor:      "cyan",
 	}
-	w.src.borderColor = Themes[theme]["red"]
-	w.src.foregroundColor = Themes[theme]["fg"]
-	w.src.selectedColor = Themes[theme]["gray"]
-	w.src.prefixColor = Themes[theme]["cyan"]
-	w.src.pressColor = Themes[theme]["purple"]
-	w.src.labelColor = Themes[theme]["yellow"]
-	w.dst.foregroundColor = Themes[theme]["fg"]
-	w.dst.selectedColor = Themes[theme]["gray"]
-	w.dst.borderColor = Themes[theme]["blue"]
-	w.dst.prefixColor = Themes[theme]["cyan"]
+}
+
+func (s Style) BackgroundColor() tcell.Color {
+	if transparent {
+		return Transparent
+	}
+	return Themes[theme][s.backgroundColor]
+}
+
+func (s Style) ForegroundColor() tcell.Color {
+	return Themes[theme][s.foregroundColor]
+}
+
+func (s Style) SelectedColor() tcell.Color {
+	return Themes[theme][s.selectedColor]
+}
+
+func (s Style) PrefixColor() tcell.Color {
+	return Themes[theme][s.prefixColor]
+}
+
+func (s Style) LabelColor() tcell.Color {
+	return Themes[theme][s.labelColor]
+}
+
+func (s Style) PressColor() tcell.Color {
+	return Themes[theme][s.pressColor]
+}
+
+func (s Style) SrcBorderColor() tcell.Color {
+	return Themes[theme][s.src.borderColor]
+}
+
+func (s Style) DstBorderColor() tcell.Color {
+	return Themes[theme][s.dst.borderColor]
+}
+
+func (s *Style) SetSrcBorderColor(color string) *Style {
+	s.src.borderColor = color
+	return s
+}
+
+func (s *Style) SetDstBorderColor(color string) *Style {
+	s.dst.borderColor = color
+	return s
 }
--- a/config.go
+++ b/config.go
@ -24,6 +24,8 @@ func configInit() {
 		config.Set("theme", "Gruvbox")
 		config.Set("source.language", "English")
 		config.Set("destination.language", "Chinese (Traditional)")
+		config.Set("source.borderColor", "red")
+		config.Set("destination.borderColor", "blue")
 		if _, err = os.Stat(defaultConfigPath); os.IsNotExist(err) {
 			os.Mkdir(defaultConfigPath, os.ModePerm)
 		}
@ -35,6 +37,8 @@ func configInit() {
 	transparent = config.GetBool("transparent")
 	translator.srcLang = config.GetString("source.language")
 	translator.dstLang = config.GetString("destination.language")
+	style.SetSrcBorderColor(config.GetString("source.borderColor"))
+	style.SetDstBorderColor(config.GetString("destination.borderColor"))
 }

 // Check if need to modify config file when quit program
--- a/main.go
+++ b/main.go
@ -27,7 +27,7 @@ var (
 	langWindow          = tview.NewFlex()
 	styleWindow         = tview.NewFlex()
 	mainPage            = tview.NewPages()
-	window              Window
+	style               = NewStyle()
 	// config
 	config      = viper.New()
 	theme       string
@ -37,7 +37,6 @@ var (
 func main() {
 	SetTermTitle("GTT")
 	configInit()
-	window.colorInit()
 	uiInit()

 	mainPage.AddPage("translatePage", translateWindow, true, true)
--- a/ui.go
+++ b/ui.go
@ -36,119 +36,119 @@ func (ui *UICycle) GetCurrentUI() tview.Primitive {

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

 	// dropdown
-	srcLangDropDown.SetBackgroundColor(window.src.backgroundColor)
+	srcLangDropDown.SetBackgroundColor(style.BackgroundColor())
 	srcLangDropDown.SetListStyles(tcell.StyleDefault.
-		Background(window.src.backgroundColor).
-		Foreground(window.src.foregroundColor),
+		Background(style.BackgroundColor()).
+		Foreground(style.ForegroundColor()),
 		tcell.StyleDefault.
-			Background(window.src.selectedColor).
-			Foreground(window.src.prefixColor))
-	dstLangDropDown.SetBackgroundColor(window.dst.backgroundColor)
+			Background(style.SelectedColor()).
+			Foreground(style.PrefixColor()))
+	dstLangDropDown.SetBackgroundColor(style.BackgroundColor())
 	dstLangDropDown.SetListStyles(tcell.StyleDefault.
-		Background(window.src.backgroundColor).
-		Foreground(window.src.foregroundColor),
+		Background(style.BackgroundColor()).
+		Foreground(style.ForegroundColor()),
 		tcell.StyleDefault.
-			Background(window.src.selectedColor).
-			Foreground(window.src.prefixColor))
-	themeDropDown.SetBackgroundColor(window.src.backgroundColor)
+			Background(style.SelectedColor()).
+			Foreground(style.PrefixColor()))
+	themeDropDown.SetBackgroundColor(style.BackgroundColor())
 	themeDropDown.SetListStyles(tcell.StyleDefault.
-		Background(window.src.backgroundColor).
-		Foreground(window.src.foregroundColor),
+		Background(style.BackgroundColor()).
+		Foreground(style.ForegroundColor()),
 		tcell.StyleDefault.
-			Background(window.src.selectedColor).
-			Foreground(window.src.prefixColor))
-	transparentDropDown.SetBackgroundColor(window.src.backgroundColor)
+			Background(style.SelectedColor()).
+			Foreground(style.PrefixColor()))
+	transparentDropDown.SetBackgroundColor(style.BackgroundColor())
 	transparentDropDown.SetListStyles(tcell.StyleDefault.
-		Background(window.src.backgroundColor).
-		Foreground(window.src.foregroundColor),
+		Background(style.BackgroundColor()).
+		Foreground(style.ForegroundColor()),
 		tcell.StyleDefault.
-			Background(window.src.selectedColor).
-			Foreground(window.src.prefixColor))
-	srcBorderDropDown.SetBackgroundColor(window.src.backgroundColor)
+			Background(style.SelectedColor()).
+			Foreground(style.PrefixColor()))
+	srcBorderDropDown.SetBackgroundColor(style.BackgroundColor())
 	srcBorderDropDown.SetListStyles(tcell.StyleDefault.
-		Background(window.src.backgroundColor).
-		Foreground(window.src.foregroundColor),
+		Background(style.BackgroundColor()).
+		Foreground(style.ForegroundColor()),
 		tcell.StyleDefault.
-			Background(window.src.selectedColor).
-			Foreground(window.src.prefixColor))
-	dstBorderDropDown.SetBackgroundColor(window.dst.backgroundColor)
+			Background(style.SelectedColor()).
+			Foreground(style.PrefixColor()))
+	dstBorderDropDown.SetBackgroundColor(style.BackgroundColor())
 	dstBorderDropDown.SetListStyles(tcell.StyleDefault.
-		Background(window.src.backgroundColor).
-		Foreground(window.src.foregroundColor),
+		Background(style.BackgroundColor()).
+		Foreground(style.ForegroundColor()),
 		tcell.StyleDefault.
-			Background(window.src.selectedColor).
-			Foreground(window.src.prefixColor))
+			Background(style.SelectedColor()).
+			Foreground(style.PrefixColor()))
 }

 func updateBorderColor() {
 	// box
-	srcBox.SetBorderColor(window.src.borderColor).
-		SetTitleColor(window.src.borderColor)
-	dstBox.SetBorderColor(window.dst.borderColor).
-		SetTitleColor(window.dst.borderColor)
+	srcBox.SetBorderColor(style.SrcBorderColor()).
+		SetTitleColor(style.SrcBorderColor())
+	dstBox.SetBorderColor(style.DstBorderColor()).
+		SetTitleColor(style.DstBorderColor())

 	// 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)
+	srcLangDropDown.SetBorderColor(style.SrcBorderColor()).
+		SetTitleColor(style.SrcBorderColor())
+	dstLangDropDown.SetBorderColor(style.DstBorderColor()).
+		SetTitleColor(style.DstBorderColor())
+	srcBorderDropDown.SetBorderColor(style.SrcBorderColor()).
+		SetTitleColor(style.SrcBorderColor())
+	dstBorderDropDown.SetBorderColor(style.DstBorderColor()).
+		SetTitleColor(style.DstBorderColor())
 }

 func updateNonConfigColor() {
 	// box
 	srcBox.SetSelectedStyle(tcell.StyleDefault.
-		Background(window.src.selectedColor).
-		Foreground(window.src.foregroundColor))
-	dstBox.SetTextColor(window.dst.foregroundColor)
+		Background(style.SelectedColor()).
+		Foreground(style.ForegroundColor()))
+	dstBox.SetTextColor(style.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)
+	srcLangDropDown.SetFieldBackgroundColor(style.SelectedColor()).
+		SetFieldTextColor(style.ForegroundColor()).
+		SetPrefixTextColor(style.PrefixColor())
+	dstLangDropDown.SetFieldBackgroundColor(style.SelectedColor()).
+		SetFieldTextColor(style.ForegroundColor()).
+		SetPrefixTextColor(style.PrefixColor())
+	themeDropDown.SetLabelColor(style.LabelColor())
+	themeDropDown.SetFieldBackgroundColor(style.SelectedColor()).
+		SetFieldTextColor(style.ForegroundColor()).
+		SetPrefixTextColor(style.PrefixColor())
+	transparentDropDown.SetLabelColor(style.LabelColor())
+	transparentDropDown.SetFieldBackgroundColor(style.SelectedColor()).
+		SetFieldTextColor(style.ForegroundColor()).
+		SetPrefixTextColor(style.PrefixColor())
+	srcBorderDropDown.SetLabelColor(style.LabelColor())
+	srcBorderDropDown.SetFieldBackgroundColor(style.SelectedColor()).
+		SetFieldTextColor(style.ForegroundColor()).
+		SetPrefixTextColor(style.PrefixColor())
+	dstBorderDropDown.SetLabelColor(style.LabelColor())
+	dstBorderDropDown.SetFieldBackgroundColor(style.SelectedColor()).
+		SetFieldTextColor(style.ForegroundColor()).
+		SetPrefixTextColor(style.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)
+	langButton.SetLabelColor(style.ForegroundColor()).
+		SetBackgroundColorActivated(style.PressColor()).
+		SetLabelColorActivated(style.ForegroundColor()).
+		SetBackgroundColor(style.SelectedColor())
+	styleButton.SetLabelColor(style.ForegroundColor()).
+		SetBackgroundColorActivated(style.PressColor()).
+		SetLabelColorActivated(style.ForegroundColor()).
+		SetBackgroundColor(style.SelectedColor())
+	menuButton.SetLabelColor(style.ForegroundColor()).
+		SetBackgroundColorActivated(style.PressColor()).
+		SetLabelColorActivated(style.ForegroundColor()).
+		SetBackgroundColor(style.SelectedColor())
 }

 func updateAllColor() {
@ -277,15 +277,8 @@ func pagesHandler(event *tcell.EventKey) *tcell.EventKey {

 	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
+		updateBackgroundColor()
 		transparentDropDown.SetCurrentOption(
 			IndexOf(strconv.FormatBool(transparent),
 				[]string{"true", "false"}))
@ -422,18 +415,10 @@ func styleDropDownHandler(key tcell.Key) {

 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()
 }