126 lines
4.2 KiB
Go
126 lines
4.2 KiB
Go
package config
|
||
|
||
import (
|
||
"encoding/json"
|
||
"io/ioutil"
|
||
)
|
||
|
||
// DNSConfig DNS配置
|
||
type DNSConfig struct {
|
||
Port int `json:"port"`
|
||
UpstreamDNS []string `json:"upstreamDNS"`
|
||
Timeout int `json:"timeout"`
|
||
StatsFile string `json:"statsFile"` // 统计数据持久化文件
|
||
SaveInterval int `json:"saveInterval"` // 数据保存间隔（秒）
|
||
}
|
||
|
||
// HTTPConfig HTTP控制台配置
|
||
type HTTPConfig struct {
|
||
Port int `json:"port"`
|
||
Host string `json:"host"`
|
||
EnableAPI bool `json:"enableAPI"`
|
||
}
|
||
|
||
// 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分钟保存一次
|
||
}
|
||
if config.HTTP.Port == 0 {
|
||
config.HTTP.Port = 8080
|
||
}
|
||
if config.HTTP.Host == "" {
|
||
config.HTTP.Host = "0.0.0.0"
|
||
}
|
||
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
|
||
}
|