package config

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

	"github.com/BurntSushi/toml"
)

// Project represents a named workspace project used for webhook routing.
type Project struct {
	Name string `toml:"name"`
	Dir  string `toml:"dir"`
}

type Config struct {
	DataDir          string    `toml:"data_dir"`
	DBPath           string    `toml:"-"`
	LogDir           string    `toml:"-"`
	DropsDir         string    `toml:"-"`
	SSHAuthSock      string    `toml:"ssh_auth_sock"`
	ClaudeBinaryPath string    `toml:"claude_binary_path"`
	GeminiBinaryPath string    `toml:"gemini_binary_path"`
	ClaudeImage      string    `toml:"claude_image"`
	GeminiImage      string    `toml:"gemini_image"`
	MaxConcurrent    int       `toml:"max_concurrent"`
	DefaultTimeout   string    `toml:"default_timeout"`
	ServerAddr       string    `toml:"server_addr"`
	WebhookURL       string    `toml:"webhook_url"`
	WorkspaceRoot    string    `toml:"workspace_root"`
	WebhookSecret    string    `toml:"webhook_secret"`
	Projects         []Project `toml:"projects"`
	VAPIDPublicKey   string    `toml:"vapid_public_key"`
	VAPIDPrivateKey  string    `toml:"vapid_private_key"`
	VAPIDEmail       string    `toml:"vapid_email"`
	ClaudeConfigDir  string    `toml:"claude_config_dir"`
}

func Default() (*Config, error) {
	home, err := os.UserHomeDir()
	if err != nil {
		return nil, fmt.Errorf("cannot determine home directory: %w", err)
	}
	if home == "" {
		return nil, errors.New("cannot determine home directory: HOME is empty")
	}
	dataDir := filepath.Join(home, ".claudomator")
	return &Config{
		DataDir:          dataDir,
		DBPath:           filepath.Join(dataDir, "claudomator.db"),
		LogDir:           filepath.Join(dataDir, "executions"),
		DropsDir:         filepath.Join(dataDir, "drops"),
		SSHAuthSock:      os.Getenv("SSH_AUTH_SOCK"),
		ClaudeBinaryPath: "claude",
		GeminiBinaryPath: "gemini",
		ClaudeImage:      "claudomator-agent:latest",
		GeminiImage:      "claudomator-agent:latest",
		MaxConcurrent:    3,
		DefaultTimeout:   "15m",
		ServerAddr:       ":8484",
		WorkspaceRoot:    "/workspace",
		ClaudeConfigDir:  "/workspace/claudomator/credentials/claude",
	}, nil
}

// LoadFile loads a TOML config file on top of the defaults.
// Fields not present in the file retain their default values.
func LoadFile(path string) (*Config, error) {
	cfg, err := Default()
	if err != nil {
		return nil, err
	}
	if _, err := toml.DecodeFile(path, cfg); err != nil {
		return nil, fmt.Errorf("loading config file %q: %w", path, err)
	}
	return cfg, nil
}

// EnsureDirs creates the data directory structure.
func (c *Config) EnsureDirs() error {
	for _, dir := range []string{c.DataDir, c.LogDir, c.DropsDir} {
		if err := os.MkdirAll(dir, 0700); err != nil {
			return err
		}
	}
	return nil
}