refactor: 项目结构重组,src/ 扁平化为根目录,提取 pkg/ 子包
- 模块名重命名 yunshu -> hub.gaomia.site/titor/YunShu - Go 版本升级 1.21 -> 1.25 - src/ 目录删除,所有文件移至根目录 - 新增 pkg/mdprint/: Markdown AST 解析+ANSI 渲染 - 新增 pkg/style/: 终端颜色样式(8色 ANSI + 24位真彩色) - 新增 pkg/termui/: 终端输入组件(交互式输入/密码/确认) - 更新文档:AGENTS.md、architecture.md、changelog.md、taolun.md - gitignore 通配符修复 yunshu.exe -> yunshu.exe*
This commit is contained in:
157
config.go
Normal file
157
config.go
Normal file
@@ -0,0 +1,157 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
|
||||
"gopkg.in/yaml.v3"
|
||||
)
|
||||
|
||||
// Config 存储云枢 Agent 的所有配置
|
||||
type Config struct {
|
||||
LLM LLMConfig `yaml:"llm"`
|
||||
}
|
||||
|
||||
// LLMConfig 大模型连接配置
|
||||
type LLMConfig struct {
|
||||
Host string `yaml:"host"`
|
||||
Model string `yaml:"model"`
|
||||
Key string `yaml:"key"`
|
||||
}
|
||||
|
||||
// ConfigDir 返回全局配置目录路径 ~/.config/yunshu/
|
||||
func ConfigDir() string {
|
||||
home, err := os.UserHomeDir()
|
||||
if err != nil {
|
||||
return "./.config/yunshu"
|
||||
}
|
||||
return filepath.Join(home, ".config", "yunshu")
|
||||
}
|
||||
|
||||
// oldConfigDir 旧版本配置目录,用于迁移
|
||||
func oldConfigDir() string {
|
||||
home, err := os.UserHomeDir()
|
||||
if err != nil {
|
||||
return ""
|
||||
}
|
||||
return filepath.Join(home, ".config", "weather-cli")
|
||||
}
|
||||
|
||||
// migrateOldConfig 从旧目录 weather-cli 迁移配置到 yunshu
|
||||
func migrateOldConfig() {
|
||||
old := oldConfigDir()
|
||||
if old == "" {
|
||||
return
|
||||
}
|
||||
// 旧目录不存在,无需迁移
|
||||
if _, err := os.Stat(old); os.IsNotExist(err) {
|
||||
return
|
||||
}
|
||||
// 新目录已存在,说明已迁移过
|
||||
new := ConfigDir()
|
||||
if _, err := os.Stat(new); err == nil {
|
||||
return
|
||||
}
|
||||
|
||||
// 复制旧目录到新目录
|
||||
copyDirRecursive(old, new)
|
||||
os.RemoveAll(old)
|
||||
}
|
||||
|
||||
// LoadConfig 从 ~/.config/yunshu/config.yaml 读取配置
|
||||
// 如果新目录不存在,自动从旧 weather-cli 目录迁移
|
||||
func LoadConfig() (*Config, error) {
|
||||
// 尝试自动迁移旧配置
|
||||
migrateOldConfig()
|
||||
|
||||
path := filepath.Join(ConfigDir(), "config.yaml")
|
||||
data, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var cfg Config
|
||||
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &cfg, nil
|
||||
}
|
||||
|
||||
// SaveConfig 将配置写入 ~/.config/yunshu/config.yaml
|
||||
func SaveConfig(cfg *Config) error {
|
||||
dir := ConfigDir()
|
||||
if err := os.MkdirAll(dir, 0755); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
path := filepath.Join(dir, "config.yaml")
|
||||
data, err := yaml.Marshal(cfg)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 0600 - 仅所有者可读写,保护 API Key
|
||||
return os.WriteFile(path, data, 0600)
|
||||
}
|
||||
|
||||
// SearchFile 在项目目录和全局配置目录中搜索文件
|
||||
// 项目目录优先,全局配置目录作为后备
|
||||
func SearchFile(relativePath string) string {
|
||||
paths := []string{
|
||||
relativePath, // 项目目录
|
||||
filepath.Join(ConfigDir(), relativePath), // 全局配置目录
|
||||
}
|
||||
for _, p := range paths {
|
||||
if _, err := os.Stat(p); err == nil {
|
||||
return p
|
||||
}
|
||||
}
|
||||
return relativePath
|
||||
}
|
||||
|
||||
// CopyDefaultDir 将项目目录下的默认文件复制到全局配置目录
|
||||
func CopyDefaultDir(srcDir, dstDir string) {
|
||||
entries, err := os.ReadDir(srcDir)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
dstBase := filepath.Join(ConfigDir(), dstDir)
|
||||
for _, e := range entries {
|
||||
srcPath := filepath.Join(srcDir, e.Name())
|
||||
dstPath := filepath.Join(dstBase, e.Name())
|
||||
|
||||
if e.IsDir() {
|
||||
os.MkdirAll(dstPath, 0755)
|
||||
copyDirRecursive(srcPath, dstPath)
|
||||
} else {
|
||||
data, err := os.ReadFile(srcPath)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
os.MkdirAll(filepath.Dir(dstPath), 0755)
|
||||
os.WriteFile(dstPath, data, 0644)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func copyDirRecursive(src, dst string) {
|
||||
entries, err := os.ReadDir(src)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
os.MkdirAll(dst, 0755)
|
||||
for _, e := range entries {
|
||||
srcPath := filepath.Join(src, e.Name())
|
||||
dstPath := filepath.Join(dst, e.Name())
|
||||
if e.IsDir() {
|
||||
copyDirRecursive(srcPath, dstPath)
|
||||
} else {
|
||||
data, err := os.ReadFile(srcPath)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
os.WriteFile(dstPath, data, 0644)
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user