feat: custom theme support

go.mod

@@ -12,6 +12,7 @@ require (
 )
 
 require (
+	github.com/BurntSushi/toml v1.5.0 // indirect
 	github.com/atotto/clipboard v0.1.4 // indirect
 	github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
 	github.com/charmbracelet/colorprofile v0.3.2 // indirect

go.sum

@@ -1,3 +1,5 @@
+github.com/BurntSushi/toml v1.5.0 h1:W5quZX/G/csjUnuI8SUYlsHs9M38FC7znL0lIO+DvMg=
+github.com/BurntSushi/toml v1.5.0/go.mod h1:ukJfTF/6rtPPRCnwkur4qwRxa8vTRFBF0uk2lLoLwho=
 github.com/MakeNowJust/heredoc v1.0.0 h1:cXCdzVdstXyiTqTvfqk9SDHpKNjxuom+DOlyEeQ4pzQ=
 github.com/MakeNowJust/heredoc v1.0.0/go.mod h1:mG5amYoWBHf8vpLOuehzbGGw0EHxpZZ6lCpQ4fNJ8LE=
 github.com/atotto/clipboard v0.1.4 h1:EH0zSVneZPSuFR11BlR9YppQTVDbh5+16AmcJi4g1z4=

internal/tui/model.go

@@ -51,7 +51,26 @@ type Model struct {
 
 // initialModel creates the initial state of the application.
 func initialModel() Model {
-	themeNames := ThemeNames()
+	themeNames := ThemeNames() //built-in themes load
+	cfg, _ := load_config()
+
+	var selectedThemeName string
+	if t, ok := Themes[cfg.Theme]; ok{
+		selectedThemeName = cfg.Theme
+		_ = t // to avoid unused variable warning
+	} else{
+		if _, err := load_custom_theme(cfg.Theme); err == nil{
+			selectedThemeName = cfg.Theme 
+		} else{
+			//fallback
+			selectedThemeName = themeNames[0]
+		}
+	}
+
+	themeNames = ThemeNames() // reload
+
+	// fmt.Println("All themes:", themeNames)
+
 	gc := git.NewGitCommands()
 	repoName, branchName, _ := gc.GetRepoInfo()
 	initialContent := initialContentLoading
@@ -77,9 +96,9 @@ func initialModel() Model {
 	ta.SetHeight(5)
 
 	return Model{
-		theme:             Themes[themeNames[0]],
+		theme:             Themes[selectedThemeName],
 		themeNames:        themeNames,
-		themeIndex:        0,
+		themeIndex:        indexOf(themeNames, selectedThemeName),
 		focusedPanel:      StatusPanel,
 		activeSourcePanel: StatusPanel,
 		help:              help.New(),
@@ -95,6 +114,15 @@ func initialModel() Model {
 	}
 }
 
+func indexOf(arr []string, val string) int{
+	for i, s := range arr{
+		if s == val{
+			return i
+		}
+	}
+	return 0
+}
+
 // Init is the first command that is run when the program starts.
 func (m Model) Init() tea.Cmd {
 	// fetch initial content for all panels.

internal/tui/theme.go

@@ -4,8 +4,19 @@ import (
 	"sort"
 
 	"github.com/charmbracelet/lipgloss"
+
+	"os"
+
+	"path/filepath"
+
+	"github.com/BurntSushi/toml"
+
+	"fmt"
 )
 
+// DefaultThemeName is the name of the default theme.
+const DefaultThemeName = "GitHub Dark"
+
 // Palette defines a set of colors for a theme.
 type Palette struct {
 	Black, Red, Green, Yellow, Blue, Magenta, Cyan, White,
@@ -139,6 +150,20 @@ type TreeStyle struct {
 	Connector, ConnectorLast, Prefix, PrefixLast string
 }
 
+//config.toml
+type themeConfig struct{
+	Theme string `toml:"theme"` 
+}
+
+// custom_theme.toml
+type ThemeFile struct{
+	Fg	string	`toml:"fg"`
+	Bg	string	`toml:"bg"`
+	Normal map[string]string `toml:"normal"`
+	Bright map[string]string `toml:"bright"`
+	Dark map[string]string `toml:"dark"`
+}
+
 // Themes holds all the available themes, generated from palettes.
 var Themes = map[string]Theme{}
 
@@ -216,3 +241,49 @@ func ThemeNames() []string {
 	sort.Strings(names)
 	return names
 }
+
+func load_config() (*themeConfig, error){
+	cfgPath := filepath.Join(os.Getenv("HOME"), ".config", "gitx", "config.toml")
+	if _,err := os.Stat(cfgPath); os.IsNotExist(err) {
+		return &themeConfig{Theme: DefaultThemeName}, nil //fallback
+	}
+
+	var cfg themeConfig
+	if _, err := toml.DecodeFile(cfgPath, &cfg); err != nil {
+		return nil, err
+	}
+
+	return &cfg, nil
+}
+
+func load_custom_theme(name string) (*Palette, error){
+	themePath := filepath.Join(os.Getenv("HOME"), ".config", "gitx", "themes", name+".toml")
+	if _,err := os.Stat(themePath); os.IsNotExist(err) {
+		return nil, fmt.Errorf("theme not found: %s", name)
+	}
+
+	var tf ThemeFile
+	if _, err := toml.DecodeFile(themePath, &tf); err != nil {
+		return nil, err
+	}
+
+	// Create a Palette from the ThemeFile
+	p := Palette{
+		Fg: tf.Fg,
+		Bg: tf.Bg,
+		Black: tf.Normal["Black"], Red: tf.Normal["Red"], Green: tf.Normal["Green"], Yellow: tf.Normal["Yellow"],
+        Blue: tf.Normal["Blue"], Magenta: tf.Normal["Magenta"], Cyan: tf.Normal["Cyan"], White: tf.Normal["White"],
+
+        BrightBlack: tf.Bright["Black"], BrightRed: tf.Bright["Red"], BrightGreen: tf.Bright["Green"], BrightYellow: tf.Bright["Yellow"],
+        BrightBlue: tf.Bright["Blue"], BrightMagenta: tf.Bright["Magenta"], BrightCyan: tf.Bright["Cyan"], BrightWhite: tf.Bright["White"],
+
+        DarkBlack: tf.Dark["Black"], DarkRed: tf.Dark["Red"], DarkGreen: tf.Dark["Green"], DarkYellow: tf.Dark["Yellow"],
+        DarkBlue: tf.Dark["Blue"], DarkMagenta: tf.Dark["Magenta"], DarkCyan: tf.Dark["Cyan"], DarkWhite: tf.Dark["White"],
+
+	}
+
+	Palettes[name] = p // Add to Palettes map for future use
+	Themes[name] = NewThemeFromPalette(p) // Add to Themes map
+
+	return &p, nil
+}