164 lines
5.9 KiB
Go
164 lines
5.9 KiB
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查询的专用服务器
|
||
SaveInterval int `json:"saveInterval"` // 数据保存间隔（秒）
|
||
CacheTTL int `json:"cacheTTL"` // DNS缓存过期时间（分钟）
|
||
EnableDNSSEC bool `json:"enableDNSSEC"` // 是否启用DNSSEC支持
|
||
QueryMode string `json:"queryMode"` // 查询模式："loadbalance"（负载均衡）、"parallel"（并行请求）、"fastest-ip"（最快的IP地址）
|
||
QueryTimeout int `json:"queryTimeout"` // 查询超时时间（毫秒）
|
||
EnableFastReturn bool `json:"enableFastReturn"` // 是否启用快速返回机制
|
||
DomainSpecificDNS DomainSpecificDNS `json:"domainSpecificDNS"` // 域名特定DNS服务器配置
|
||
NoDNSSECDomains []string `json:"noDNSSECDomains"` // 不验证DNSSEC的域名模式列表
|
||
EnableIPv6 bool `json:"enableIPv6"` // 是否启用IPv6解析（AAAA记录）
|
||
}
|
||
|
||
// 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 {
|
||
Blacklists []BlacklistEntry `json:"blacklists"`
|
||
UpdateInterval int `json:"updateInterval"`
|
||
BlockMethod string `json:"blockMethod"` // 屏蔽方法: "NXDOMAIN", "refused", "emptyIP", "customIP"
|
||
CustomBlockIP string `json:"customBlockIP"` // 自定义屏蔽IP，当BlockMethod为"customIP"时使用
|
||
StatsSaveInterval int `json:"statsSaveInterval"` // 计数数据保存间隔（秒）
|
||
}
|
||
|
||
// LogConfig 日志配置
|
||
type LogConfig struct {
|
||
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.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支持
|
||
// IPv6默认配置
|
||
config.DNS.EnableIPv6 = true // 默认启用IPv6解析
|
||
// 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" // 默认使用并行请求模式
|
||
}
|
||
// 查询超时默认配置（毫秒）
|
||
if config.DNS.QueryTimeout == 0 {
|
||
config.DNS.QueryTimeout = 500 // 默认超时时间为500ms
|
||
}
|
||
// 快速返回机制默认配置
|
||
if config.DNS.EnableFastReturn == false {
|
||
config.DNS.EnableFastReturn = true // 默认启用快速返回机制
|
||
}
|
||
// 域名特定DNS服务器配置默认值
|
||
if config.DNS.DomainSpecificDNS == nil {
|
||
config.DNS.DomainSpecificDNS = make(DomainSpecificDNS) // 默认为空映射
|
||
}
|
||
|
||
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.StatsSaveInterval == 0 {
|
||
config.Shield.StatsSaveInterval = 300 // 默认5分钟保存一次
|
||
}
|
||
|
||
// 如果黑名单列表为空，添加一些默认的黑名单
|
||
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
|
||
}
|