2019-12-08 04:17:24 +00:00
|
|
|
package outboundgroup
|
2018-07-12 15:28:38 +00:00
|
|
|
|
|
|
|
import (
|
2019-10-12 15:55:39 +00:00
|
|
|
"context"
|
2018-11-21 05:47:46 +00:00
|
|
|
"encoding/json"
|
2018-07-12 15:28:38 +00:00
|
|
|
"errors"
|
|
|
|
|
2021-06-10 06:05:56 +00:00
|
|
|
"github.com/Dreamacro/clash/adapter/outbound"
|
2021-11-07 08:48:51 +00:00
|
|
|
"github.com/Dreamacro/clash/component/dialer"
|
2018-07-12 15:28:38 +00:00
|
|
|
C "github.com/Dreamacro/clash/constant"
|
2021-07-04 12:32:59 +00:00
|
|
|
"github.com/Dreamacro/clash/constant/provider"
|
2018-07-12 15:28:38 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type Selector struct {
|
2022-04-28 11:01:13 +00:00
|
|
|
*GroupBase
|
2020-11-13 13:48:52 +00:00
|
|
|
disableUDP bool
|
|
|
|
selected string
|
2018-10-02 07:26:36 +00:00
|
|
|
}
|
|
|
|
|
2021-04-29 03:23:14 +00:00
|
|
|
// DialContext implements C.ProxyAdapter
|
2021-11-07 08:48:51 +00:00
|
|
|
func (s *Selector) DialContext(ctx context.Context, metadata *C.Metadata, opts ...dialer.Option) (C.Conn, error) {
|
|
|
|
c, err := s.selectedProxy(true).DialContext(ctx, metadata, s.Base.DialOptions(opts...)...)
|
2019-08-08 17:28:37 +00:00
|
|
|
if err == nil {
|
|
|
|
c.AppendToChains(s)
|
|
|
|
}
|
|
|
|
return c, err
|
2018-07-12 15:28:38 +00:00
|
|
|
}
|
|
|
|
|
2021-10-15 13:44:53 +00:00
|
|
|
// ListenPacketContext implements C.ProxyAdapter
|
2021-11-07 08:48:51 +00:00
|
|
|
func (s *Selector) ListenPacketContext(ctx context.Context, metadata *C.Metadata, opts ...dialer.Option) (C.PacketConn, error) {
|
|
|
|
pc, err := s.selectedProxy(true).ListenPacketContext(ctx, metadata, s.Base.DialOptions(opts...)...)
|
2019-08-08 17:28:37 +00:00
|
|
|
if err == nil {
|
|
|
|
pc.AppendToChains(s)
|
|
|
|
}
|
2020-01-31 06:43:54 +00:00
|
|
|
return pc, err
|
2019-04-23 15:29:36 +00:00
|
|
|
}
|
|
|
|
|
2021-04-29 03:23:14 +00:00
|
|
|
// SupportUDP implements C.ProxyAdapter
|
2019-04-23 15:29:36 +00:00
|
|
|
func (s *Selector) SupportUDP() bool {
|
2020-11-13 13:48:52 +00:00
|
|
|
if s.disableUDP {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
2020-11-18 16:53:22 +00:00
|
|
|
return s.selectedProxy(false).SupportUDP()
|
2019-04-23 15:29:36 +00:00
|
|
|
}
|
|
|
|
|
2021-04-29 03:23:14 +00:00
|
|
|
// MarshalJSON implements C.ProxyAdapter
|
2018-11-21 05:47:46 +00:00
|
|
|
func (s *Selector) MarshalJSON() ([]byte, error) {
|
2022-03-23 05:48:21 +00:00
|
|
|
all := []string{}
|
2022-04-28 11:01:13 +00:00
|
|
|
for _, proxy := range s.GetProxies(false) {
|
2019-12-08 04:17:24 +00:00
|
|
|
all = append(all, proxy.Name())
|
|
|
|
}
|
|
|
|
|
2022-03-16 04:10:13 +00:00
|
|
|
return json.Marshal(map[string]any{
|
2018-11-21 05:47:46 +00:00
|
|
|
"type": s.Type().String(),
|
|
|
|
"now": s.Now(),
|
2019-12-08 04:17:24 +00:00
|
|
|
"all": all,
|
2018-11-21 05:47:46 +00:00
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *Selector) Now() string {
|
2020-11-18 16:53:22 +00:00
|
|
|
return s.selectedProxy(false).Name()
|
2018-07-12 15:28:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (s *Selector) Set(name string) error {
|
2022-04-28 11:01:13 +00:00
|
|
|
for _, proxy := range s.GetProxies(false) {
|
2019-12-08 04:17:24 +00:00
|
|
|
if proxy.Name() == name {
|
2020-04-27 13:23:03 +00:00
|
|
|
s.selected = name
|
2019-12-08 04:17:24 +00:00
|
|
|
return nil
|
|
|
|
}
|
2018-07-12 15:28:38 +00:00
|
|
|
}
|
|
|
|
|
2020-08-25 14:19:59 +00:00
|
|
|
return errors.New("proxy not exist")
|
2019-12-08 04:17:24 +00:00
|
|
|
}
|
2018-07-12 15:28:38 +00:00
|
|
|
|
2021-04-29 03:23:14 +00:00
|
|
|
// Unwrap implements C.ProxyAdapter
|
2022-10-30 15:08:18 +00:00
|
|
|
func (s *Selector) Unwrap(metadata *C.Metadata, touch bool) C.Proxy {
|
|
|
|
return s.selectedProxy(touch)
|
2020-05-07 13:42:52 +00:00
|
|
|
}
|
|
|
|
|
2020-11-18 16:53:22 +00:00
|
|
|
func (s *Selector) selectedProxy(touch bool) C.Proxy {
|
2022-04-28 11:01:13 +00:00
|
|
|
proxies := s.GetProxies(touch)
|
|
|
|
for _, proxy := range proxies {
|
|
|
|
if proxy.Name() == s.selected {
|
|
|
|
return proxy
|
2020-04-27 13:23:03 +00:00
|
|
|
}
|
2022-04-28 11:01:13 +00:00
|
|
|
}
|
2020-04-27 13:23:03 +00:00
|
|
|
|
2022-04-28 11:01:13 +00:00
|
|
|
return proxies[0]
|
2019-12-08 04:17:24 +00:00
|
|
|
}
|
2018-10-18 15:24:04 +00:00
|
|
|
|
2021-11-07 08:48:51 +00:00
|
|
|
func NewSelector(option *GroupCommonOption, providers []provider.ProxyProvider) *Selector {
|
2019-12-08 04:17:24 +00:00
|
|
|
return &Selector{
|
2022-04-28 11:01:13 +00:00
|
|
|
GroupBase: NewGroupBase(GroupBaseOption{
|
|
|
|
outbound.BaseOption{
|
|
|
|
Name: option.Name,
|
|
|
|
Type: C.Selector,
|
|
|
|
Interface: option.Interface,
|
|
|
|
RoutingMark: option.RoutingMark,
|
|
|
|
},
|
|
|
|
option.Filter,
|
2022-11-09 00:06:37 +00:00
|
|
|
option.ExcludeFilter,
|
2023-01-07 04:24:28 +00:00
|
|
|
option.ExcludeType,
|
2022-04-28 11:01:13 +00:00
|
|
|
providers,
|
2021-11-07 08:48:51 +00:00
|
|
|
}),
|
2022-01-21 14:38:02 +00:00
|
|
|
selected: "COMPATIBLE",
|
2021-11-07 08:48:51 +00:00
|
|
|
disableUDP: option.DisableUDP,
|
2018-07-12 15:28:38 +00:00
|
|
|
}
|
|
|
|
}
|