mirror of
https://github.com/black-ant/Ant-Browser.git
synced 2026-07-14 18:48:55 +08:00
Ant Browser Release Bot
6f58a6c19a
channel: master version: 1.0.0 source-ref: master published-at-utc: 2026-03-13T15:19:28Z
97 lines
2.5 KiB
Go
97 lines
2.5 KiB
Go
package browser
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
// CoreDAO 内核配置持久化接口
|
|
type CoreDAO interface {
|
|
List() ([]Core, error)
|
|
Upsert(core Core) error
|
|
Delete(coreId string) error
|
|
SetDefault(coreId string) error
|
|
}
|
|
|
|
// SQLiteCoreDAO 基于 SQLite 的 CoreDAO 实现
|
|
type SQLiteCoreDAO struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
// NewSQLiteCoreDAO 创建 SQLiteCoreDAO
|
|
func NewSQLiteCoreDAO(db *sql.DB) *SQLiteCoreDAO {
|
|
return &SQLiteCoreDAO{db: db}
|
|
}
|
|
|
|
// List 查询所有内核，按 sort_order 升序
|
|
func (d *SQLiteCoreDAO) List() ([]Core, error) {
|
|
rows, err := d.db.Query(`
|
|
SELECT core_id, core_name, core_path, is_default
|
|
FROM browser_cores ORDER BY sort_order ASC, created_at ASC`)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("查询内核列表失败: %w", err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
var list []Core
|
|
for rows.Next() {
|
|
var c Core
|
|
var isDefault int
|
|
if err := rows.Scan(&c.CoreId, &c.CoreName, &c.CorePath, &isDefault); err != nil {
|
|
return nil, fmt.Errorf("读取内核行失败: %w", err)
|
|
}
|
|
c.IsDefault = isDefault == 1
|
|
list = append(list, c)
|
|
}
|
|
return list, rows.Err()
|
|
}
|
|
|
|
// Upsert 新增或更新内核配置
|
|
func (d *SQLiteCoreDAO) Upsert(core Core) error {
|
|
now := time.Now().Format(time.RFC3339)
|
|
isDefault := 0
|
|
if core.IsDefault {
|
|
isDefault = 1
|
|
}
|
|
_, err := d.db.Exec(`
|
|
INSERT INTO browser_cores (core_id, core_name, core_path, is_default, created_at)
|
|
VALUES (?, ?, ?, ?, ?)
|
|
ON CONFLICT(core_id) DO UPDATE SET
|
|
core_name = excluded.core_name,
|
|
core_path = excluded.core_path,
|
|
is_default = excluded.is_default`,
|
|
core.CoreId, core.CoreName, core.CorePath, isDefault, now,
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("保存内核配置失败: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Delete 删除内核配置
|
|
func (d *SQLiteCoreDAO) Delete(coreId string) error {
|
|
_, err := d.db.Exec(`DELETE FROM browser_cores WHERE core_id = ?`, coreId)
|
|
if err != nil {
|
|
return fmt.Errorf("删除内核配置失败: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// SetDefault 设置默认内核（先清除所有默认标记，再设置指定内核）
|
|
func (d *SQLiteCoreDAO) SetDefault(coreId string) error {
|
|
tx, err := d.db.Begin()
|
|
if err != nil {
|
|
return fmt.Errorf("开启事务失败: %w", err)
|
|
}
|
|
defer tx.Rollback()
|
|
|
|
if _, err := tx.Exec(`UPDATE browser_cores SET is_default = 0`); err != nil {
|
|
return fmt.Errorf("清除默认内核失败: %w", err)
|
|
}
|
|
if _, err := tx.Exec(`UPDATE browser_cores SET is_default = 1 WHERE core_id = ?`, coreId); err != nil {
|
|
return fmt.Errorf("设置默认内核失败: %w", err)
|
|
}
|
|
return tx.Commit()
|
|
}
|