summaryrefslogtreecommitdiff
path: root/internal/config/config.go
blob: 71258c1c5e992e12899575d123c188cc2f992c26 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package config

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

	"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"`
	ShutdownTimeout  time.Duration `toml:"shutdown_timeout"`
	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
}