make language persistent in .keymaster config file

Author: ToeiRei

cmd/keymaster/main.go

@@ -82,14 +82,17 @@ func init() {
 	// Database flags
 	rootCmd.PersistentFlags().String("db-type", "sqlite", "Database type (e.g., sqlite, postgres)")
 	rootCmd.PersistentFlags().String("db-dsn", "./keymaster.db", "Database connection string (DSN)")
+	rootCmd.PersistentFlags().String("lang", "en", `TUI language ("en", "de")`)
 
 	// Bind flags to viper
 	viper.BindPFlag("database.type", rootCmd.PersistentFlags().Lookup("db-type"))
 	viper.BindPFlag("database.dsn", rootCmd.PersistentFlags().Lookup("db-dsn"))
+	viper.BindPFlag("language", rootCmd.PersistentFlags().Lookup("lang"))
 
 	// Set defaults in viper. These are used if not set in the config file or by flags.
 	viper.SetDefault("database.type", "sqlite")
 	viper.SetDefault("database.dsn", "./keymaster.db")
+	viper.SetDefault("language", "en")
 
 	// Add commands
 	rootCmd.AddCommand(deployCmd)
@@ -143,6 +146,9 @@ database:
   # For SQLite, this is the path to the database file.
   dsn: ./keymaster.db
 
+# The default language for the TUI. Supported: "en", "de".
+language: en
+
 # Example for future PostgreSQL configuration:
 # database:
 #   type: postgres

internal/i18n/i18n.go

@@ -25,12 +25,17 @@ var localeFS embed.FS
 // bundle stores all the loaded translation messages from the locale files.
 var bundle *i18n.Bundle
 
-// localizer is used to translate messages into a specific language.
-var localizer *i18n.Localizer
+var (
+	// localizer is used to translate messages into a specific language.
+	localizer *i18n.Localizer
+	// currentLang stores the tag of the currently active language.
+	currentLang string
+)
 
 // Init initializes the i18n bundle and sets up the localizer for a specific language.
 // It parses all embedded YAML files from the 'locales' directory.
 func Init(lang string) {
+	currentLang = lang
 	bundle = i18n.NewBundle(language.English)
 	bundle.RegisterUnmarshalFunc("yaml", yaml.Unmarshal)
 
@@ -66,3 +71,8 @@ func T(messageID string) string {
 func SetLang(lang string) {
 	Init(lang)
 }
+
+// GetLang returns the string tag of the currently active language.
+func GetLang() string {
+	return currentLang
+}

internal/tui/tui.go

@@ -17,6 +17,7 @@ import (
 	"github.com/toeirei/keymaster/internal/model"
 
 	tea "github.com/charmbracelet/bubbletea"
+	"github.com/spf13/viper"
 )
 
 // viewState represents which part of the UI is currently active.
@@ -263,6 +264,23 @@ func (m mainModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 					i18n.T("menu.view_audit_log"),
 					i18n.T("menu.view_accounts_by_tag"),
 				}
+				// Save the new language setting
+				viper.Set("language", i18n.GetLang())
+				if err := viper.WriteConfig(); err != nil {
+					// This is not a fatal error, so we just note it.
+					// The language will still change for the current session.
+					m.err = fmt.Errorf("could not save language setting: %w", err)
+				}
+				// Re-initialize models that depend on i18n strings
+				m.menu.choices = []string{
+					i18n.T("menu.manage_accounts"),
+					i18n.T("menu.manage_public_keys"),
+					i18n.T("menu.assign_keys"),
+					i18n.T("menu.rotate_system_keys"),
+					i18n.T("menu.deploy_to_fleet"),
+					i18n.T("menu.view_audit_log"),
+					i18n.T("menu.view_accounts_by_tag"),
+				}
 				return m, nil
 			}
 		}
@@ -394,6 +412,9 @@ func (m menuModel) View(data dashboardData, width int) string {
 
 // Run is the main entrypoint for the TUI. It initializes and runs the Bubble Tea program.
 func Run() {
+	// Initialize i18n with the language from config
+	i18n.Init(viper.GetString("language"))
+
 	if _, err := tea.NewProgram(initialModel()).Run(); err != nil {
 		fmt.Printf("Alas, there's been an error: %v", err)
 		os.Exit(1)