✨ Add extension management service
This commit is contained in:
278
internal/services/extension_service.go
Normal file
278
internal/services/extension_service.go
Normal file
@@ -0,0 +1,278 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
"sync"
|
||||
"voidraft/internal/models"
|
||||
|
||||
jsonparser "github.com/knadh/koanf/parsers/json"
|
||||
"github.com/knadh/koanf/providers/file"
|
||||
"github.com/knadh/koanf/providers/structs"
|
||||
"github.com/knadh/koanf/v2"
|
||||
"github.com/wailsapp/wails/v3/pkg/services/log"
|
||||
)
|
||||
|
||||
// ExtensionService 扩展管理服务
|
||||
type ExtensionService struct {
|
||||
koanf *koanf.Koanf
|
||||
logger *log.LoggerService
|
||||
pathManager *PathManager
|
||||
fileProvider *file.File
|
||||
|
||||
mu sync.RWMutex
|
||||
ctx context.Context
|
||||
cancel context.CancelFunc
|
||||
initOnce sync.Once
|
||||
|
||||
// 配置迁移服务
|
||||
migrationService *ConfigMigrationService[*models.ExtensionSettings]
|
||||
}
|
||||
|
||||
// ExtensionError 扩展错误
|
||||
type ExtensionError struct {
|
||||
Operation string
|
||||
Extension string
|
||||
Err error
|
||||
}
|
||||
|
||||
func (e *ExtensionError) Error() string {
|
||||
if e.Extension != "" {
|
||||
return fmt.Sprintf("extension %s for %s: %v", e.Operation, e.Extension, e.Err)
|
||||
}
|
||||
return fmt.Sprintf("extension %s: %v", e.Operation, e.Err)
|
||||
}
|
||||
|
||||
func (e *ExtensionError) Unwrap() error {
|
||||
return e.Err
|
||||
}
|
||||
|
||||
func (e *ExtensionError) Is(target error) bool {
|
||||
var extensionError *ExtensionError
|
||||
return errors.As(target, &extensionError)
|
||||
}
|
||||
|
||||
// NewExtensionService 创建扩展服务实例
|
||||
func NewExtensionService(logger *log.LoggerService, pathManager *PathManager) *ExtensionService {
|
||||
if logger == nil {
|
||||
logger = log.New()
|
||||
}
|
||||
if pathManager == nil {
|
||||
pathManager = NewPathManager()
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
|
||||
k := koanf.New(".")
|
||||
|
||||
migrationService := NewExtensionMigrationService(logger, pathManager)
|
||||
|
||||
service := &ExtensionService{
|
||||
koanf: k,
|
||||
logger: logger,
|
||||
pathManager: pathManager,
|
||||
ctx: ctx,
|
||||
cancel: cancel,
|
||||
migrationService: migrationService,
|
||||
}
|
||||
|
||||
// 异步初始化
|
||||
go service.initialize()
|
||||
|
||||
return service
|
||||
}
|
||||
|
||||
// initialize 初始化配置
|
||||
func (es *ExtensionService) initialize() {
|
||||
es.initOnce.Do(func() {
|
||||
if err := es.initConfig(); err != nil {
|
||||
es.logger.Error("failed to initialize extension config", "error", err)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// setDefaults 设置默认值
|
||||
func (es *ExtensionService) setDefaults() error {
|
||||
defaultConfig := models.NewDefaultExtensionSettings()
|
||||
|
||||
if err := es.koanf.Load(structs.Provider(defaultConfig, "json"), nil); err != nil {
|
||||
return &ExtensionError{"load_defaults", "", err}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// initConfig 初始化配置
|
||||
func (es *ExtensionService) initConfig() error {
|
||||
es.mu.Lock()
|
||||
defer es.mu.Unlock()
|
||||
|
||||
// 检查配置文件是否存在
|
||||
configPath := es.pathManager.GetExtensionsPath()
|
||||
if _, err := os.Stat(configPath); os.IsNotExist(err) {
|
||||
return es.createDefaultConfig()
|
||||
}
|
||||
|
||||
// 配置文件存在，先加载现有配置
|
||||
es.fileProvider = file.Provider(configPath)
|
||||
if err := es.koanf.Load(es.fileProvider, jsonparser.Parser()); err != nil {
|
||||
return &ExtensionError{"load_config_file", "", err}
|
||||
}
|
||||
|
||||
// 检查并执行配置迁移
|
||||
if es.migrationService != nil {
|
||||
result, err := es.migrationService.MigrateConfig(es.koanf)
|
||||
if err != nil {
|
||||
return &ExtensionError{"migrate_config", "", err}
|
||||
}
|
||||
|
||||
if result.Migrated && result.ConfigUpdated {
|
||||
// 迁移完成且配置已更新，重新创建文件提供器以监听新文件
|
||||
es.fileProvider = file.Provider(configPath)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// createDefaultConfig 创建默认配置文件
|
||||
func (es *ExtensionService) createDefaultConfig() error {
|
||||
if err := es.pathManager.EnsureConfigDir(); err != nil {
|
||||
return &ExtensionError{"create_config_dir", "", err}
|
||||
}
|
||||
|
||||
if err := es.setDefaults(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
configBytes, err := es.koanf.Marshal(jsonparser.Parser())
|
||||
if err != nil {
|
||||
return &ExtensionError{"marshal_config", "", err}
|
||||
}
|
||||
|
||||
if err := os.WriteFile(es.pathManager.GetExtensionsPath(), configBytes, 0644); err != nil {
|
||||
return &ExtensionError{"write_config", "", err}
|
||||
}
|
||||
|
||||
// 创建文件提供器
|
||||
es.fileProvider = file.Provider(es.pathManager.GetExtensionsPath())
|
||||
if err = es.koanf.Load(es.fileProvider, jsonparser.Parser()); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// saveConfig 保存配置到文件
|
||||
func (es *ExtensionService) saveExtensionConfig() error {
|
||||
configBytes, err := es.koanf.Marshal(jsonparser.Parser())
|
||||
if err != nil {
|
||||
return &ExtensionError{"marshal_config", "", err}
|
||||
}
|
||||
|
||||
if err := os.WriteFile(es.pathManager.GetExtensionsPath(), configBytes, 0644); err != nil {
|
||||
return &ExtensionError{"write_config", "", err}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetExtensionSettings 获取完整扩展配置
|
||||
func (es *ExtensionService) GetExtensionSettings() (*models.ExtensionSettings, error) {
|
||||
es.mu.RLock()
|
||||
defer es.mu.RUnlock()
|
||||
|
||||
var settings models.ExtensionSettings
|
||||
if err := es.koanf.Unmarshal("", &settings); err != nil {
|
||||
return nil, &ExtensionError{"unmarshal_config", "", err}
|
||||
}
|
||||
return &settings, nil
|
||||
}
|
||||
|
||||
// GetAllExtensions 获取所有扩展配置
|
||||
func (es *ExtensionService) GetAllExtensions() ([]models.Extension, error) {
|
||||
settings, err := es.GetExtensionSettings()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return settings.Extensions, nil
|
||||
}
|
||||
|
||||
// EnableExtension 启用扩展
|
||||
func (es *ExtensionService) EnableExtension(id models.ExtensionID) error {
|
||||
return es.UpdateExtensionState(id, true, nil)
|
||||
}
|
||||
|
||||
// DisableExtension 禁用扩展
|
||||
func (es *ExtensionService) DisableExtension(id models.ExtensionID) error {
|
||||
return es.UpdateExtensionState(id, false, nil)
|
||||
}
|
||||
|
||||
// UpdateExtensionState 更新扩展状态
|
||||
func (es *ExtensionService) UpdateExtensionState(id models.ExtensionID, enabled bool, config models.ExtensionConfig) error {
|
||||
es.mu.Lock()
|
||||
defer es.mu.Unlock()
|
||||
|
||||
// 获取当前配置
|
||||
var settings models.ExtensionSettings
|
||||
if err := es.koanf.Unmarshal("", &settings); err != nil {
|
||||
return &ExtensionError{"unmarshal_config", string(id), err}
|
||||
}
|
||||
|
||||
// 更新扩展状态
|
||||
if !settings.UpdateExtension(id, enabled, config) {
|
||||
return &ExtensionError{"extension_not_found", string(id), nil}
|
||||
}
|
||||
|
||||
// 重新加载到koanf
|
||||
if err := es.koanf.Load(structs.Provider(&settings, "json"), nil); err != nil {
|
||||
return &ExtensionError{"reload_config", string(id), err}
|
||||
}
|
||||
|
||||
// 保存到文件
|
||||
if err := es.saveExtensionConfig(); err != nil {
|
||||
return &ExtensionError{"save_config", string(id), err}
|
||||
}
|
||||
|
||||
es.logger.Info("extension state updated", "id", id, "enabled", enabled)
|
||||
return nil
|
||||
}
|
||||
|
||||
// ResetExtensionToDefault 重置扩展到默认状态
|
||||
func (es *ExtensionService) ResetExtensionToDefault(id models.ExtensionID) error {
|
||||
// 获取默认配置
|
||||
defaultSettings := models.NewDefaultExtensionSettings()
|
||||
defaultExtension := defaultSettings.GetExtensionByID(id)
|
||||
if defaultExtension == nil {
|
||||
return &ExtensionError{"default_extension_not_found", string(id), nil}
|
||||
}
|
||||
|
||||
return es.UpdateExtensionState(id, defaultExtension.Enabled, defaultExtension.Config)
|
||||
}
|
||||
|
||||
// ResetAllExtensionsToDefault 重置所有扩展到默认状态
|
||||
func (es *ExtensionService) ResetAllExtensionsToDefault() error {
|
||||
es.mu.Lock()
|
||||
defer es.mu.Unlock()
|
||||
|
||||
// 加载默认配置
|
||||
defaultSettings := models.NewDefaultExtensionSettings()
|
||||
if err := es.koanf.Load(structs.Provider(defaultSettings, "json"), nil); err != nil {
|
||||
return &ExtensionError{"load_defaults", "", err}
|
||||
}
|
||||
|
||||
// 保存到文件
|
||||
if err := es.saveExtensionConfig(); err != nil {
|
||||
return &ExtensionError{"save_config", "", err}
|
||||
}
|
||||
|
||||
es.logger.Info("all extensions reset to default")
|
||||
return nil
|
||||
}
|
||||
|
||||
// ServiceShutdown 关闭服务
|
||||
func (es *ExtensionService) ServiceShutdown() error {
|
||||
es.cancel()
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user