mihomo/adapters/remote/selector.go

package adapters

import (
	"errors"
	"sort"

	C "github.com/Dreamacro/clash/constant"
)

type Selector struct {
	name     string
	selected C.Proxy
	proxies  map[string]C.Proxy
}

func (s *Selector) Name() string {
	return s.name
}

func (s *Selector) Type() C.AdapterType {
	return C.Selector
}

func (s *Selector) Generator(addr *C.Addr) (adapter C.ProxyAdapter, err error) {
	return s.selected.Generator(addr)
}

func (s *Selector) Now() string {
	return s.selected.Name()
}

func (s *Selector) All() []string {
	var all []string
	for k := range s.proxies {
		all = append(all, k)
	}
	sort.Strings(all)
	return all
}

func (s *Selector) Set(name string) error {
	proxy, exist := s.proxies[name]
	if !exist {
		return errors.New("Proxy does not exist")
	}
	s.selected = proxy
	return nil
}

func NewSelector(name string, proxies map[string]C.Proxy) (*Selector, error) {
	if len(proxies) == 0 {
		return nil, errors.New("Provide at least one proxy")
	}

	mapping := make(map[string]C.Proxy)
	var init string
	for k, v := range proxies {
		mapping[k] = v
		init = k
	}
	s := &Selector{
		name:     name,
		proxies:  mapping,
		selected: proxies[init],
	}
	return s, nil
}