97 lines
2.8 KiB
Go
97 lines
2.8 KiB
Go
package provider
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/Dreamacro/clash/adapter/provider"
|
|
"github.com/Dreamacro/clash/common/structure"
|
|
C "github.com/Dreamacro/clash/constant"
|
|
P "github.com/Dreamacro/clash/constant/provider"
|
|
RC "github.com/Dreamacro/clash/rule/common"
|
|
"time"
|
|
)
|
|
|
|
type ruleProviderSchema struct {
|
|
Type string `provider:"type"`
|
|
Behavior string `provider:"behavior"`
|
|
Path string `provider:"path"`
|
|
URL string `provider:"url,omitempty"`
|
|
Interval int `provider:"interval,omitempty"`
|
|
}
|
|
|
|
func ParseRuleProvider(name string, mapping map[string]interface{}) (P.RuleProvider, error) {
|
|
schema := &ruleProviderSchema{}
|
|
decoder := structure.NewDecoder(structure.Option{TagName: "provider", WeaklyTypedInput: true})
|
|
if err := decoder.Decode(mapping, schema); err != nil {
|
|
return nil, err
|
|
}
|
|
var behavior P.RuleType
|
|
|
|
switch schema.Behavior {
|
|
case "domain":
|
|
behavior = P.Domain
|
|
case "ipcidr":
|
|
behavior = P.IPCIDR
|
|
case "classical":
|
|
behavior = P.Classical
|
|
default:
|
|
return nil, fmt.Errorf("unsupported behavior type: %s", schema.Behavior)
|
|
}
|
|
|
|
path := C.Path.Resolve(schema.Path)
|
|
var vehicle P.Vehicle
|
|
switch schema.Type {
|
|
case "file":
|
|
vehicle = provider.NewFileVehicle(path)
|
|
case "http":
|
|
vehicle = provider.NewHTTPVehicle(schema.URL, path)
|
|
default:
|
|
return nil, fmt.Errorf("unsupported vehicle type: %s", schema.Type)
|
|
}
|
|
|
|
return NewRuleSetProvider(name, behavior, time.Duration(uint(schema.Interval))*time.Second, vehicle), nil
|
|
}
|
|
|
|
func parseRule(tp, payload, target string, params []string) (C.Rule, error) {
|
|
var (
|
|
parseErr error
|
|
parsed C.Rule
|
|
)
|
|
|
|
switch tp {
|
|
case "DOMAIN":
|
|
parsed = RC.NewDomain(payload, target)
|
|
case "DOMAIN-SUFFIX":
|
|
parsed = RC.NewDomainSuffix(payload, target)
|
|
case "DOMAIN-KEYWORD":
|
|
parsed = RC.NewDomainKeyword(payload, target)
|
|
case "GEOSITE":
|
|
parsed, parseErr = RC.NewGEOSITE(payload, target)
|
|
case "IP-CIDR", "IP-CIDR6":
|
|
noResolve := RC.HasNoResolve(params)
|
|
parsed, parseErr = RC.NewIPCIDR(payload, target, RC.WithIPCIDRNoResolve(noResolve))
|
|
case "SRC-IP-CIDR":
|
|
parsed, parseErr = RC.NewIPCIDR(payload, target, RC.WithIPCIDRSourceIP(true), RC.WithIPCIDRNoResolve(true))
|
|
case "SRC-PORT":
|
|
parsed, parseErr = RC.NewPort(payload, target, true)
|
|
case "DST-PORT":
|
|
parsed, parseErr = RC.NewPort(payload, target, false)
|
|
case "PROCESS-NAME":
|
|
parsed, parseErr = RC.NewProcess(payload, target, true)
|
|
case "PROCESS-PATH":
|
|
parsed, parseErr = RC.NewProcess(payload, target, false)
|
|
case "GEOIP":
|
|
noResolve := RC.HasNoResolve(params)
|
|
parsed, parseErr = RC.NewGEOIP(payload, target, noResolve)
|
|
default:
|
|
parseErr = fmt.Errorf("unsupported rule type %s", tp)
|
|
}
|
|
if parseErr != nil {
|
|
return nil, parseErr
|
|
}
|
|
ruleExtra := &C.RuleExtra{
|
|
Network: RC.FindNetwork(params),
|
|
SourceIPs: RC.FindSourceIPs(params),
|
|
}
|
|
parsed.SetRuleExtra(ruleExtra)
|
|
return parsed, parseErr
|
|
}
|