// Package config 定义应用配置的结构、默认值与磁盘持久化。
package config

import (
	"encoding/json"
	"fmt"
	"os"
	"path/filepath"
	"strings"

	"viewsagent/internal/capture"
)

// Profile 是一个 OpenAI 兼容协议的模型接入点。
type Profile struct {
	Name         string            `json:"name"`
	BaseURL      string            `json:"baseURL"`
	APIKey       string            `json:"apiKey"`
	Model        string            `json:"model"`
	ExtraHeaders map[string]string `json:"extraHeaders,omitempty"`
	MaxTokens    int               `json:"maxTokens"`
	Temperature  float64           `json:"temperature"`
}

// Capture 是抓屏与触发策略参数。
type Capture struct {
	FPS           float64 `json:"fps"`           // 监听帧率，1~5
	MaxEdge       int     `json:"maxEdge"`       // 发给模型的图片长边上限
	JPEGQuality   int     `json:"jpegQuality"`   // 1~100
	DiffThreshold int     `json:"diffThreshold"` // 感知哈希汉明距离阈值
	CooldownMs    int     `json:"cooldownMs"`    // 两次自动提问的最小间隔
	StableFrames  int     `json:"stableFrames"`  // 连续稳定帧数（去抖）
	HistoryTurns  int     `json:"historyTurns"`  // 带上的历史文本轮数
}

// Hotkeys 保存全局热键描述，语法如 "Ctrl+Alt+Q"。
type Hotkeys struct {
	SelectRegion string `json:"selectRegion"`
	ToggleWatch  string `json:"toggleWatch"`
}

// Template 是一套系统提示词。
type Template struct {
	Key    string `json:"key"`
	Name   string `json:"name"`
	System string `json:"system"`
}

// Config 是完整的应用配置。
type Config struct {
	ActiveProfile   string         `json:"activeProfile"`
	Profiles        []Profile      `json:"profiles"`
	Capture         Capture        `json:"capture"`
	Hotkeys         Hotkeys        `json:"hotkeys"`
	PromptTemplate  string         `json:"promptTemplate"`
	Templates       []Template     `json:"templates"`
	AlwaysOnTopCard bool           `json:"alwaysOnTopCard"`
	Region          capture.Region `json:"region"`
}

// DefaultTemplates 返回内置提示词模板。
func DefaultTemplates() []Template {
	return []Template{
		{
			Key:  "exercise",
			Name: "习题解答",
			System: `你是屏幕内容实时助手。用户会持续截取一块屏幕区域给你看，用途通常是做练习题、看试卷或读文档。

必须遵守：
1. 只针对画面中"当前这道题"给出答案，不要复述题目。
2. 如果画面没有出现新内容、或你无法辨认题目，只回复「(无新内容)」，绝对不要编造。
3. 先给结论，必要时补一行关键解析，整体控制在 5 行以内。
4. 数学公式只用纯文本和 Unicode 符号（如 x²=4、√2/2、≤、π、∫），禁止 LaTeX、禁止 Markdown 表格。
5. 题干信息不足时，直接说明缺少什么，不要猜。`,
		},
		{
			Key:  "tutor",
			Name: "逐步讲解",
			System: `你是屏幕辅导老师。用户会截取题目区域给你看。

必须遵守：
1. 分步骤讲解解题思路，每一步单独一行，最多 6 步。
2. 公式只用纯文本与 Unicode 符号（如 x²=4、√2/2、≤、π），禁止 LaTeX。
3. 画面没有新内容时只回复「(无新内容)」。
4. 不确定的地方明确指出，不要虚构条件。`,
		},
		{
			Key:  "code",
			Name: "代码解读",
			System: `你是代码解读助手。用户会截取代码区域给你看。

必须遵守：
1. 先用一句话说明这段代码在做什么。
2. 再指出明显的 bug、边界问题或性能风险。
3. 最后给出修改建议，用文字描述，不要输出整段重写的代码。
4. 画面没有新内容时只回复「(无新内容)」。`,
		},
		{
			Key:  "translate",
			Name: "翻译",
			System: `你是翻译助手。识别画面中的外文并翻译成简体中文，保持原有段落结构。

必须遵守：
1. 只输出译文，不要输出原文，也不要解释。
2. 画面没有新内容时只回复「(无新内容)」。`,
		},
		{
			Key:  "chat",
			Name: "自由对话",
			System: `你是桌面视觉助手。用户会把屏幕上的内容截给你看并提问，请用简洁准确的中文回答。
公式使用纯文本与 Unicode 符号，不要使用 LaTeX。`,
		},
	}
}

// Default 返回一份可直接使用的默认配置。
func Default() *Config {
	return &Config{
		ActiveProfile: "qwen",
		Profiles: []Profile{
			{
				Name: "qwen", BaseURL: "https://dashscope.aliyuncs.com/compatible-mode/v1",
				Model: "qwen3-vl-plus", MaxTokens: 1024, Temperature: 0.3,
			},
			{
				Name: "doubao", BaseURL: "https://ark.cn-beijing.volces.com/api/v3",
				Model: "doubao-seed-1-6-vision", MaxTokens: 1024, Temperature: 0.3,
			},
			{
				Name: "zhipu", BaseURL: "https://open.bigmodel.cn/api/paas/v4",
				Model: "glm-4.6v", MaxTokens: 1024, Temperature: 0.3,
			},
			{
				Name: "kimi", BaseURL: "https://api.moonshot.cn/v1",
				Model: "kimi-latest", MaxTokens: 1024, Temperature: 0.3,
			},
			{
				Name: "ollama", BaseURL: "http://localhost:11434/v1",
				Model: "qwen2.5vl:7b", MaxTokens: 1024, Temperature: 0.3,
			},
		},
		Capture: Capture{
			FPS: 2, MaxEdge: 1280, JPEGQuality: 75,
			DiffThreshold: 5, CooldownMs: 3000, StableFrames: 2, HistoryTurns: 6,
		},
		Hotkeys: Hotkeys{
			SelectRegion: "Ctrl+Alt+X",
			ToggleWatch:  "Ctrl+Alt+Space",
		},
		PromptTemplate:  "exercise",
		Templates:       DefaultTemplates(),
		AlwaysOnTopCard: true,
	}
}

// Profile 按名称查找接入点。
func (c *Config) Profile(name string) (Profile, bool) {
	for _, p := range c.Profiles {
		if p.Name == name {
			return p, true
		}
	}
	return Profile{}, false
}

// Active 返回当前启用的接入点，找不到时回退到第一个。
func (c *Config) Active() Profile {
	if p, ok := c.Profile(c.ActiveProfile); ok {
		return p
	}
	if len(c.Profiles) > 0 {
		return c.Profiles[0]
	}
	return Profile{}
}

// Template 按 key 查找提示词模板。
func (c *Config) Template(key string) Template {
	for _, t := range c.Templates {
		if t.Key == key {
			return t
		}
	}
	if len(c.Templates) > 0 {
		return c.Templates[0]
	}
	return Template{Key: "chat", Name: "自由对话", System: "你是桌面视觉助手。"}
}

// Clone 返回一份深拷贝，避免设置窗口与引擎共享同一份可变配置。
func (c *Config) Clone() *Config {
	if c == nil {
		return Default()
	}
	out := *c
	out.Profiles = append([]Profile(nil), c.Profiles...)
	for i := range out.Profiles {
		if c.Profiles[i].ExtraHeaders != nil {
			h := make(map[string]string, len(c.Profiles[i].ExtraHeaders))
			for k, v := range c.Profiles[i].ExtraHeaders {
				h[k] = v
			}
			out.Profiles[i].ExtraHeaders = h
		}
	}
	out.Templates = append([]Template(nil), c.Templates...)
	return &out
}

// Normalize 补全缺失字段，保证运行期不会出现零值。
func (c *Config) Normalize() {
	def := Default()
	if len(c.Profiles) == 0 {
		c.Profiles = def.Profiles
	}
	if c.ActiveProfile == "" {
		c.ActiveProfile = c.Profiles[0].Name
	}
	for i := range c.Profiles {
		p := &c.Profiles[i]
		if p.MaxTokens <= 0 {
			p.MaxTokens = 1024
		}
		if p.Temperature <= 0 {
			p.Temperature = 0.3
		}
		p.BaseURL = strings.TrimRight(strings.TrimSpace(p.BaseURL), "/")
	}
	if len(c.Templates) == 0 {
		c.Templates = def.Templates
	}
	if c.Capture.FPS <= 0 {
		c.Capture.FPS = 2
	}
	if c.Capture.MaxEdge <= 0 {
		c.Capture.MaxEdge = 1280
	}
	if c.Capture.JPEGQuality <= 0 {
		c.Capture.JPEGQuality = 75
	}
	if c.Capture.DiffThreshold <= 0 {
		c.Capture.DiffThreshold = 5
	}
	if c.Capture.CooldownMs <= 0 {
		c.Capture.CooldownMs = 3000
	}
	if c.Capture.StableFrames <= 0 {
		c.Capture.StableFrames = 2
	}
	if c.Capture.HistoryTurns <= 0 {
		c.Capture.HistoryTurns = 6
	}
	if c.Hotkeys.SelectRegion == "" {
		c.Hotkeys.SelectRegion = def.Hotkeys.SelectRegion
	}
	if c.Hotkeys.ToggleWatch == "" {
		c.Hotkeys.ToggleWatch = def.Hotkeys.ToggleWatch
	}
	if c.PromptTemplate == "" {
		c.PromptTemplate = c.Templates[0].Key
	}
}

// Store 负责配置文件的读写。
type Store struct {
	path string
}

// NewStore 创建配置文件存储，默认位于 %APPDATA%\viewsagent\config.json。
func NewStore() (*Store, error) {
	dir, err := os.UserConfigDir()
	if err != nil {
		return nil, fmt.Errorf("定位用户配置目录失败: %w", err)
	}
	base := filepath.Join(dir, "viewsagent")
	if err := os.MkdirAll(base, 0o755); err != nil {
		return nil, fmt.Errorf("创建配置目录失败: %w", err)
	}
	return &Store{path: filepath.Join(base, "config.json")}, nil
}

// Path 返回配置文件路径。
func (s *Store) Path() string { return s.path }

// Dir 返回配置目录。
func (s *Store) Dir() string { return filepath.Dir(s.path) }

// Load 读取配置；文件不存在或损坏时返回默认配置（损坏文件会被备份）。
func (s *Store) Load() *Config {
	raw, err := os.ReadFile(s.path)
	if err != nil {
		cfg := Default()
		cfg.Normalize()
		return cfg
	}
	cfg := Default()
	if err := json.Unmarshal(raw, cfg); err != nil {
		_ = os.WriteFile(s.path+".broken", raw, 0o644)
		fresh := Default()
		fresh.Normalize()
		return fresh
	}
	cfg.Normalize()
	cfg.migrate()
	return cfg
}

// migrate 把旧版本的默认值升级到当前默认值，避免老配置文件一直沿用已知冲突的热键。
func (c *Config) migrate() {
	if strings.EqualFold(strings.TrimSpace(c.Hotkeys.SelectRegion), "Ctrl+Alt+Q") {
		c.Hotkeys.SelectRegion = "Ctrl+Alt+X"
	}
}

// Save 原子写入配置文件。
func (s *Store) Save(cfg *Config) error {
	raw, err := json.MarshalIndent(cfg, "", "  ")
	if err != nil {
		return fmt.Errorf("序列化配置失败: %w", err)
	}
	tmp := s.path + ".tmp"
	if err := os.WriteFile(tmp, raw, 0o600); err != nil {
		return fmt.Errorf("写入临时配置失败: %w", err)
	}
	return os.Rename(tmp, s.path)
}