mihomo/adapter/provider/parser.go

71 lines
2.1 KiB
Go
Raw Normal View History

2019-12-08 12:17:24 +08:00
package provider
import (
"errors"
"fmt"
"github.com/Dreamacro/clash/component/resource"
2019-12-08 12:17:24 +08:00
"time"
"github.com/Dreamacro/clash/common/structure"
C "github.com/Dreamacro/clash/constant"
types "github.com/Dreamacro/clash/constant/provider"
2019-12-08 12:17:24 +08:00
)
2021-10-10 23:44:09 +08:00
var errVehicleType = errors.New("unsupport vehicle type")
2019-12-08 12:17:24 +08:00
type healthCheckSchema struct {
Enable bool `provider:"enable"`
URL string `provider:"url"`
Interval int `provider:"interval"`
Lazy bool `provider:"lazy,omitempty"`
2019-12-08 12:17:24 +08:00
}
type proxyProviderSchema struct {
Type string `provider:"type"`
Path string `provider:"path"`
URL string `provider:"url,omitempty"`
Interval int `provider:"interval,omitempty"`
Filter string `provider:"filter,omitempty"`
ExcludeFilter string `provider:"exclude-filter,omitempty"`
ExcludeType string `provider:"exclude-type,omitempty"`
HealthCheck healthCheckSchema `provider:"health-check,omitempty"`
2019-12-08 12:17:24 +08:00
}
2022-03-16 12:10:13 +08:00
func ParseProxyProvider(name string, mapping map[string]any) (types.ProxyProvider, error) {
2019-12-08 12:17:24 +08:00
decoder := structure.NewDecoder(structure.Option{TagName: "provider", WeaklyTypedInput: true})
schema := &proxyProviderSchema{
HealthCheck: healthCheckSchema{
Lazy: true,
},
}
2019-12-08 12:17:24 +08:00
if err := decoder.Decode(mapping, schema); err != nil {
return nil, err
}
2021-03-24 01:00:21 +08:00
var hcInterval uint
2019-12-08 12:17:24 +08:00
if schema.HealthCheck.Enable {
hcInterval = uint(schema.HealthCheck.Interval)
2019-12-08 12:17:24 +08:00
}
hc := NewHealthCheck([]C.Proxy{}, schema.HealthCheck.URL, hcInterval, schema.HealthCheck.Lazy)
2019-12-08 12:17:24 +08:00
2020-01-30 17:03:11 +08:00
path := C.Path.Resolve(schema.Path)
2019-12-08 12:17:24 +08:00
var vehicle types.Vehicle
2019-12-08 12:17:24 +08:00
switch schema.Type {
case "file":
vehicle = resource.NewFileVehicle(path)
2019-12-08 12:17:24 +08:00
case "http":
vehicle = resource.NewHTTPVehicle(schema.URL, path)
2019-12-08 12:17:24 +08:00
default:
return nil, fmt.Errorf("%w: %s", errVehicleType, schema.Type)
}
interval := time.Duration(uint(schema.Interval)) * time.Second
filter := schema.Filter
excludeFilter := schema.ExcludeFilter
excludeType:=schema.ExcludeType
return NewProxySetProvider(name, interval, filter, excludeFilter,excludeType,vehicle, hc)
2019-12-08 12:17:24 +08:00
}