✨ Added preset theme
This commit is contained in:
@@ -3,6 +3,7 @@ package services
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"errors"
|
||||
"fmt"
|
||||
"time"
|
||||
"voidraft/internal/models"
|
||||
@@ -43,96 +44,141 @@ func (ts *ThemeService) getDB() *sql.DB {
|
||||
return ts.databaseService.db
|
||||
}
|
||||
|
||||
// initializeDefaultThemes 初始化默认主题
|
||||
// initializeDefaultThemes 初始化所有预设主题
|
||||
func (ts *ThemeService) initializeDefaultThemes() error {
|
||||
db := ts.getDB()
|
||||
if db == nil {
|
||||
return fmt.Errorf("database not available")
|
||||
}
|
||||
|
||||
// 检查是否已存在默认主题
|
||||
var count int
|
||||
err := db.QueryRow("SELECT COUNT(*) FROM themes WHERE is_default = 1").Scan(&count)
|
||||
// 获取所有已存在的主题名称
|
||||
existingThemes := make(map[string]bool)
|
||||
rows, err := db.Query("SELECT name FROM themes")
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to check existing themes: %w", err)
|
||||
}
|
||||
|
||||
if count > 0 {
|
||||
return nil // 默认主题已存在
|
||||
}
|
||||
|
||||
// 创建默认深色主题
|
||||
now := time.Now().Format("2006-01-02 15:04:05")
|
||||
darkTheme := &models.Theme{
|
||||
Name: "Default Dark",
|
||||
Type: models.ThemeTypeDark,
|
||||
Colors: *models.NewDefaultDarkTheme(),
|
||||
IsDefault: true,
|
||||
CreatedAt: now,
|
||||
UpdatedAt: now,
|
||||
}
|
||||
|
||||
// 创建默认浅色主题
|
||||
lightTheme := &models.Theme{
|
||||
Name: "Default Light",
|
||||
Type: models.ThemeTypeLight,
|
||||
Colors: *models.NewDefaultLightTheme(),
|
||||
IsDefault: true,
|
||||
CreatedAt: now,
|
||||
UpdatedAt: now,
|
||||
}
|
||||
|
||||
// 插入默认主题
|
||||
if _, err := ts.CreateTheme(darkTheme); err != nil {
|
||||
return fmt.Errorf("failed to create default dark theme: %w", err)
|
||||
}
|
||||
|
||||
if _, err := ts.CreateTheme(lightTheme); err != nil {
|
||||
return fmt.Errorf("failed to create default light theme: %w", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetDefaultThemes 获取默认主题
|
||||
func (ts *ThemeService) GetDefaultThemes() (map[string]*models.Theme, error) {
|
||||
query := `
|
||||
SELECT id, name, type, colors, is_default, created_at, updated_at
|
||||
FROM themes
|
||||
WHERE is_default = 1
|
||||
ORDER BY type
|
||||
`
|
||||
|
||||
db := ts.getDB()
|
||||
rows, err := db.Query(query)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to query default themes: %w", err)
|
||||
return fmt.Errorf("failed to query existing themes: %w", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
themes := make(map[string]*models.Theme)
|
||||
for rows.Next() {
|
||||
theme := &models.Theme{}
|
||||
err := rows.Scan(
|
||||
&theme.ID,
|
||||
&theme.Name,
|
||||
&theme.Type,
|
||||
&theme.Colors,
|
||||
&theme.IsDefault,
|
||||
&theme.CreatedAt,
|
||||
&theme.UpdatedAt,
|
||||
var name string
|
||||
if err := rows.Scan(&name); err != nil {
|
||||
return fmt.Errorf("failed to scan theme name: %w", err)
|
||||
}
|
||||
existingThemes[name] = true
|
||||
}
|
||||
|
||||
// 定义所有预设主题配置
|
||||
now := time.Now().Format("2006-01-02 15:04:05")
|
||||
presetThemes := []struct {
|
||||
config *models.ThemeColorConfig
|
||||
themeType models.ThemeType
|
||||
isDefault bool
|
||||
}{
|
||||
// 默认主题
|
||||
{models.NewDefaultDarkTheme(), models.ThemeTypeDark, true},
|
||||
{models.NewDefaultLightTheme(), models.ThemeTypeLight, true},
|
||||
|
||||
// 深色主题预设
|
||||
{models.NewDraculaTheme(), models.ThemeTypeDark, false},
|
||||
{models.NewAuraTheme(), models.ThemeTypeDark, false},
|
||||
{models.NewGitHubDarkTheme(), models.ThemeTypeDark, false},
|
||||
{models.NewMaterialDarkTheme(), models.ThemeTypeDark, false},
|
||||
{models.NewOneDarkTheme(), models.ThemeTypeDark, false},
|
||||
{models.NewSolarizedDarkTheme(), models.ThemeTypeDark, false},
|
||||
{models.NewTokyoNightTheme(), models.ThemeTypeDark, false},
|
||||
{models.NewTokyoNightStormTheme(), models.ThemeTypeDark, false},
|
||||
|
||||
// 浅色主题预设
|
||||
{models.NewGitHubLightTheme(), models.ThemeTypeLight, false},
|
||||
{models.NewMaterialLightTheme(), models.ThemeTypeLight, false},
|
||||
{models.NewSolarizedLightTheme(), models.ThemeTypeLight, false},
|
||||
{models.NewTokyoNightDayTheme(), models.ThemeTypeLight, false},
|
||||
}
|
||||
|
||||
// 筛选出需要创建的主题
|
||||
var themesToCreate []*models.Theme
|
||||
for _, preset := range presetThemes {
|
||||
if !existingThemes[preset.config.Name] {
|
||||
themesToCreate = append(themesToCreate, &models.Theme{
|
||||
Name: preset.config.Name,
|
||||
Type: preset.themeType,
|
||||
Colors: *preset.config,
|
||||
IsDefault: preset.isDefault,
|
||||
CreatedAt: now,
|
||||
UpdatedAt: now,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
if len(themesToCreate) == 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
// 批量插入主题
|
||||
tx, err := db.Begin()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to begin transaction: %w", err)
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
stmt, err := tx.Prepare(`
|
||||
INSERT INTO themes (name, type, colors, is_default, created_at, updated_at)
|
||||
VALUES (?, ?, ?, ?, ?, ?)
|
||||
`)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to prepare statement: %w", err)
|
||||
}
|
||||
defer stmt.Close()
|
||||
|
||||
for _, theme := range themesToCreate {
|
||||
_, err := stmt.Exec(
|
||||
theme.Name,
|
||||
theme.Type,
|
||||
theme.Colors,
|
||||
theme.IsDefault,
|
||||
theme.CreatedAt,
|
||||
theme.UpdatedAt,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to scan theme: %w", err)
|
||||
return fmt.Errorf("failed to insert theme %s: %w", theme.Name, err)
|
||||
}
|
||||
themes[string(theme.Type)] = theme
|
||||
}
|
||||
|
||||
if err := rows.Err(); err != nil {
|
||||
return nil, fmt.Errorf("failed to iterate themes: %w", err)
|
||||
if err := tx.Commit(); err != nil {
|
||||
return fmt.Errorf("failed to commit transaction: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetThemeByID 根据ID获取主题
|
||||
func (ts *ThemeService) GetThemeByID(id int) (*models.Theme, error) {
|
||||
query := `
|
||||
SELECT id, name, type, colors, is_default, created_at, updated_at
|
||||
FROM themes
|
||||
WHERE id = ?
|
||||
LIMIT 1
|
||||
`
|
||||
|
||||
theme := &models.Theme{}
|
||||
db := ts.getDB()
|
||||
err := db.QueryRow(query, id).Scan(
|
||||
&theme.ID,
|
||||
&theme.Name,
|
||||
&theme.Type,
|
||||
&theme.Colors,
|
||||
&theme.IsDefault,
|
||||
&theme.CreatedAt,
|
||||
&theme.UpdatedAt,
|
||||
)
|
||||
|
||||
if err != nil {
|
||||
if errors.Is(err, sql.ErrNoRows) {
|
||||
return nil, fmt.Errorf("theme not found with id: %d", id)
|
||||
}
|
||||
return nil, fmt.Errorf("failed to get theme by id: %w", err)
|
||||
}
|
||||
|
||||
return themes, nil
|
||||
return theme, nil
|
||||
}
|
||||
|
||||
// GetThemeByType 根据类型获取默认主题
|
||||
@@ -166,37 +212,123 @@ func (ts *ThemeService) GetThemeByType(themeType models.ThemeType) (*models.Them
|
||||
return theme, nil
|
||||
}
|
||||
|
||||
// UpdateThemeColors 更新主题颜色
|
||||
func (ts *ThemeService) UpdateThemeColors(themeType models.ThemeType, colors models.ThemeColorConfig) error {
|
||||
// GetThemesByType 根据类型获取所有主题
|
||||
func (ts *ThemeService) GetThemesByType(themeType models.ThemeType) ([]*models.Theme, error) {
|
||||
query := `
|
||||
UPDATE themes
|
||||
SET colors = ?, updated_at = ?
|
||||
WHERE type = ? AND is_default = 1
|
||||
SELECT id, name, type, colors, is_default, created_at, updated_at
|
||||
FROM themes
|
||||
WHERE type = ?
|
||||
ORDER BY is_default DESC, name ASC
|
||||
`
|
||||
|
||||
db := ts.getDB()
|
||||
_, err := db.Exec(query, colors, time.Now().Format("2006-01-02 15:04:05"), themeType)
|
||||
rows, err := db.Query(query, themeType)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to update theme colors: %w", err)
|
||||
return nil, fmt.Errorf("failed to query themes by type: %w", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var themes []*models.Theme
|
||||
for rows.Next() {
|
||||
theme := &models.Theme{}
|
||||
err := rows.Scan(
|
||||
&theme.ID,
|
||||
&theme.Name,
|
||||
&theme.Type,
|
||||
&theme.Colors,
|
||||
&theme.IsDefault,
|
||||
&theme.CreatedAt,
|
||||
&theme.UpdatedAt,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to scan theme: %w", err)
|
||||
}
|
||||
themes = append(themes, theme)
|
||||
}
|
||||
|
||||
if err := rows.Err(); err != nil {
|
||||
return nil, fmt.Errorf("failed to iterate themes: %w", err)
|
||||
}
|
||||
|
||||
return themes, nil
|
||||
}
|
||||
|
||||
// UpdateTheme 更新主题
|
||||
func (ts *ThemeService) UpdateTheme(id int, colors models.ThemeColorConfig) error {
|
||||
query := `
|
||||
UPDATE themes
|
||||
SET colors = ?, updated_at = ?
|
||||
WHERE id = ?
|
||||
`
|
||||
|
||||
db := ts.getDB()
|
||||
result, err := db.Exec(query, colors, time.Now().Format("2006-01-02 15:04:05"), id)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to update theme: %w", err)
|
||||
}
|
||||
|
||||
rowsAffected, err := result.RowsAffected()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to get rows affected: %w", err)
|
||||
}
|
||||
|
||||
if rowsAffected == 0 {
|
||||
return fmt.Errorf("theme not found with id: %d", id)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// ResetThemeColors 重置主题颜色为默认值
|
||||
func (ts *ThemeService) ResetThemeColors(themeType models.ThemeType) error {
|
||||
var defaultColors models.ThemeColorConfig
|
||||
|
||||
switch themeType {
|
||||
case models.ThemeTypeDark:
|
||||
defaultColors = *models.NewDefaultDarkTheme()
|
||||
case models.ThemeTypeLight:
|
||||
defaultColors = *models.NewDefaultLightTheme()
|
||||
default:
|
||||
return fmt.Errorf("unknown theme type: %s", themeType)
|
||||
// ResetTheme 重置主题为预设配置
|
||||
func (ts *ThemeService) ResetTheme(id int) error {
|
||||
// 先获取主题信息
|
||||
theme, err := ts.GetThemeByID(id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return ts.UpdateThemeColors(themeType, defaultColors)
|
||||
// 根据主题名称获取预设配置
|
||||
var presetConfig *models.ThemeColorConfig
|
||||
switch theme.Name {
|
||||
// 默认主题
|
||||
case "default-dark":
|
||||
presetConfig = models.NewDefaultDarkTheme()
|
||||
case "default-light":
|
||||
presetConfig = models.NewDefaultLightTheme()
|
||||
|
||||
// 深色主题预设
|
||||
case "dracula":
|
||||
presetConfig = models.NewDraculaTheme()
|
||||
case "aura":
|
||||
presetConfig = models.NewAuraTheme()
|
||||
case "github-dark":
|
||||
presetConfig = models.NewGitHubDarkTheme()
|
||||
case "material-dark":
|
||||
presetConfig = models.NewMaterialDarkTheme()
|
||||
case "one-dark":
|
||||
presetConfig = models.NewOneDarkTheme()
|
||||
case "solarized-dark":
|
||||
presetConfig = models.NewSolarizedDarkTheme()
|
||||
case "tokyo-night":
|
||||
presetConfig = models.NewTokyoNightTheme()
|
||||
case "tokyo-night-storm":
|
||||
presetConfig = models.NewTokyoNightStormTheme()
|
||||
|
||||
// 浅色主题预设
|
||||
case "github-light":
|
||||
presetConfig = models.NewGitHubLightTheme()
|
||||
case "material-light":
|
||||
presetConfig = models.NewMaterialLightTheme()
|
||||
case "solarized-light":
|
||||
presetConfig = models.NewSolarizedLightTheme()
|
||||
case "tokyo-night-day":
|
||||
presetConfig = models.NewTokyoNightDayTheme()
|
||||
|
||||
default:
|
||||
return fmt.Errorf("no preset configuration found for theme: %s", theme.Name)
|
||||
}
|
||||
|
||||
return ts.UpdateTheme(id, *presetConfig)
|
||||
}
|
||||
|
||||
// CreateTheme 创建新主题
|
||||
@@ -235,7 +367,7 @@ func (ts *ThemeService) GetAllThemes() ([]*models.Theme, error) {
|
||||
query := `
|
||||
SELECT id, name, type, colors, is_default, created_at, updated_at
|
||||
FROM themes
|
||||
ORDER BY is_default DESC, created_at ASC
|
||||
ORDER BY is_default DESC, type DESC, name ASC
|
||||
`
|
||||
|
||||
db := ts.getDB()
|
||||
|
||||
Reference in New Issue
Block a user