dns-server/config/config.go

package config

import (
	"encoding/json"
	"io/ioutil"
)

// DomainSpecificDNS 域名特定DNS服务器配置
// 格式：{"domainMatch": ["dns1", "dns2"]}
// domainMatch: 域名匹配字符串，当域名中包含该字符串时使用对应的DNS服务器
// dns1, dns2: 用于解析匹配域名的DNS服务器列表

type DomainSpecificDNS map[string][]string

// DNSConfig DNS配置
type DNSConfig struct {
	Port             int              `json:"port"`
	UpstreamDNS      []string         `json:"upstreamDNS"`
	DNSSECUpstreamDNS []string        `json:"dnssecUpstreamDNS"` // 用于DNSSEC查询的专用服务器
	Timeout          int              `json:"timeout"`
	StatsFile        string           `json:"statsFile"`    // 统计数据持久化文件
	SaveInterval     int              `json:"saveInterval"` // 数据保存间隔（秒）
	CacheTTL         int              `json:"cacheTTL"`     // DNS缓存过期时间（分钟）
	EnableDNSSEC     bool             `json:"enableDNSSEC"` // 是否启用DNSSEC支持
	QueryMode        string           `json:"queryMode"`    // 查询模式："loadbalance"（负载均衡）、"parallel"（并行请求）、"fastest-ip"（最快的IP地址）
	DomainSpecificDNS DomainSpecificDNS `json:"domainSpecificDNS"` // 域名特定DNS服务器配置
	NoDNSSECDomains  []string         `json:"noDNSSECDomains"` // 不验证DNSSEC的域名模式列表
	PrefixDomain     []string         `json:"prefixDomain"` // 搜索域名前缀列表，对应/etc/resolv.conf中的search domain
}

// HTTPConfig HTTP控制台配置
type HTTPConfig struct {
	Port      int    `json:"port"`
	Host      string `json:"host"`
	EnableAPI bool   `json:"enableAPI"`
	Username  string `json:"username"` // 登录用户名
	Password  string `json:"password"` // 登录密码
}

// BlacklistEntry 黑名单条目
type BlacklistEntry struct {
	Name           string `json:"name"`
	URL            string `json:"url"`
	Enabled        bool   `json:"enabled"`
	RuleCount      int    `json:"ruleCount,omitempty"`
	LastUpdateTime string `json:"lastUpdateTime,omitempty"`
}

// ShieldConfig 屏蔽规则配置
type ShieldConfig struct {
	LocalRulesFile      string           `json:"localRulesFile"`
	Blacklists          []BlacklistEntry `json:"blacklists"`
	UpdateInterval      int              `json:"updateInterval"`
	HostsFile           string           `json:"hostsFile"`
	BlockMethod         string           `json:"blockMethod"`         // 屏蔽方法: "NXDOMAIN", "refused", "emptyIP", "customIP"
	CustomBlockIP       string           `json:"customBlockIP"`       // 自定义屏蔽IP，当BlockMethod为"customIP"时使用
	StatsFile           string           `json:"statsFile"`           // 计数数据持久化文件
	StatsSaveInterval   int              `json:"statsSaveInterval"`   // 计数数据保存间隔（秒）
	RemoteRulesCacheDir string           `json:"remoteRulesCacheDir"` // 远程规则缓存目录
}

// LogConfig 日志配置
type LogConfig struct {
	File       string `json:"file"`
	Level      string `json:"level"`
	MaxSize    int    `json:"maxSize"`
	MaxBackups int    `json:"maxBackups"`
	MaxAge     int    `json:"maxAge"`
}

// Config 整体配置
type Config struct {
	DNS    DNSConfig    `json:"dns"`
	HTTP   HTTPConfig   `json:"http"`
	Shield ShieldConfig `json:"shield"`
	Log    LogConfig    `json:"log"`
}

// LoadConfig 加载配置文件
func LoadConfig(path string) (*Config, error) {
	data, err := ioutil.ReadFile(path)
	if err != nil {
		return nil, err
	}

	var config Config
	err = json.Unmarshal(data, &config)
	if err != nil {
		return nil, err
	}

	// 设置默认值
	if config.DNS.Port == 0 {
		config.DNS.Port = 53
	}
	if len(config.DNS.UpstreamDNS) == 0 {
		config.DNS.UpstreamDNS = []string{"223.5.5.5:53", "223.6.6.6:53"}
	}
	if config.DNS.StatsFile == "" {
		config.DNS.StatsFile = "./data/stats.json" // 默认统计数据文件路径
	}
	if config.DNS.SaveInterval == 0 {
		config.DNS.SaveInterval = 300 // 默认5分钟保存一次
	}
	// 默认DNS缓存TTL为30分钟
	if config.DNS.CacheTTL == 0 {
		config.DNS.CacheTTL = 30 // 默认30分钟
	}
	// DNSSEC默认配置
	config.DNS.EnableDNSSEC = true // 默认启用DNSSEC支持
	// DNSSEC专用服务器默认配置
	if len(config.DNS.DNSSECUpstreamDNS) == 0 {
		config.DNS.DNSSECUpstreamDNS = []string{"8.8.8.8:53", "1.1.1.1:53"}
	}
	// 查询模式默认配置
	if config.DNS.QueryMode == "" {
		config.DNS.QueryMode = "parallel" // 默认使用并行请求模式
	}
	// 域名特定DNS服务器配置默认值
	if config.DNS.DomainSpecificDNS == nil {
		config.DNS.DomainSpecificDNS = make(DomainSpecificDNS) // 默认为空映射
	}
	// PrefixDomain默认值处理
	if config.DNS.PrefixDomain == nil {
		config.DNS.PrefixDomain = []string{} // 默认为空切片
	}
	if config.HTTP.Port == 0 {
		config.HTTP.Port = 8080
	}
	if config.HTTP.Host == "" {
		config.HTTP.Host = "0.0.0.0"
	}
	// 默认用户名和密码，如果未配置则使用admin/admin
	if config.HTTP.Username == "" {
		config.HTTP.Username = "admin"
	}
	if config.HTTP.Password == "" {
		config.HTTP.Password = "admin"
	}
	if config.Shield.UpdateInterval == 0 {
		config.Shield.UpdateInterval = 3600
	}
	if config.Shield.BlockMethod == "" {
		config.Shield.BlockMethod = "NXDOMAIN" // 默认屏蔽方法为NXDOMAIN
	}
	if config.Shield.StatsFile == "" {
		config.Shield.StatsFile = "./data/shield_stats.json" // 默认Shield统计数据文件路径
	}
	if config.Shield.StatsSaveInterval == 0 {
		config.Shield.StatsSaveInterval = 300 // 默认5分钟保存一次
	}
	if config.Shield.RemoteRulesCacheDir == "" {
		config.Shield.RemoteRulesCacheDir = "./data/remote_rules" // 默认远程规则缓存目录
	}

	// 如果黑名单列表为空，添加一些默认的黑名单
	if len(config.Shield.Blacklists) == 0 {
		config.Shield.Blacklists = []BlacklistEntry{
			{Name: "AdGuard DNS filter", URL: "https://gitea.amazehome.xyz/AMAZEHOME/hosts-and-filters/raw/branch/main/filter.txt", Enabled: true},
			{Name: "Adaway Default Blocklist", URL: "https://gitea.amazehome.xyz/AMAZEHOME/hosts-and-Filters/raw/branch/main/hosts/adaway.txt", Enabled: true},
			{Name: "CHN-anti-AD", URL: "https://gitea.amazehome.xyz/AMAZEHOME/hosts-and-Filters/raw/branch/main/list/easylist.txt", Enabled: true},
			{Name: "My GitHub Rules", URL: "https://gitea.amazehome.xyz/AMAZEHOME/hosts-and-filters/raw/branch/main/rules/costomize.txt", Enabled: true},
		}
	}
	if config.Log.Level == "" {
		config.Log.Level = "info"
	}

	return &config, nil
}