mirror of
https://github.com/nadoo/glider.git
synced 2025-02-23 17:35:40 +08:00
76 lines
2.0 KiB
Go
76 lines
2.0 KiB
Go
package rule
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/nadoo/conflag"
|
|
|
|
"github.com/nadoo/glider/strategy"
|
|
)
|
|
|
|
// Config , every rule dialer points to a rule file
|
|
type Config struct {
|
|
name string
|
|
|
|
Forward []string
|
|
StrategyConfig strategy.Config
|
|
|
|
DNSServers []string
|
|
IPSet string
|
|
|
|
Domain []string
|
|
IP []string
|
|
CIDR []string
|
|
}
|
|
|
|
// NewConfFromFile .
|
|
func NewConfFromFile(ruleFile string) (*Config, error) {
|
|
p := &Config{name: ruleFile}
|
|
|
|
f := conflag.NewFromFile("rule", ruleFile)
|
|
f.StringSliceUniqVar(&p.Forward, "forward", nil, "forward url, format: SCHEME://[USER|METHOD:PASSWORD@][HOST]:PORT?PARAMS[,SCHEME://[USER|METHOD:PASSWORD@][HOST]:PORT?PARAMS]")
|
|
f.StringVar(&p.StrategyConfig.Strategy, "strategy", "rr", "forward strategy, default: rr")
|
|
f.StringVar(&p.StrategyConfig.CheckWebSite, "checkwebsite", "www.apple.com", "proxy check HTTP(NOT HTTPS) website address, format: HOST[:PORT], default port: 80")
|
|
// TODO: change to checkinterval
|
|
f.IntVar(&p.StrategyConfig.CheckInterval, "checkduration", 30, "proxy check interval(seconds)")
|
|
f.StringVar(&p.StrategyConfig.IntFace, "interface", "", "source ip or source interface")
|
|
|
|
f.StringSliceUniqVar(&p.DNSServers, "dnsserver", nil, "remote dns server")
|
|
f.StringVar(&p.IPSet, "ipset", "", "ipset name")
|
|
|
|
f.StringSliceUniqVar(&p.Domain, "domain", nil, "domain")
|
|
f.StringSliceUniqVar(&p.IP, "ip", nil, "ip")
|
|
f.StringSliceUniqVar(&p.CIDR, "cidr", nil, "cidr")
|
|
|
|
err := f.Parse()
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "ERROR: %s\n", err)
|
|
return nil, err
|
|
}
|
|
|
|
return p, err
|
|
}
|
|
|
|
// ListDir returns file list named with suffix in dirPth
|
|
func ListDir(dirPth string, suffix string) (files []string, err error) {
|
|
files = make([]string, 0, 10)
|
|
dir, err := ioutil.ReadDir(dirPth)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
PthSep := string(os.PathSeparator)
|
|
suffix = strings.ToUpper(suffix)
|
|
for _, fi := range dir {
|
|
if fi.IsDir() {
|
|
continue
|
|
}
|
|
if strings.HasSuffix(strings.ToUpper(fi.Name()), suffix) {
|
|
files = append(files, dirPth+PthSep+fi.Name())
|
|
}
|
|
}
|
|
return files, nil
|
|
}
|