package outboundgroup import ( "errors" "fmt" "github.com/Dreamacro/clash/adapter/outbound" "github.com/Dreamacro/clash/adapter/provider" "github.com/Dreamacro/clash/common/structure" C "github.com/Dreamacro/clash/constant" types "github.com/Dreamacro/clash/constant/provider" ) var ( errFormat = errors.New("format error") errType = errors.New("unsupport type") errMissProxy = errors.New("`use` or `proxies` missing") errMissHealthCheck = errors.New("`url` or `interval` missing") errDuplicateProvider = errors.New("duplicate provider name") ) type GroupCommonOption struct { outbound.BasicOption Name string `group:"name"` Type string `group:"type"` Proxies []string `group:"proxies,omitempty"` Use []string `group:"use,omitempty"` URL string `group:"url,omitempty"` Interval int `group:"interval,omitempty"` Lazy bool `group:"lazy,omitempty"` DisableUDP bool `group:"disable-udp,omitempty"` } func ParseProxyGroup(config map[string]interface{}, proxyMap map[string]C.Proxy, providersMap map[string]types.ProxyProvider) (C.ProxyAdapter, error) { decoder := structure.NewDecoder(structure.Option{TagName: "group", WeaklyTypedInput: true}) groupOption := &GroupCommonOption{ Lazy: true, } if err := decoder.Decode(config, groupOption); err != nil { return nil, errFormat } if groupOption.Type == "" || groupOption.Name == "" { return nil, errFormat } groupName := groupOption.Name providers := []types.ProxyProvider{} if len(groupOption.Proxies) == 0 && len(groupOption.Use) == 0 { return nil, errMissProxy } if len(groupOption.Proxies) != 0 { ps, err := getProxies(proxyMap, groupOption.Proxies) if err != nil { return nil, err } if _, ok := providersMap[groupName]; ok { return nil, errDuplicateProvider } // select don't need health check if groupOption.Type == "select" || groupOption.Type == "relay" { hc := provider.NewHealthCheck(ps, "", 0, true) pd, err := provider.NewCompatibleProvider(groupName, ps, hc) if err != nil { return nil, err } providers = append(providers, pd) providersMap[groupName] = pd } else { if groupOption.URL == "" || groupOption.Interval == 0 { return nil, errMissHealthCheck } hc := provider.NewHealthCheck(ps, groupOption.URL, uint(groupOption.Interval), groupOption.Lazy) pd, err := provider.NewCompatibleProvider(groupName, ps, hc) if err != nil { return nil, err } providers = append(providers, pd) providersMap[groupName] = pd } } if len(groupOption.Use) != 0 { list, err := getProviders(providersMap, groupOption.Use) if err != nil { return nil, err } providers = append(providers, list...) } var group C.ProxyAdapter switch groupOption.Type { case "url-test": opts := parseURLTestOption(config) group = NewURLTest(groupOption, providers, opts...) case "select": group = NewSelector(groupOption, providers) case "fallback": group = NewFallback(groupOption, providers) case "load-balance": strategy := parseStrategy(config) return NewLoadBalance(groupOption, providers, strategy) case "relay": group = NewRelay(groupOption, providers) default: return nil, fmt.Errorf("%w: %s", errType, groupOption.Type) } return group, nil } func getProxies(mapping map[string]C.Proxy, list []string) ([]C.Proxy, error) { var ps []C.Proxy for _, name := range list { p, ok := mapping[name] if !ok { return nil, fmt.Errorf("'%s' not found", name) } ps = append(ps, p) } return ps, nil } func getProviders(mapping map[string]types.ProxyProvider, list []string) ([]types.ProxyProvider, error) { var ps []types.ProxyProvider for _, name := range list { p, ok := mapping[name] if !ok { return nil, fmt.Errorf("'%s' not found", name) } if p.VehicleType() == types.Compatible { return nil, fmt.Errorf("proxy group %s can't contains in `use`", name) } ps = append(ps, p) } return ps, nil }