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
431 lines
15 KiB
Go
431 lines
15 KiB
Go
package config
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
"strings"
|
||
|
||
"gopkg.in/yaml.v3"
|
||
)
|
||
|
||
// LaunchServerConfig Launch HTTP 服务配置
|
||
type LaunchServerConfig struct {
|
||
// Port <= 0 时自动分配随机可用端口（推荐）。
|
||
Port int `yaml:"port"`
|
||
}
|
||
|
||
// Config 应用配置
|
||
type Config struct {
|
||
Database DatabaseConfig `yaml:"database"`
|
||
App AppConfig `yaml:"app"`
|
||
Runtime RuntimeConfig `yaml:"runtime"`
|
||
Logging LoggingConfig `yaml:"logging"`
|
||
Browser BrowserConfig `yaml:"browser"`
|
||
LaunchServer LaunchServerConfig `yaml:"launch_server"`
|
||
}
|
||
|
||
// DatabaseConfig 数据库配置
|
||
type DatabaseConfig struct {
|
||
Type string `yaml:"type"`
|
||
SQLite SQLiteConfig `yaml:"sqlite"`
|
||
}
|
||
|
||
// SQLiteConfig SQLite 配置
|
||
type SQLiteConfig struct {
|
||
Path string `yaml:"path"`
|
||
}
|
||
|
||
// AppConfig 应用配置
|
||
type AppConfig struct {
|
||
Name string `yaml:"name"`
|
||
Window WindowConfig `yaml:"window"`
|
||
MaxProfileLimit int `yaml:"max_profile_limit"`
|
||
UsedCDKeys []string `yaml:"used_cd_keys"`
|
||
}
|
||
|
||
// WindowConfig 窗口配置
|
||
type WindowConfig struct {
|
||
Width int `yaml:"width"`
|
||
Height int `yaml:"height"`
|
||
MinWidth int `yaml:"min_width"`
|
||
MinHeight int `yaml:"min_height"`
|
||
}
|
||
|
||
// RuntimeConfig 运行时配置
|
||
type RuntimeConfig struct {
|
||
MaxMemoryMB int `yaml:"max_memory_mb"` // 最大内存限制（MB）
|
||
GCPercent int `yaml:"gc_percent"` // GC 触发百分比
|
||
}
|
||
|
||
type BrowserBookmark struct {
|
||
Name string `yaml:"name" json:"name"`
|
||
URL string `yaml:"url" json:"url"`
|
||
}
|
||
|
||
type BrowserConfig struct {
|
||
UserDataRoot string `yaml:"user_data_root"`
|
||
DefaultFingerprintArgs []string `yaml:"default_fingerprint_args"`
|
||
DefaultLaunchArgs []string `yaml:"default_launch_args"`
|
||
DefaultProxy string `yaml:"default_proxy"`
|
||
DefaultBookmarks []BrowserBookmark `yaml:"default_bookmarks,omitempty"`
|
||
Cores []BrowserCore `yaml:"cores,omitempty"`
|
||
Proxies []BrowserProxy `yaml:"proxies,omitempty"`
|
||
Profiles []BrowserProfileConfig `yaml:"profiles,omitempty"`
|
||
// 废弃字段，保留用于迁移
|
||
ChromeBinaryPath string `yaml:"chrome_binary_path,omitempty"`
|
||
ClashBinaryPath string `yaml:"clash_binary_path,omitempty"`
|
||
XrayBinaryPath string `yaml:"xray_binary_path,omitempty"`
|
||
SingBoxBinaryPath string `yaml:"singbox_binary_path,omitempty"`
|
||
CoreRoot string `yaml:"core_root,omitempty"`
|
||
DefaultCoreId string `yaml:"default_core_id,omitempty"`
|
||
DefaultConnectorType string `yaml:"default_connector_type,omitempty"`
|
||
Environments []BrowserEnvironment `yaml:"environments,omitempty"`
|
||
}
|
||
|
||
type BrowserCore struct {
|
||
CoreId string `yaml:"core_id" json:"coreId"`
|
||
CoreName string `yaml:"core_name" json:"coreName"`
|
||
CorePath string `yaml:"core_path" json:"corePath"`
|
||
IsDefault bool `yaml:"is_default" json:"isDefault"`
|
||
}
|
||
|
||
type BrowserProxy struct {
|
||
ProxyId string `yaml:"proxy_id" json:"proxyId"`
|
||
ProxyName string `yaml:"proxy_name" json:"proxyName"`
|
||
ProxyConfig string `yaml:"proxy_config" json:"proxyConfig"`
|
||
DnsServers string `yaml:"dns_servers,omitempty" json:"dnsServers,omitempty"`
|
||
GroupName string `yaml:"group_name,omitempty" json:"groupName,omitempty"`
|
||
SortOrder int `yaml:"sort_order,omitempty" json:"sortOrder,omitempty"`
|
||
SourceID string `yaml:"source_id,omitempty" json:"sourceId,omitempty"`
|
||
SourceURL string `yaml:"source_url,omitempty" json:"sourceUrl,omitempty"`
|
||
// URL 导入时的名称前缀，用于后续自动刷新时重建同名策略
|
||
SourceNamePrefix string `yaml:"source_name_prefix,omitempty" json:"sourceNamePrefix,omitempty"`
|
||
// URL 导入自动刷新开关与间隔（分钟）
|
||
SourceAutoRefresh bool `yaml:"source_auto_refresh,omitempty" json:"sourceAutoRefresh,omitempty"`
|
||
SourceRefreshIntervalM int `yaml:"source_refresh_interval_m,omitempty" json:"sourceRefreshIntervalM,omitempty"`
|
||
SourceLastRefreshAt string `yaml:"source_last_refresh_at,omitempty" json:"sourceLastRefreshAt,omitempty"`
|
||
// 测速结果（运行时字段，不写入 yaml）
|
||
LastLatencyMs int64 `yaml:"-" json:"lastLatencyMs"`
|
||
LastTestOk bool `yaml:"-" json:"lastTestOk"`
|
||
LastTestedAt string `yaml:"-" json:"lastTestedAt"`
|
||
// IP 健康检测原始结果（运行时字段，不写入 yaml）
|
||
LastIPHealthJSON string `yaml:"-" json:"lastIPHealthJson,omitempty"`
|
||
}
|
||
|
||
type BrowserEnvironment struct {
|
||
CoreId string `yaml:"core_id" json:"coreId"`
|
||
CoreName string `yaml:"core_name" json:"coreName"`
|
||
CorePath string `yaml:"core_path" json:"corePath"`
|
||
ProxyConfig string `yaml:"proxy_config" json:"proxyConfig"`
|
||
ConnectorType string `yaml:"connector_type" json:"connectorType"`
|
||
IsDefault bool `yaml:"is_default" json:"isDefault"`
|
||
}
|
||
|
||
type BrowserProfileConfig struct {
|
||
ProfileId string `yaml:"profile_id" json:"profileId"`
|
||
ProfileName string `yaml:"profile_name" json:"profileName"`
|
||
UserDataDir string `yaml:"user_data_dir" json:"userDataDir"`
|
||
CoreId string `yaml:"core_id" json:"coreId"`
|
||
FingerprintArgs []string `yaml:"fingerprint_args" json:"fingerprintArgs"`
|
||
ProxyId string `yaml:"proxy_id" json:"proxyId"`
|
||
ProxyConfig string `yaml:"proxy_config" json:"proxyConfig"`
|
||
LaunchArgs []string `yaml:"launch_args" json:"launchArgs"`
|
||
Tags []string `yaml:"tags" json:"tags"`
|
||
Keywords []string `yaml:"keywords,omitempty" json:"keywords,omitempty"`
|
||
CreatedAt string `yaml:"created_at" json:"createdAt"`
|
||
UpdatedAt string `yaml:"updated_at" json:"updatedAt"`
|
||
}
|
||
|
||
// LoggingConfig 日志配置
|
||
type LoggingConfig struct {
|
||
Level string `yaml:"level"`
|
||
FileEnabled bool `yaml:"file_enabled"`
|
||
FilePath string `yaml:"file_path"`
|
||
Format string `yaml:"format"` // "text" or "json"
|
||
|
||
// 性能配置
|
||
BufferSize int `yaml:"buffer_size"` // 缓冲区大小（KB）
|
||
AsyncQueueSize int `yaml:"async_queue_size"` // 异步队列大小
|
||
FlushIntervalMs int `yaml:"flush_interval_ms"` // 刷新间隔（毫秒）
|
||
|
||
// 分片配置
|
||
Rotation RotationConfig `yaml:"rotation"`
|
||
|
||
// 方法拦截配置
|
||
Interceptor InterceptorConfig `yaml:"interceptor"`
|
||
}
|
||
|
||
// RotationConfig 日志分片配置
|
||
type RotationConfig struct {
|
||
Enabled bool `yaml:"enabled"`
|
||
MaxSizeMB int `yaml:"max_size_mb"` // 单文件最大大小（MB）
|
||
MaxAge int `yaml:"max_age"` // 保留天数
|
||
MaxBackups int `yaml:"max_backups"` // 保留文件数
|
||
TimeInterval string `yaml:"time_interval"` // 时间间隔: "daily", "hourly"
|
||
}
|
||
|
||
// InterceptorConfig 方法拦截器配置
|
||
type InterceptorConfig struct {
|
||
Enabled bool `yaml:"enabled"`
|
||
LogParameters bool `yaml:"log_parameters"` // 是否记录参数
|
||
LogResults bool `yaml:"log_results"` // 是否记录返回值
|
||
SensitiveFields []string `yaml:"sensitive_fields"` // 敏感字段（脱敏）
|
||
}
|
||
|
||
// Load 加载配置文件
|
||
func Load(configPath string) (*Config, error) {
|
||
data, err := os.ReadFile(configPath)
|
||
if err != nil {
|
||
if os.IsNotExist(err) {
|
||
return DefaultConfig(), nil
|
||
}
|
||
return nil, fmt.Errorf("读取配置文件失败: %w", err)
|
||
}
|
||
|
||
var config Config
|
||
if err := yaml.Unmarshal(data, &config); err != nil {
|
||
return nil, fmt.Errorf("解析配置文件失败: %w", err)
|
||
}
|
||
|
||
normalizeConfig(&config)
|
||
|
||
return &config, nil
|
||
}
|
||
|
||
// normalizeConfig 对历史配置进行字段补齐，不覆盖用户已配置值。
|
||
func normalizeConfig(config *Config) {
|
||
defaultConfig := DefaultConfig()
|
||
|
||
if strings.TrimSpace(config.Database.Type) == "" {
|
||
config.Database.Type = defaultConfig.Database.Type
|
||
}
|
||
if strings.TrimSpace(config.Database.SQLite.Path) == "" {
|
||
config.Database.SQLite.Path = defaultConfig.Database.SQLite.Path
|
||
}
|
||
|
||
if strings.TrimSpace(config.App.Name) == "" {
|
||
config.App.Name = defaultConfig.App.Name
|
||
}
|
||
if config.App.Window.Width <= 0 {
|
||
config.App.Window.Width = defaultConfig.App.Window.Width
|
||
}
|
||
if config.App.Window.Height <= 0 {
|
||
config.App.Window.Height = defaultConfig.App.Window.Height
|
||
}
|
||
if config.App.Window.MinWidth <= 0 {
|
||
config.App.Window.MinWidth = defaultConfig.App.Window.MinWidth
|
||
}
|
||
if config.App.Window.MinHeight <= 0 {
|
||
config.App.Window.MinHeight = defaultConfig.App.Window.MinHeight
|
||
}
|
||
if config.App.UsedCDKeys == nil {
|
||
config.App.UsedCDKeys = []string{}
|
||
}
|
||
|
||
// 兼容老版本配置: 如果之前没有 max_profile_limit，它会被解析成 0，
|
||
// 若用户在 0 状态下兑换了额度（比如 0+3=3），基础的 3 额度会被覆盖。
|
||
// 这里通过统计兑换记录重新保底验证它的额度即可修复。
|
||
expectedLimit := defaultConfig.App.MaxProfileLimit
|
||
for _, k := range config.App.UsedCDKeys {
|
||
if k == "GITHUB_STAR_REWARD" {
|
||
expectedLimit += 3
|
||
} else {
|
||
expectedLimit += 3
|
||
}
|
||
}
|
||
if config.App.MaxProfileLimit < expectedLimit {
|
||
config.App.MaxProfileLimit = expectedLimit
|
||
}
|
||
|
||
if config.Runtime.MaxMemoryMB <= 0 {
|
||
config.Runtime.MaxMemoryMB = defaultConfig.Runtime.MaxMemoryMB
|
||
}
|
||
if config.Runtime.GCPercent <= 0 {
|
||
config.Runtime.GCPercent = defaultConfig.Runtime.GCPercent
|
||
}
|
||
|
||
if strings.TrimSpace(config.Logging.Level) == "" {
|
||
config.Logging.Level = defaultConfig.Logging.Level
|
||
}
|
||
if isLegacyDefaultLogPath(config.Logging.FilePath) || strings.TrimSpace(config.Logging.FilePath) == "" {
|
||
config.Logging.FilePath = defaultConfig.Logging.FilePath
|
||
}
|
||
if strings.TrimSpace(config.Logging.Format) == "" {
|
||
config.Logging.Format = defaultConfig.Logging.Format
|
||
}
|
||
if config.Logging.BufferSize <= 0 {
|
||
config.Logging.BufferSize = defaultConfig.Logging.BufferSize
|
||
}
|
||
if config.Logging.AsyncQueueSize <= 0 {
|
||
config.Logging.AsyncQueueSize = defaultConfig.Logging.AsyncQueueSize
|
||
}
|
||
if config.Logging.FlushIntervalMs <= 0 {
|
||
config.Logging.FlushIntervalMs = defaultConfig.Logging.FlushIntervalMs
|
||
}
|
||
if config.Logging.Rotation.MaxSizeMB <= 0 {
|
||
config.Logging.Rotation.MaxSizeMB = defaultConfig.Logging.Rotation.MaxSizeMB
|
||
}
|
||
if config.Logging.Rotation.MaxAge <= 0 {
|
||
config.Logging.Rotation.MaxAge = defaultConfig.Logging.Rotation.MaxAge
|
||
}
|
||
if config.Logging.Rotation.MaxBackups <= 0 {
|
||
config.Logging.Rotation.MaxBackups = defaultConfig.Logging.Rotation.MaxBackups
|
||
}
|
||
if strings.TrimSpace(config.Logging.Rotation.TimeInterval) == "" {
|
||
config.Logging.Rotation.TimeInterval = defaultConfig.Logging.Rotation.TimeInterval
|
||
}
|
||
|
||
interceptorAllZero := !config.Logging.Interceptor.Enabled &&
|
||
!config.Logging.Interceptor.LogParameters &&
|
||
!config.Logging.Interceptor.LogResults &&
|
||
config.Logging.Interceptor.SensitiveFields == nil
|
||
if interceptorAllZero {
|
||
config.Logging.Interceptor = cloneInterceptorConfig(defaultConfig.Logging.Interceptor)
|
||
} else if config.Logging.Interceptor.SensitiveFields == nil {
|
||
config.Logging.Interceptor.SensitiveFields = append([]string{}, defaultConfig.Logging.Interceptor.SensitiveFields...)
|
||
}
|
||
|
||
if strings.TrimSpace(config.Browser.UserDataRoot) == "" {
|
||
config.Browser.UserDataRoot = defaultConfig.Browser.UserDataRoot
|
||
}
|
||
if len(config.Browser.DefaultFingerprintArgs) == 0 {
|
||
config.Browser.DefaultFingerprintArgs = append([]string{}, defaultConfig.Browser.DefaultFingerprintArgs...)
|
||
}
|
||
if len(config.Browser.DefaultLaunchArgs) == 0 {
|
||
config.Browser.DefaultLaunchArgs = append([]string{}, defaultConfig.Browser.DefaultLaunchArgs...)
|
||
}
|
||
if config.Browser.DefaultBookmarks == nil {
|
||
config.Browser.DefaultBookmarks = []BrowserBookmark{}
|
||
}
|
||
if config.Browser.Cores == nil {
|
||
config.Browser.Cores = []BrowserCore{}
|
||
}
|
||
if config.Browser.Proxies == nil {
|
||
config.Browser.Proxies = []BrowserProxy{}
|
||
}
|
||
if config.Browser.Profiles == nil {
|
||
config.Browser.Profiles = []BrowserProfileConfig{}
|
||
}
|
||
|
||
if config.LaunchServer.Port < 0 {
|
||
config.LaunchServer.Port = defaultConfig.LaunchServer.Port
|
||
}
|
||
}
|
||
|
||
func cloneInterceptorConfig(src InterceptorConfig) InterceptorConfig {
|
||
dst := src
|
||
dst.SensitiveFields = append([]string{}, src.SensitiveFields...)
|
||
return dst
|
||
}
|
||
|
||
func isLegacyDefaultLogPath(path string) bool {
|
||
return strings.EqualFold(filepath.ToSlash(strings.TrimSpace(path)), "logs/app.log")
|
||
}
|
||
|
||
// DefaultConfig 返回默认配置
|
||
func DefaultConfig() *Config {
|
||
return &Config{
|
||
Database: DatabaseConfig{
|
||
Type: "sqlite",
|
||
SQLite: SQLiteConfig{
|
||
Path: "data/app.db",
|
||
},
|
||
},
|
||
App: AppConfig{
|
||
Name: "Ant Browser",
|
||
Window: WindowConfig{
|
||
Width: 1750,
|
||
Height: 1000,
|
||
MinWidth: 1200,
|
||
MinHeight: 700,
|
||
},
|
||
MaxProfileLimit: 3,
|
||
UsedCDKeys: []string{},
|
||
},
|
||
Runtime: RuntimeConfig{
|
||
MaxMemoryMB: 1024, // 默认 1GB
|
||
GCPercent: 100, // 默认 100%
|
||
},
|
||
Browser: BrowserConfig{
|
||
UserDataRoot: "data",
|
||
DefaultFingerprintArgs: []string{"--fingerprint-brand=Chrome", "--fingerprint-platform=windows"},
|
||
DefaultLaunchArgs: []string{"--disable-sync", "--no-first-run"},
|
||
DefaultProxy: "",
|
||
},
|
||
Logging: LoggingConfig{
|
||
Level: "info",
|
||
FileEnabled: false,
|
||
FilePath: "data/logs/app.log",
|
||
Format: "text",
|
||
BufferSize: 4, // 4KB
|
||
AsyncQueueSize: 1000,
|
||
FlushIntervalMs: 1000, // 1秒
|
||
Rotation: RotationConfig{
|
||
Enabled: false,
|
||
MaxSizeMB: 100,
|
||
MaxAge: 7,
|
||
MaxBackups: 5,
|
||
TimeInterval: "daily",
|
||
},
|
||
Interceptor: InterceptorConfig{
|
||
Enabled: true,
|
||
LogParameters: true,
|
||
LogResults: true,
|
||
SensitiveFields: []string{"password", "token", "secret"},
|
||
},
|
||
},
|
||
LaunchServer: LaunchServerConfig{
|
||
Port: 0,
|
||
},
|
||
}
|
||
}
|
||
|
||
// Save 保存配置到文件
|
||
func (c *Config) Save(configPath string) error {
|
||
data, err := yaml.Marshal(c)
|
||
if err != nil {
|
||
return fmt.Errorf("序列化配置失败: %w", err)
|
||
}
|
||
|
||
if err := os.WriteFile(configPath, data, 0644); err != nil {
|
||
return fmt.Errorf("写入配置文件失败: %w", err)
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// ProxyStore 代理数据文件结构
|
||
type ProxyStore struct {
|
||
Proxies []BrowserProxy `yaml:"proxies"`
|
||
}
|
||
|
||
// LoadProxies 从独立文件加载代理列表
|
||
func LoadProxies(path string) ([]BrowserProxy, error) {
|
||
data, err := os.ReadFile(path)
|
||
if err != nil {
|
||
if os.IsNotExist(err) {
|
||
return nil, nil
|
||
}
|
||
return nil, fmt.Errorf("读取代理文件失败: %w", err)
|
||
}
|
||
var store ProxyStore
|
||
if err := yaml.Unmarshal(data, &store); err != nil {
|
||
return nil, fmt.Errorf("解析代理文件失败: %w", err)
|
||
}
|
||
return store.Proxies, nil
|
||
}
|
||
|
||
// SaveProxies 将代理列表保存到独立文件
|
||
func SaveProxies(path string, proxies []BrowserProxy) error {
|
||
store := ProxyStore{Proxies: proxies}
|
||
data, err := yaml.Marshal(store)
|
||
if err != nil {
|
||
return fmt.Errorf("序列化代理数据失败: %w", err)
|
||
}
|
||
if err := os.WriteFile(path, data, 0644); err != nil {
|
||
return fmt.Errorf("写入代理文件失败: %w", err)
|
||
}
|
||
return nil
|
||
}
|