package outboundgroup import ( "context" "encoding/json" "errors" "github.com/Dreamacro/clash/adapters/outbound" "github.com/Dreamacro/clash/adapters/provider" "github.com/Dreamacro/clash/common/singledo" C "github.com/Dreamacro/clash/constant" ) type Selector struct { *outbound.Base disableUDP bool single *singledo.Single selected string providers []provider.ProxyProvider } func (s *Selector) DialContext(ctx context.Context, metadata *C.Metadata) (C.Conn, error) { c, err := s.selectedProxy(true).DialContext(ctx, metadata) if err == nil { c.AppendToChains(s) } return c, err } func (s *Selector) DialUDP(metadata *C.Metadata) (C.PacketConn, error) { pc, err := s.selectedProxy(true).DialUDP(metadata) if err == nil { pc.AppendToChains(s) } return pc, err } func (s *Selector) SupportUDP() bool { if s.disableUDP { return false } return s.selectedProxy(false).SupportUDP() } func (s *Selector) MarshalJSON() ([]byte, error) { var all []string for _, proxy := range getProvidersProxies(s.providers, false) { all = append(all, proxy.Name()) } return json.Marshal(map[string]interface{}{ "type": s.Type().String(), "now": s.Now(), "all": all, }) } func (s *Selector) Now() string { return s.selectedProxy(false).Name() } func (s *Selector) Set(name string) error { for _, proxy := range getProvidersProxies(s.providers, false) { if proxy.Name() == name { s.selected = name s.single.Reset() return nil } } return errors.New("proxy not exist") } func (s *Selector) Unwrap(metadata *C.Metadata) C.Proxy { return s.selectedProxy(true) } func (s *Selector) selectedProxy(touch bool) C.Proxy { elm, _, _ := s.single.Do(func() (interface{}, error) { proxies := getProvidersProxies(s.providers, touch) for _, proxy := range proxies { if proxy.Name() == s.selected { return proxy, nil } } return proxies[0], nil }) return elm.(C.Proxy) } func NewSelector(options *GroupCommonOption, providers []provider.ProxyProvider) *Selector { selected := providers[0].Proxies()[0].Name() return &Selector{ Base: outbound.NewBase(options.Name, "", C.Selector, false), single: singledo.NewSingle(defaultGetProxiesDuration), providers: providers, selected: selected, disableUDP: options.DisableUDP, } }