package config

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

	"gitcode.com/JianFeeeee/HomeAgent/pkg/types"
	"gopkg.in/yaml.v3"
)

const DefaultConfigPath = "/etc/homeagent/config.yaml"

func DefaultConfig() types.Config {
	return types.Config{
		Daemon: types.DaemonConfig{
			ListenAddr:        ":8080",
			DataDir:           "/var/lib/homeagent",
			HeartbeatInterval: 15 * time.Second,
			CheckInterval:     30 * time.Second,
			LogLevel:          "info",
		},
		LLM: types.LLMConfig{
			Provider:    "openai",
			Model:       "deepseek-v4-flash",
			BaseURL:     "https://api.deepseek.com",
			Temperature: 0.7,
			MaxTokens:   4096,
		},
		Defaults: types.AgentConfig{
			Image:  "homeagent/agent-base:latest",
			LLMEndpoints: []string{"https://api.openai.com/v1"},
			SnapshotPolicy: types.SnapshotPolicy{
				Interval:     10 * time.Minute,
				MaxSnapshots: 20,
				PreAction:    true,
				PostAction:   false,
			},
			RollbackPolicy: types.RollbackPolicy{
				MaxRetries:      3,
				HealthThreshold: types.HealthDown,
				CooldownPeriod:  30 * time.Second,
				AutoRollback:    true,
			},
			ResourceLimit: types.ResourceLimit{
				CPU:     "2",
				Memory:  "2g",
				Disk:    "10g",
				Network: true,
			},
			OpenClawEnabled: true,
		},
	}
}

func Load(path string) (*types.Config, error) {
	cfg := DefaultConfig()

	data, err := os.ReadFile(path)
	if err != nil {
		if os.IsNotExist(err) {
			return &cfg, nil
		}
		return nil, fmt.Errorf("read config: %w", err)
	}

	if err := yaml.Unmarshal(data, &cfg); err != nil {
		return nil, fmt.Errorf("parse config: %w", err)
	}

	return &cfg, nil
}

func Save(path string, cfg *types.Config) error {
	dir := filepath.Dir(path)
	if err := os.MkdirAll(dir, 0755); err != nil {
		return fmt.Errorf("create config dir: %w", err)
	}

	data, err := yaml.Marshal(cfg)
	if err != nil {
		return fmt.Errorf("marshal config: %w", err)
	}

	if err := os.WriteFile(path, data, 0644); err != nil {
		return fmt.Errorf("write config: %w", err)
	}

	return nil
}