config.go

package main

import (
	"fmt"
	"os"
	"path/filepath"

	"github.com/spf13/viper"
)

// Config holds the application configuration.
type Config struct {
	ShowHidden      bool     `mapstructure:"show_hidden"`
	ShowIcons       bool     `mapstructure:"show_icons"`
	PreviewEnabled  bool     `mapstructure:"preview_enabled"`
	PreviewMaxLines int      `mapstructure:"preview_max_lines"`
	GitEnabled      bool     `mapstructure:"git_enabled"`
	Theme           string   `mapstructure:"theme"`
	SortBy          string   `mapstructure:"sort_by"`
	SortReverse     bool     `mapstructure:"sort_reverse"`
	Bookmarks       []string `mapstructure:"bookmarks"`
}

// DefaultConfig returns the default configuration.
func DefaultConfig() *Config {
	return &Config{
		ShowHidden:      false,
		ShowIcons:       true,
		PreviewEnabled:  true,
		PreviewMaxLines: 500,
		GitEnabled:      true,
		Theme:           "default",
		SortBy:          "name",
		SortReverse:     false,
	}
}

// ConfigPath returns the path to the configuration file.
func ConfigPath() string {
	dir, err := os.UserConfigDir()
	if err != nil {
		dir = "."
	}
	return filepath.Join(dir, "traverse", "config.yaml")
}

// LoadConfig loads the configuration from file.
func LoadConfig() (*Config, error) {
	cfg := DefaultConfig()

	v := viper.New()
	v.SetConfigFile(ConfigPath())

	// Set defaults
	v.SetDefault("show_hidden", cfg.ShowHidden)
	v.SetDefault("show_icons", cfg.ShowIcons)
	v.SetDefault("preview_enabled", cfg.PreviewEnabled)
	v.SetDefault("preview_max_lines", cfg.PreviewMaxLines)
	v.SetDefault("git_enabled", cfg.GitEnabled)
	v.SetDefault("theme", cfg.Theme)
	v.SetDefault("sort_by", cfg.SortBy)
	v.SetDefault("sort_reverse", cfg.SortReverse)

	// Try to read config file
	if err := v.ReadInConfig(); err != nil {
		if _, ok := err.(viper.ConfigFileNotFoundError); !ok {
			return nil, fmt.Errorf("error reading config: %w", err)
		}
		// Config file not found, use defaults - that's OK
	}

	if err := v.Unmarshal(cfg); err != nil {
		return nil, fmt.Errorf("error unmarshaling config: %w", err)
	}

	return cfg, nil
}

// SaveConfig saves the configuration to file.
func SaveConfig(cfg *Config) error {
	configFile := ConfigPath()
	dir := filepath.Dir(configFile)

	if err := os.MkdirAll(dir, 0755); err != nil {
		return fmt.Errorf("error creating config directory: %w", err)
	}

	v := viper.New()
	v.SetConfigFile(configFile)

	v.Set("show_hidden", cfg.ShowHidden)
	v.Set("show_icons", cfg.ShowIcons)
	v.Set("preview_enabled", cfg.PreviewEnabled)
	v.Set("preview_max_lines", cfg.PreviewMaxLines)
	v.Set("git_enabled", cfg.GitEnabled)
	v.Set("theme", cfg.Theme)
	v.Set("sort_by", cfg.SortBy)
	v.Set("sort_reverse", cfg.SortReverse)
	v.Set("bookmarks", cfg.Bookmarks)

	return v.WriteConfigAs(configFile)
}

// SortModeFromConfig returns the SortMode from config string.
func SortModeFromConfig(s string) SortMode {
	switch s {
	case "size":
		return SortBySize
	case "modified", "time":
		return SortByModified
	case "type":
		return SortByType
	default:
		return SortByName
	}
}