新增项目配置功能
This commit is contained in:
zyj
75
internal/config/config.go
Normal file
75
internal/config/config.go
Normal file
@@ -0,0 +1,75 @@
|
|||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"gopkg.in/yaml.v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
// AppConfig 应用配置结构体，对应 config.yaml
|
||||||
|
type AppConfig struct {
|
||||||
|
// 日志设置
|
||||||
|
LogLevel string `yaml:"log_level"` // 日志级别: trace|debug|info|warn|error
|
||||||
|
LogFile string `yaml:"log_file"` // 日志文件路径（可选）
|
||||||
|
|
||||||
|
// 默认 Profile
|
||||||
|
DefaultProfile string `yaml:"default_profile"`
|
||||||
|
|
||||||
|
// 还原选项
|
||||||
|
Restore RestoreConfig `yaml:"restore"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RestoreConfig 还原相关配置
|
||||||
|
type RestoreConfig struct {
|
||||||
|
ConflictStrategy string `yaml:"conflict_strategy"` // skip|overwrite|merge|prompt|newest
|
||||||
|
CreateRestorePoint bool `yaml:"create_restore_point"` // 是否创建还原点
|
||||||
|
DryRunFirst bool `yaml:"dry_run_first"` // 是否先执行干运行
|
||||||
|
}
|
||||||
|
|
||||||
|
// DefaultAppConfig 返回带有合理默认值的配置
|
||||||
|
func DefaultAppConfig() *AppConfig {
|
||||||
|
return &AppConfig{
|
||||||
|
LogLevel: "info",
|
||||||
|
DefaultProfile: "standard",
|
||||||
|
Restore: RestoreConfig{
|
||||||
|
ConflictStrategy: "prompt",
|
||||||
|
CreateRestorePoint: true,
|
||||||
|
DryRunFirst: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoadConfig 从指定路径加载配置文件，文件不存在时返回默认配置
|
||||||
|
func LoadConfig(path string) (*AppConfig, error) {
|
||||||
|
cfg := DefaultAppConfig()
|
||||||
|
|
||||||
|
data, err := os.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
// 配置文件不存在，返回默认配置
|
||||||
|
return cfg, nil
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("读取配置文件失败: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := yaml.Unmarshal(data, cfg); err != nil {
|
||||||
|
return nil, fmt.Errorf("解析配置文件失败: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return cfg, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// SaveConfig 将配置保存到指定路径
|
||||||
|
func SaveConfig(path string, cfg *AppConfig) error {
|
||||||
|
data, err := yaml.Marshal(cfg)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("序列化配置失败: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := os.WriteFile(path, data, 0o644); err != nil {
|
||||||
|
return fmt.Errorf("写入配置文件失败: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
329
internal/config/config_test.go
Normal file
329
internal/config/config_test.go
Normal file
@@ -0,0 +1,329 @@
|
|||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ==================== DefaultAppConfig 测试 ====================
|
||||||
|
|
||||||
|
func TestDefaultAppConfig(t *testing.T) {
|
||||||
|
cfg := DefaultAppConfig()
|
||||||
|
|
||||||
|
if cfg.LogLevel != "info" {
|
||||||
|
t.Errorf("默认 LogLevel 应为 info, got: %s", cfg.LogLevel)
|
||||||
|
}
|
||||||
|
if cfg.LogFile != "" {
|
||||||
|
t.Errorf("默认 LogFile 应为空, got: %s", cfg.LogFile)
|
||||||
|
}
|
||||||
|
if cfg.DefaultProfile != "standard" {
|
||||||
|
t.Errorf("默认 DefaultProfile 应为 standard, got: %s", cfg.DefaultProfile)
|
||||||
|
}
|
||||||
|
if cfg.Restore.ConflictStrategy != "prompt" {
|
||||||
|
t.Errorf("默认 ConflictStrategy 应为 prompt, got: %s", cfg.Restore.ConflictStrategy)
|
||||||
|
}
|
||||||
|
if !cfg.Restore.CreateRestorePoint {
|
||||||
|
t.Error("默认 CreateRestorePoint 应为 true")
|
||||||
|
}
|
||||||
|
if cfg.Restore.DryRunFirst {
|
||||||
|
t.Error("默认 DryRunFirst 应为 false")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ==================== LoadConfig 测试 ====================
|
||||||
|
|
||||||
|
func TestLoadConfig_FileNotExist(t *testing.T) {
|
||||||
|
cfg, err := LoadConfig("/not/exist/config.yaml")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("文件不存在时不应报错, got: %v", err)
|
||||||
|
}
|
||||||
|
// 应返回默认配置
|
||||||
|
if cfg.LogLevel != "info" {
|
||||||
|
t.Errorf("应返回默认 LogLevel, got: %s", cfg.LogLevel)
|
||||||
|
}
|
||||||
|
if cfg.DefaultProfile != "standard" {
|
||||||
|
t.Errorf("应返回默认 DefaultProfile, got: %s", cfg.DefaultProfile)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoadConfig_ValidFile(t *testing.T) {
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
path := filepath.Join(tmpDir, "config.yaml")
|
||||||
|
|
||||||
|
content := `log_level: debug
|
||||||
|
log_file: /tmp/devpack.log
|
||||||
|
default_profile: golang-dev
|
||||||
|
restore:
|
||||||
|
conflict_strategy: overwrite
|
||||||
|
create_restore_point: false
|
||||||
|
dry_run_first: true
|
||||||
|
`
|
||||||
|
if err := os.WriteFile(path, []byte(content), 0o644); err != nil {
|
||||||
|
t.Fatalf("写入测试文件失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
cfg, err := LoadConfig(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("加载配置不应报错, got: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if cfg.LogLevel != "debug" {
|
||||||
|
t.Errorf("LogLevel 应为 debug, got: %s", cfg.LogLevel)
|
||||||
|
}
|
||||||
|
if cfg.LogFile != "/tmp/devpack.log" {
|
||||||
|
t.Errorf("LogFile 应为 /tmp/devpack.log, got: %s", cfg.LogFile)
|
||||||
|
}
|
||||||
|
if cfg.DefaultProfile != "golang-dev" {
|
||||||
|
t.Errorf("DefaultProfile 应为 golang-dev, got: %s", cfg.DefaultProfile)
|
||||||
|
}
|
||||||
|
if cfg.Restore.ConflictStrategy != "overwrite" {
|
||||||
|
t.Errorf("ConflictStrategy 应为 overwrite, got: %s", cfg.Restore.ConflictStrategy)
|
||||||
|
}
|
||||||
|
if cfg.Restore.CreateRestorePoint {
|
||||||
|
t.Error("CreateRestorePoint 应为 false")
|
||||||
|
}
|
||||||
|
if !cfg.Restore.DryRunFirst {
|
||||||
|
t.Error("DryRunFirst 应为 true")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoadConfig_PartialFile(t *testing.T) {
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
path := filepath.Join(tmpDir, "config.yaml")
|
||||||
|
|
||||||
|
// 只设置部分字段，其余应保留默认值
|
||||||
|
content := `log_level: warn
|
||||||
|
`
|
||||||
|
if err := os.WriteFile(path, []byte(content), 0o644); err != nil {
|
||||||
|
t.Fatalf("写入测试文件失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
cfg, err := LoadConfig(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("加载配置不应报错, got: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 显式设置的字段
|
||||||
|
if cfg.LogLevel != "warn" {
|
||||||
|
t.Errorf("LogLevel 应为 warn, got: %s", cfg.LogLevel)
|
||||||
|
}
|
||||||
|
// 未设置的字段应保留默认值
|
||||||
|
if cfg.DefaultProfile != "standard" {
|
||||||
|
t.Errorf("未设置的 DefaultProfile 应保留默认值 standard, got: %s", cfg.DefaultProfile)
|
||||||
|
}
|
||||||
|
if cfg.Restore.ConflictStrategy != "prompt" {
|
||||||
|
t.Errorf("未设置的 ConflictStrategy 应保留默认值 prompt, got: %s", cfg.Restore.ConflictStrategy)
|
||||||
|
}
|
||||||
|
if !cfg.Restore.CreateRestorePoint {
|
||||||
|
t.Error("未设置的 CreateRestorePoint 应保留默认值 true")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoadConfig_InvalidYAML(t *testing.T) {
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
path := filepath.Join(tmpDir, "config.yaml")
|
||||||
|
|
||||||
|
content := `log_level: [invalid yaml
|
||||||
|
this is broken`
|
||||||
|
if err := os.WriteFile(path, []byte(content), 0o644); err != nil {
|
||||||
|
t.Fatalf("写入测试文件失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err := LoadConfig(path)
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("无效 YAML 应返回错误")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoadConfig_EmptyFile(t *testing.T) {
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
path := filepath.Join(tmpDir, "config.yaml")
|
||||||
|
|
||||||
|
if err := os.WriteFile(path, []byte(""), 0o644); err != nil {
|
||||||
|
t.Fatalf("写入测试文件失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
cfg, err := LoadConfig(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("空文件不应报错, got: %v", err)
|
||||||
|
}
|
||||||
|
// 空文件应返回默认配置
|
||||||
|
if cfg.LogLevel != "info" {
|
||||||
|
t.Errorf("空文件应保留默认 LogLevel, got: %s", cfg.LogLevel)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ==================== SaveConfig 测试 ====================
|
||||||
|
|
||||||
|
func TestSaveConfig(t *testing.T) {
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
path := filepath.Join(tmpDir, "config.yaml")
|
||||||
|
|
||||||
|
cfg := &AppConfig{
|
||||||
|
LogLevel: "debug",
|
||||||
|
LogFile: "/var/log/devpack.log",
|
||||||
|
DefaultProfile: "full",
|
||||||
|
Restore: RestoreConfig{
|
||||||
|
ConflictStrategy: "skip",
|
||||||
|
CreateRestorePoint: true,
|
||||||
|
DryRunFirst: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := SaveConfig(path, cfg); err != nil {
|
||||||
|
t.Fatalf("保存配置不应报错, got: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 验证文件已创建
|
||||||
|
if _, err := os.Stat(path); os.IsNotExist(err) {
|
||||||
|
t.Fatal("配置文件应已被创建")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 重新加载验证内容一致
|
||||||
|
loaded, err := LoadConfig(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("重新加载不应报错, got: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if loaded.LogLevel != cfg.LogLevel {
|
||||||
|
t.Errorf("LogLevel 不匹配: want %s, got %s", cfg.LogLevel, loaded.LogLevel)
|
||||||
|
}
|
||||||
|
if loaded.LogFile != cfg.LogFile {
|
||||||
|
t.Errorf("LogFile 不匹配: want %s, got %s", cfg.LogFile, loaded.LogFile)
|
||||||
|
}
|
||||||
|
if loaded.DefaultProfile != cfg.DefaultProfile {
|
||||||
|
t.Errorf("DefaultProfile 不匹配: want %s, got %s", cfg.DefaultProfile, loaded.DefaultProfile)
|
||||||
|
}
|
||||||
|
if loaded.Restore.ConflictStrategy != cfg.Restore.ConflictStrategy {
|
||||||
|
t.Errorf("ConflictStrategy 不匹配: want %s, got %s", cfg.Restore.ConflictStrategy, loaded.Restore.ConflictStrategy)
|
||||||
|
}
|
||||||
|
if loaded.Restore.CreateRestorePoint != cfg.Restore.CreateRestorePoint {
|
||||||
|
t.Errorf("CreateRestorePoint 不匹配: want %v, got %v", cfg.Restore.CreateRestorePoint, loaded.Restore.CreateRestorePoint)
|
||||||
|
}
|
||||||
|
if loaded.Restore.DryRunFirst != cfg.Restore.DryRunFirst {
|
||||||
|
t.Errorf("DryRunFirst 不匹配: want %v, got %v", cfg.Restore.DryRunFirst, loaded.Restore.DryRunFirst)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSaveConfig_InvalidPath(t *testing.T) {
|
||||||
|
cfg := DefaultAppConfig()
|
||||||
|
// 写入不存在的目录应报错
|
||||||
|
err := SaveConfig("/not/exist/dir/config.yaml", cfg)
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("写入不存在的目录应返回错误")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ==================== Save → Load 往返测试 ====================
|
||||||
|
|
||||||
|
func TestSaveAndLoad_RoundTrip(t *testing.T) {
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
path := filepath.Join(tmpDir, "roundtrip.yaml")
|
||||||
|
|
||||||
|
original := &AppConfig{
|
||||||
|
LogLevel: "trace",
|
||||||
|
LogFile: "logs/app.log",
|
||||||
|
DefaultProfile: "minimal",
|
||||||
|
Restore: RestoreConfig{
|
||||||
|
ConflictStrategy: "newest",
|
||||||
|
CreateRestorePoint: false,
|
||||||
|
DryRunFirst: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
// Save
|
||||||
|
if err := SaveConfig(path, original); err != nil {
|
||||||
|
t.Fatalf("SaveConfig 失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Load
|
||||||
|
loaded, err := LoadConfig(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("LoadConfig 失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 逐字段对比
|
||||||
|
if *loaded != *original {
|
||||||
|
t.Errorf("往返后配置不一致:\n 原始: %+v\n 加载: %+v", original, loaded)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ==================== DefaultPaths 测试 ====================
|
||||||
|
|
||||||
|
func TestDefaultPaths(t *testing.T) {
|
||||||
|
paths, err := DefaultPaths()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("DefaultPaths 不应报错, got: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
home, _ := os.UserHomeDir()
|
||||||
|
expected := filepath.Join(home, ".devpack")
|
||||||
|
|
||||||
|
if paths.Home != expected {
|
||||||
|
t.Errorf("Home 应为 %s, got: %s", expected, paths.Home)
|
||||||
|
}
|
||||||
|
if paths.Config != filepath.Join(expected, "config.yaml") {
|
||||||
|
t.Errorf("Config 路径不正确, got: %s", paths.Config)
|
||||||
|
}
|
||||||
|
if paths.Profiles != filepath.Join(expected, "profiles") {
|
||||||
|
t.Errorf("Profiles 路径不正确, got: %s", paths.Profiles)
|
||||||
|
}
|
||||||
|
if paths.Packs != filepath.Join(expected, "packs") {
|
||||||
|
t.Errorf("Packs 路径不正确, got: %s", paths.Packs)
|
||||||
|
}
|
||||||
|
if paths.Logs != filepath.Join(expected, "logs") {
|
||||||
|
t.Errorf("Logs 路径不正确, got: %s", paths.Logs)
|
||||||
|
}
|
||||||
|
if paths.Temp != filepath.Join(expected, "tmp") {
|
||||||
|
t.Errorf("Temp 路径不正确, got: %s", paths.Temp)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEnsureDirs(t *testing.T) {
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
|
||||||
|
paths := &Paths{
|
||||||
|
Home: filepath.Join(tmpDir, ".devpack"),
|
||||||
|
Profiles: filepath.Join(tmpDir, ".devpack", "profiles"),
|
||||||
|
Packs: filepath.Join(tmpDir, ".devpack", "packs"),
|
||||||
|
Logs: filepath.Join(tmpDir, ".devpack", "logs"),
|
||||||
|
Temp: filepath.Join(tmpDir, ".devpack", "tmp"),
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := paths.EnsureDirs(); err != nil {
|
||||||
|
t.Fatalf("EnsureDirs 不应报错, got: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 验证所有目录都已创建
|
||||||
|
dirs := []string{paths.Home, paths.Profiles, paths.Packs, paths.Logs, paths.Temp}
|
||||||
|
for _, dir := range dirs {
|
||||||
|
info, err := os.Stat(dir)
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
t.Errorf("目录应已创建: %s", dir)
|
||||||
|
}
|
||||||
|
if err == nil && !info.IsDir() {
|
||||||
|
t.Errorf("应为目录而非文件: %s", dir)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEnsureDirs_Idempotent(t *testing.T) {
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
|
||||||
|
paths := &Paths{
|
||||||
|
Home: filepath.Join(tmpDir, ".devpack"),
|
||||||
|
Profiles: filepath.Join(tmpDir, ".devpack", "profiles"),
|
||||||
|
Packs: filepath.Join(tmpDir, ".devpack", "packs"),
|
||||||
|
Logs: filepath.Join(tmpDir, ".devpack", "logs"),
|
||||||
|
Temp: filepath.Join(tmpDir, ".devpack", "tmp"),
|
||||||
|
}
|
||||||
|
|
||||||
|
// 调用两次应幂等
|
||||||
|
if err := paths.EnsureDirs(); err != nil {
|
||||||
|
t.Fatalf("第一次 EnsureDirs 失败: %v", err)
|
||||||
|
}
|
||||||
|
if err := paths.EnsureDirs(); err != nil {
|
||||||
|
t.Fatalf("第二次 EnsureDirs 应幂等, got: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user