327 lines
9.6 KiB
Go
327 lines
9.6 KiB
Go
// 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)
|
|
}
|