Merge pull request #18 from ai-coding-workshop/jx/config-option

feat: add config file support

Author: jiangxin

.dict-be.yml.example

@@ -0,0 +1,5 @@
+llm:
+  url: "https://api.openai.com/v1"
+  model: "gpt-4o-mini"
+  token: "your-api-token"
+  type: "openai"

internal/cli/root.go

@@ -3,6 +3,10 @@ package cli
 import (
 	"fmt"
 	"os"
+	"path/filepath"
+	"strings"
+
+	"dict-be/internal/config"
 
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
@@ -14,6 +18,10 @@ type Options struct {
 
 func NewRootCmd() *cobra.Command {
 	opts := &Options{}
+	defaultConfigPath := ""
+	if homeDir, err := os.UserHomeDir(); err == nil {
+		defaultConfigPath = filepath.Join(homeDir, ".dict-be.yml")
+	}
 	root := &cobra.Command{
 		Use:   "dict-be",
 		Short: "dict-be - CLI starter",
@@ -26,8 +34,8 @@ func NewRootCmd() *cobra.Command {
 	root.PersistentFlags().StringVar(
 		&opts.Config,
 		"config",
-		"",
-		"config file (default: ./dict-be.yaml)",
+		defaultConfigPath,
+		"config file (default: ~/.dict-be.yml)",
 	)
 	_ = viper.BindPFlag("config", root.PersistentFlags().Lookup("config"))
 
@@ -36,22 +44,46 @@ func NewRootCmd() *cobra.Command {
 }
 
 func initConfig(configFile string) {
+	configFile = strings.TrimSpace(configFile)
 	if configFile != "" {
-		viper.SetConfigFile(configFile)
+		viper.SetConfigFile(expandHome(configFile))
 	} else {
-		viper.SetConfigName("dict-be")
-		viper.SetConfigType("yaml")
-		viper.AddConfigPath(".")
-		viper.AddConfigPath("$HOME/.config/dict-be")
+		if homeDir, err := os.UserHomeDir(); err == nil {
+			viper.SetConfigFile(filepath.Join(homeDir, ".dict-be.yml"))
+		}
 	}
 
 	viper.SetEnvPrefix("DICT_BE")
 	viper.AutomaticEnv()
+	viper.SetDefault("llm.url", "")
+	viper.SetDefault("llm.model", "")
+	viper.SetDefault("llm.token", "")
+	viper.SetDefault("llm.type", "")
 
 	if err := viper.ReadInConfig(); err != nil {
 		if _, ok := err.(viper.ConfigFileNotFoundError); ok {
 			return
 		}
 		fmt.Fprintln(os.Stderr, err.Error())
+		return
+	}
+
+	if _, err := config.Load(); err != nil {
+		fmt.Fprintln(os.Stderr, err.Error())
+	}
+}
+
+func expandHome(path string) string {
+	if path == "~" {
+		if homeDir, err := os.UserHomeDir(); err == nil {
+			return homeDir
+		}
+		return path
+	}
+	if strings.HasPrefix(path, "~/") {
+		if homeDir, err := os.UserHomeDir(); err == nil {
+			return filepath.Join(homeDir, strings.TrimPrefix(path, "~/"))
+		}
 	}
+	return path
 }

internal/config/config.go

@@ -0,0 +1,41 @@
+package config
+
+import (
+	"fmt"
+
+	"github.com/spf13/viper"
+)
+
+type Config struct {
+	LLM LLMConfig `mapstructure:"llm"`
+}
+
+type LLMConfig struct {
+	URL   string `mapstructure:"url"`
+	Model string `mapstructure:"model"`
+	Token string `mapstructure:"token"`
+	Type  string `mapstructure:"type"`
+}
+
+func Load() (Config, error) {
+	var cfg Config
+	if err := viper.Unmarshal(&cfg); err != nil {
+		return cfg, err
+	}
+	if err := cfg.Validate(); err != nil {
+		return cfg, err
+	}
+	return cfg, nil
+}
+
+func (c Config) Validate() error {
+	if c.LLM.Type == "" {
+		return nil
+	}
+	switch c.LLM.Type {
+	case "openai", "anthropics", "gemini":
+		return nil
+	default:
+		return fmt.Errorf("invalid llm.type: %s", c.LLM.Type)
+	}
+}