121 lines
2.8 KiB
Go
121 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
|
|
"github.com/BurntSushi/toml"
|
|
"reflect"
|
|
"github.com/caarlos0/env"
|
|
)
|
|
|
|
var (
|
|
settings Settings
|
|
)
|
|
|
|
var LogLevelMap = map[string]int{
|
|
"DEBUG": LevelDebug,
|
|
"INFO": LevelInfo,
|
|
"NOTICE": LevelNotice,
|
|
"WARN": LevelWarn,
|
|
"ERROR": LevelError,
|
|
}
|
|
|
|
type Settings struct {
|
|
Version string
|
|
Debug bool
|
|
Server DNSServerSettings `toml:"server"`
|
|
ResolvConfig ResolvSettings `toml:"resolv"`
|
|
Redis RedisSettings `toml:"redis"`
|
|
Memcache MemcacheSettings `toml:"memcache"`
|
|
Log LogSettings `toml:"log"`
|
|
Cache CacheSettings `toml:"cache"`
|
|
Hosts HostsSettings `toml:"hosts"`
|
|
}
|
|
|
|
type ResolvSettings struct {
|
|
Timeout int
|
|
Interval int
|
|
SetEDNS0 bool
|
|
ServerListFile string `toml:"server-list-file" env:"SERVER_LIST_FILE"`
|
|
ResolvFile string `toml:"resolv-file" env:"RESOLV_FILE"`
|
|
DOHServer string `toml:"dns-over-https" env:"DNS_HTTPS_SERVER"`
|
|
}
|
|
|
|
type DNSServerSettings struct {
|
|
Host string `toml:"host" env:"SERVER_BIND"`
|
|
Port int `toml:"port" env:"SERVER_PORT"`
|
|
}
|
|
|
|
type RedisSettings struct {
|
|
Host string `toml:"host" env:"REDIS_HOST"`
|
|
Port int `toml:"port" env:"REDIS_PORT"`
|
|
DB int `toml:"db" env:"REDIS_DB"`
|
|
Password string `toml:"password" env:"REDIS_PASSWORD"`
|
|
}
|
|
|
|
type MemcacheSettings struct {
|
|
Servers []string `toml:"servers" env:"MEMCACHE_SERVERS"`
|
|
}
|
|
|
|
func (s RedisSettings) Addr() string {
|
|
return s.Host + ":" + strconv.Itoa(s.Port)
|
|
}
|
|
|
|
type LogSettings struct {
|
|
Stdout bool
|
|
File string
|
|
Level string
|
|
}
|
|
|
|
func (ls LogSettings) LogLevel() int {
|
|
l, ok := LogLevelMap[ls.Level]
|
|
if !ok {
|
|
panic("Config error: invalid log level: " + ls.Level)
|
|
}
|
|
return l
|
|
}
|
|
|
|
type CacheSettings struct {
|
|
Backend string `toml:"backend" env:"CACHE_BACKEND"`
|
|
Expire int `toml:"expire" env:"CACHE_EXPIRE"`
|
|
Maxcount int `toml:"maxcount" env:"CACHE_MAX_COUNT"`
|
|
}
|
|
|
|
type HostsSettings struct {
|
|
Enable bool `toml:"enable" env:"HOSTS_ENABLE"`
|
|
HostsFile string `toml:"host-file" env:"HOSTS_FILE"`
|
|
RedisEnable bool `toml:"redis-enable" env:"REDIS_HOSTS_ENABLE"`
|
|
RedisKey string `toml:"redis-key" env:"REDIS_HOSTS_KEY"`
|
|
TTL uint32 `toml:"ttl" env:"HOSTS_TTL"`
|
|
RefreshInterval uint32 `toml:"refresh-interval" env:"HOSTS_REFRESH_INTERVAL"`
|
|
}
|
|
|
|
func init() {
|
|
var configFile string
|
|
|
|
flag.StringVar(&configFile, "c", "/etc/godns.conf", "Look for godns toml-formatting config file in this directory")
|
|
flag.Parse()
|
|
|
|
if _, err := toml.DecodeFile(configFile, &settings); err != nil {
|
|
fmt.Printf("%s is not a valid toml config file\n", configFile)
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
val := reflect.ValueOf(settings)
|
|
|
|
for i := 0; i < val.NumField(); i++ {
|
|
f := val.Field(i)
|
|
|
|
if f.Kind() != reflect.Struct {
|
|
continue
|
|
}
|
|
|
|
v := f.Interface()
|
|
|
|
env.Parse(&v)
|
|
}
|
|
} |