init

config.go

@@ -0,0 +1,132 @@
+package main
+
+import (
+	"bufio"
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/pelletier/go-toml/v2"
+)
+
+type Config struct {
+	LLMProviderURL string `toml:"LLMProviderURL"`
+	APIKey         string `toml:"APIKey"`
+	ShellType      string `toml:"ShellType"`
+	ModelName      string `toml:"ModelName"`
+	ColorScheme    string `toml:"ColorScheme"`
+}
+
+func loadOrCreateConfig(path string) (*Config, error) {
+	// Check if config file exists
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		fmt.Printf("Config file %s does not exist.\n", path)
+		fmt.Printf("Would you like to create it? (Y/n) ")
+		
+		reader := bufio.NewReader(os.Stdin)
+		response, err := reader.ReadString('\n')
+		if err != nil {
+			return nil, fmt.Errorf("failed to read response: %v", err)
+		}
+		
+		response = strings.TrimSpace(strings.ToLower(response))
+		// Default to 'y' if empty, or check for 'n' to decline
+		if response == "n" || response == "no" {
+			return nil, fmt.Errorf("config file is required")
+		}
+		
+		// Create config directory if it doesn't exist
+		configDir := filepath.Dir(path)
+		if err := os.MkdirAll(configDir, 0755); err != nil {
+			return nil, fmt.Errorf("failed to create config directory: %v", err)
+		}
+		
+		// Get config values from user
+		config, err := promptForConfig()
+		if err != nil {
+			return nil, err
+		}
+		
+		// Write config file
+		file, err := os.Create(path)
+		if err != nil {
+			return nil, fmt.Errorf("failed to create config file: %v", err)
+		}
+		defer file.Close()
+		
+		encoder := toml.NewEncoder(file)
+		if err := encoder.Encode(config); err != nil {
+			return nil, fmt.Errorf("failed to write config file: %v", err)
+		}
+		
+		fmt.Printf("Config file created at %s\n", path)
+		return config, nil
+	}
+	
+	// Config file exists, load it
+	file, err := os.Open(path)
+	if err != nil {
+		return nil, fmt.Errorf("failed to open config file: %v", err)
+	}
+	defer file.Close()
+	
+	var config Config
+	decoder := toml.NewDecoder(file)
+	if err := decoder.Decode(&config); err != nil {
+		return nil, fmt.Errorf("failed to parse config file: %v", err)
+	}
+	
+	return &config, nil
+}
+
+func promptForConfig() (*Config, error) {
+	reader := bufio.NewReader(os.Stdin)
+	config := &Config{
+		LLMProviderURL: "https://openrouter.ai/api/v1/chat/completions",
+		ShellType:      "fish",
+		ModelName:      "anthropic/claude-sonnet-4",
+		ColorScheme:    "light",
+	}
+	
+	// Ask questions in sequence
+	fmt.Printf("LLM Provider URL [%s]: ", config.LLMProviderURL)
+	urlInput, _ := reader.ReadString('\n')
+	urlInput = strings.TrimSpace(urlInput)
+	if urlInput != "" {
+		config.LLMProviderURL = urlInput
+	}
+	
+	fmt.Print("API Key: ")
+	apiKey, _ := reader.ReadString('\n')
+	config.APIKey = strings.TrimSpace(apiKey)
+	if config.APIKey == "" {
+		return nil, fmt.Errorf("API key is required")
+	}
+	
+	fmt.Printf("Shell Type [%s]: ", config.ShellType)
+	shellInput, _ := reader.ReadString('\n')
+	shellInput = strings.TrimSpace(shellInput)
+	if shellInput != "" {
+		config.ShellType = shellInput
+	}
+	
+	fmt.Printf("Model Name [%s]: ", config.ModelName)
+	modelInput, _ := reader.ReadString('\n')
+	modelInput = strings.TrimSpace(modelInput)
+	if modelInput != "" {
+		config.ModelName = modelInput
+	}
+
+	fmt.Printf("Color Scheme (light/dark) [%s]: ", config.ColorScheme)
+	colorInput, _ := reader.ReadString('\n')
+	colorInput = strings.TrimSpace(colorInput)
+	if colorInput == "light" || colorInput == "dark" {
+		config.ColorScheme = colorInput
+	} else if colorInput != "" {
+		fmt.Println("Invalid color scheme, using default (light)")
+	}
+	
+	return config, nil
+}
+