landaiqing/voidraft
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 4 Wiki Activity

✨ Add extension management service

Browse Source
This commit is contained in:
landaiqing
2025-06-24 14:16:53 +08:00
parent ea025e3f5d
commit f3bcb87828
31 changed files with 1682 additions and 176 deletions
Download Patch File Download Diff File Expand all files Collapse all files

220
internal/models/extensions.go Normal file
View File

@@ -0,0 +1,220 @@
package models
import "time"
// Extension 单个扩展配置
type Extension struct {
ID ExtensionID `json:"id"` // 扩展唯一标识
Category ExtensionCategory `json:"category"` // 扩展分类
Enabled bool `json:"enabled"` // 是否启用
IsDefault bool `json:"isDefault"` // 是否为默认扩展
Config ExtensionConfig `json:"config"` // 扩展配置项
}
// ExtensionID 扩展标识符
type ExtensionID string
const (
// 编辑增强扩展
ExtensionRainbowBrackets ExtensionID = "rainbowBrackets" // 彩虹括号
ExtensionHyperlink ExtensionID = "hyperlink" // 超链接
ExtensionColorSelector ExtensionID = "colorSelector" // 颜色选择器
ExtensionFold ExtensionID = "fold"
ExtensionTextHighlight ExtensionID = "textHighlight"
// UI增强扩展
ExtensionMinimap ExtensionID = "minimap" // 小地图
ExtensionCodeBlast ExtensionID = "codeBlast" // 代码爆炸效果
// 工具扩展
ExtensionSearch ExtensionID = "search" // 搜索功能
ExtensionCodeBlock ExtensionID = "codeBlock" // 代码块
)
// ExtensionCategory 扩展分类
type ExtensionCategory string
const (
CategoryEditing ExtensionCategory = "editing" // 编辑增强
CategoryUI ExtensionCategory = "ui" // 界面增强
CategoryTools ExtensionCategory = "tools" // 工具类
)
// ExtensionConfig 扩展配置项（动态配置）
type ExtensionConfig map[string]interface{}
// ExtensionMetadata 扩展配置元数据
type ExtensionMetadata struct {
Version string `json:"version"` // 配置版本
LastUpdated string `json:"lastUpdated"` // 最后更新时间
}
// ExtensionSettings 扩展设置配置
type ExtensionSettings struct {
Extensions []Extension `json:"extensions"` // 扩展列表
Metadata ExtensionMetadata `json:"metadata"` // 配置元数据
}
// NewDefaultExtensionSettings 创建默认扩展配置
func NewDefaultExtensionSettings() *ExtensionSettings {
return &ExtensionSettings{
Extensions: NewDefaultExtensions(),
Metadata: ExtensionMetadata{
Version: "1.0.0",
LastUpdated: time.Now().Format(time.RFC3339),
},
}
}
// NewDefaultExtensions 创建默认扩展配置
func NewDefaultExtensions() []Extension {
return []Extension{
// 编辑增强扩展
{
ID: ExtensionRainbowBrackets,
Category: CategoryEditing,
Enabled: true,
IsDefault: true,
Config: ExtensionConfig{},
},
{
ID: ExtensionHyperlink,
Category: CategoryEditing,
Enabled: true,
IsDefault: true,
Config: ExtensionConfig{},
},
{
ID: ExtensionColorSelector,
Category: CategoryEditing,
Enabled: true,
IsDefault: true,
Config: ExtensionConfig{},
},
// UI增强扩展
{
ID: ExtensionMinimap,
Category: CategoryUI,
Enabled: true,
IsDefault: true,
Config: ExtensionConfig{
"displayText": "characters",
"showOverlay": "always",
"autohide": false,
},
},
{
ID: ExtensionCodeBlast,
Category: CategoryUI,
Enabled: true,
IsDefault: true,
Config: ExtensionConfig{
"effect": 1,
"shake": true,
"maxParticles": 300,
"shakeIntensity": 3,
},
},
// 工具扩展
{
ID: ExtensionSearch,
Category: CategoryTools,
Enabled: true,
IsDefault: true,
Config: ExtensionConfig{},
},
{
ID: ExtensionCodeBlock,
Category: CategoryTools,
Enabled: true,
IsDefault: true,
Config: ExtensionConfig{
"showBackground": true,
"enableAutoDetection": true,
},
},
{
ID: ExtensionFold,
Category: CategoryEditing,
Enabled: true,
IsDefault: true,
Config: ExtensionConfig{},
},
{
ID: ExtensionTextHighlight,
Category: CategoryEditing,
Enabled: true,
IsDefault: true,
Config: ExtensionConfig{},
},
}
}
// GetVersion 获取配置版本
func (es *ExtensionSettings) GetVersion() string {
return es.Metadata.Version
}
// SetVersion 设置配置版本
func (es *ExtensionSettings) SetVersion(version string) {
es.Metadata.Version = version
}
// SetLastUpdated 设置最后更新时间
func (es *ExtensionSettings) SetLastUpdated(timeStr string) {
es.Metadata.LastUpdated = timeStr
}
// GetDefaultConfig 获取默认配置
func (es *ExtensionSettings) GetDefaultConfig() any {
return NewDefaultExtensionSettings()
}
// GetExtensionByID 根据ID获取扩展
func (es *ExtensionSettings) GetExtensionByID(id ExtensionID) *Extension {
for i := range es.Extensions {
if es.Extensions[i].ID == id {
return &es.Extensions[i]
}
}
return nil
}
// GetEnabledExtensions 获取所有启用的扩展
func (es *ExtensionSettings) GetEnabledExtensions() []Extension {
var enabled []Extension
for _, ext := range es.Extensions {
if ext.Enabled {
enabled = append(enabled, ext)
}
}
return enabled
}
// GetExtensionsByCategory 根据分类获取扩展
func (es *ExtensionSettings) GetExtensionsByCategory(category ExtensionCategory) []Extension {
var extensions []Extension
for _, ext := range es.Extensions {
if ext.Category == category {
extensions = append(extensions, ext)
}
}
return extensions
}
// UpdateExtension 更新扩展配置
func (es *ExtensionSettings) UpdateExtension(id ExtensionID, enabled bool, config ExtensionConfig) bool {
for i := range es.Extensions {
if es.Extensions[i].ID == id {
es.Extensions[i].Enabled = enabled
if config != nil {
es.Extensions[i].Config = config
}
es.SetLastUpdated(time.Now().Format(time.RFC3339))
return true
}
}
return false
}

9
internal/services/config_migration_service.go
View File

@@ -22,6 +22,9 @@ const (
CurrentAppConfigVersion = "1.0.0"
// CurrentKeyBindingConfigVersion 当前快捷键配置版本
CurrentKeyBindingConfigVersion = "1.0.0"
CurrentExtensionConfigVersion = "1.0.0"
// BackupFilePattern 备份文件名模式
BackupFilePattern = "%s.backup.%s.json"
@@ -323,3 +326,9 @@ func NewKeyBindingMigrationService(logger *log.LoggerService, pathManager *PathM
return NewConfigMigrationService[*models.KeyBindingConfig](
logger, pathManager, "keybindings", CurrentKeyBindingConfigVersion, pathManager.GetKeybindsPath())
}
// NewExtensionMigrationService 创建扩展配置迁移服务
func NewExtensionMigrationService(logger *log.LoggerService, pathManager *PathManager) *ConfigMigrationService[*models.ExtensionSettings] {
return NewConfigMigrationService[*models.ExtensionSettings](
logger, pathManager, "extensions", CurrentExtensionConfigVersion, pathManager.GetExtensionsPath())
}

278
internal/services/extension_service.go Normal file
View 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
}

19
internal/services/path_manager.go
View File

@@ -7,9 +7,10 @@ import (
// PathManager 路径管理器
type PathManager struct {
configDir string // 配置目录
settingsPath string // 设置文件路径
keybindsPath string // 快捷键配置文件路径
configDir string // 配置目录
settingsPath string // 设置文件路径
keybindsPath string // 快捷键配置文件路径
extensionsPath string // 扩展配置文件路径
}
// NewPathManager 创建新的路径管理器
@@ -25,9 +26,10 @@ func NewPathManager() *PathManager {
configDir := filepath.Join(userConfigDir, ".voidraft", "config")
return &PathManager{
configDir: configDir,
settingsPath: filepath.Join(configDir, "settings.json"),
keybindsPath: filepath.Join(configDir, "keybindings.json"),
configDir: configDir,
settingsPath: filepath.Join(configDir, "settings.json"),
keybindsPath: filepath.Join(configDir, "keybindings.json"),
extensionsPath: filepath.Join(configDir, "extensions.json"),
}
}
@@ -46,6 +48,11 @@ func (pm *PathManager) GetConfigDir() string {
return pm.configDir
}
// GetExtensionsPath 获取扩展配置文件路径
func (pm *PathManager) GetExtensionsPath() string {
return pm.extensionsPath
}
// EnsureConfigDir 确保配置目录存在
func (pm *PathManager) EnsureConfigDir() error {
return os.MkdirAll(pm.configDir, 0755)

11
internal/services/service_manager.go
View File

@@ -18,6 +18,7 @@ type ServiceManager struct {
dialogService *DialogService
trayService *TrayService
keyBindingService *KeyBindingService
extensionService *ExtensionService
startupService *StartupService
updateService *UpdateService
logger *log.LoggerService
@@ -55,6 +56,9 @@ func NewServiceManager() *ServiceManager {
// 初始化快捷键服务
keyBindingService := NewKeyBindingService(logger, pathManager)
// 初始化扩展服务
extensionService := NewExtensionService(logger, pathManager)
// 初始化开机启动服务
startupService := NewStartupService(configService, logger)
@@ -92,6 +96,7 @@ func NewServiceManager() *ServiceManager {
dialogService: dialogService,
trayService: trayService,
keyBindingService: keyBindingService,
extensionService: extensionService,
startupService: startupService,
updateService: updateService,
logger: logger,
@@ -109,6 +114,7 @@ func (sm *ServiceManager) GetServices() []application.Service {
application.NewService(sm.dialogService),
application.NewService(sm.trayService),
application.NewService(sm.keyBindingService),
application.NewService(sm.extensionService),
application.NewService(sm.startupService),
application.NewService(sm.updateService),
}
@@ -149,6 +155,11 @@ func (sm *ServiceManager) GetStartupService() *StartupService {
return sm.startupService
}
// GetExtensionService 获取扩展服务实例
func (sm *ServiceManager) GetExtensionService() *ExtensionService {
return sm.extensionService
}
// GetUpdateService 获取更新服务实例
func (sm *ServiceManager) GetUpdateService() *UpdateService {
return sm.updateService