feat: config system — TOML load/save with provider auth map

internal/config/config.go

@@ -0,0 +1,67 @@
+package config
+
+import (
+	"os"
+	"path/filepath"
+
+	"github.com/BurntSushi/toml"
+)
+
+type ProviderConfig struct {
+	APIKey          string `toml:"api_key,omitempty"`
+	AccessKeyID     string `toml:"access_key_id,omitempty"`
+	AccessKeySecret string `toml:"access_key_secret,omitempty"`
+	Endpoint        string `toml:"endpoint,omitempty"`
+}
+
+type Config struct {
+	CurrentProvider string                    `toml:"current_provider"`
+	Output          string                    `toml:"output"`
+	Providers       map[string]ProviderConfig `toml:"providers"`
+}
+
+func Load(path string) (*Config, error) {
+	cfg := &Config{
+		Output:    "table",
+		Providers: make(map[string]ProviderConfig),
+	}
+	data, err := os.ReadFile(path)
+	if err != nil {
+		if os.IsNotExist(err) {
+			return cfg, nil
+		}
+		return nil, err
+	}
+	if err := toml.Unmarshal(data, cfg); err != nil {
+		return nil, err
+	}
+	if cfg.Providers == nil {
+		cfg.Providers = make(map[string]ProviderConfig)
+	}
+	if cfg.Output == "" {
+		cfg.Output = "table"
+	}
+	return cfg, nil
+}
+
+func Save(path string, cfg *Config) error {
+	dir := filepath.Dir(path)
+	if err := os.MkdirAll(dir, 0700); err != nil {
+		return err
+	}
+	f, err := os.Create(path)
+	if err != nil {
+		return err
+	}
+	defer f.Close()
+	enc := toml.NewEncoder(f)
+	return enc.Encode(cfg)
+}
+
+func DefaultPath() string {
+	home, err := os.UserHomeDir()
+	if err != nil {
+		return "config.toml"
+	}
+	return filepath.Join(home, ".config", "ry", "config.toml")
+}

internal/config/config_test.go

@@ -0,0 +1,96 @@
+package config
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+)
+
+func TestLoadTOML(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "config.toml")
+	content := `
+current_provider = "rainyun"
+output = "table"
+
+[providers.rainyun]
+api_key = "sk-test123"
+
+[providers.aliyun]
+access_key_id = "ak-xxx"
+access_key_secret = "sk-xxx"
+`
+	if err := os.WriteFile(path, []byte(content), 0600); err != nil {
+		t.Fatal(err)
+	}
+
+	cfg, err := Load(path)
+	if err != nil {
+		t.Fatalf("Load() error: %v", err)
+	}
+	if cfg.CurrentProvider != "rainyun" {
+		t.Errorf("CurrentProvider = %q, want %q", cfg.CurrentProvider, "rainyun")
+	}
+	if cfg.Output != "table" {
+		t.Errorf("Output = %q, want %q", cfg.Output, "table")
+	}
+	if cfg.Providers["rainyun"].APIKey != "sk-test123" {
+		t.Errorf("rainyun api_key = %q, want %q", cfg.Providers["rainyun"].APIKey, "sk-test123")
+	}
+	if cfg.Providers["aliyun"].AccessKeyID != "ak-xxx" {
+		t.Errorf("aliyun access_key_id = %q, want %q", cfg.Providers["aliyun"].AccessKeyID, "ak-xxx")
+	}
+}
+
+func TestLoadDefaults(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "empty.toml")
+	if err := os.WriteFile(path, []byte(""), 0600); err != nil {
+		t.Fatal(err)
+	}
+	cfg, err := Load(path)
+	if err != nil {
+		t.Fatalf("Load() error: %v", err)
+	}
+	if cfg.Output != "table" {
+		t.Errorf("default Output = %q, want %q", cfg.Output, "table")
+	}
+}
+
+func TestSave(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "out.toml")
+	cfg := &Config{
+		CurrentProvider: "test",
+		Output:          "json",
+		Providers: map[string]ProviderConfig{
+			"test": {APIKey: "secret"},
+		},
+	}
+	if err := Save(path, cfg); err != nil {
+		t.Fatalf("Save() error: %v", err)
+	}
+	loaded, err := Load(path)
+	if err != nil {
+		t.Fatalf("re-Load() error: %v", err)
+	}
+	if loaded.CurrentProvider != "test" {
+		t.Errorf("roundtrip CurrentProvider = %q", loaded.CurrentProvider)
+	}
+}
+
+func TestDefaultPath(t *testing.T) {
+	path := DefaultPath()
+	if path == "" {
+		t.Error("DefaultPath() returned empty")
+	}
+}
+
+func TestEnsureDir(t *testing.T) {
+	dir := filepath.Join(t.TempDir(), "sub", "dir")
+	path := filepath.Join(dir, "config.toml")
+	cfg := &Config{Output: "table"}
+	if err := Save(path, cfg); err != nil {
+		t.Fatalf("Save() to deep dir error: %v", err)
+	}
+}