118 lines
2.6 KiB
Go
118 lines
2.6 KiB
Go
package outboundgroup
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"github.com/Dreamacro/clash/adapter/outbound"
|
|
"github.com/Dreamacro/clash/component/dialer"
|
|
C "github.com/Dreamacro/clash/constant"
|
|
"github.com/Dreamacro/clash/constant/provider"
|
|
)
|
|
|
|
type Fallback struct {
|
|
*GroupBase
|
|
disableUDP bool
|
|
selected string
|
|
}
|
|
|
|
func (f *Fallback) Now() string {
|
|
proxy := f.findAliveProxy(false)
|
|
return proxy.Name()
|
|
}
|
|
|
|
// DialContext implements C.ProxyAdapter
|
|
func (f *Fallback) DialContext(ctx context.Context, metadata *C.Metadata, opts ...dialer.Option) (C.Conn, error) {
|
|
proxy := f.findAliveProxy(true)
|
|
c, err := proxy.DialContext(ctx, metadata, f.Base.DialOptions(opts...)...)
|
|
if err == nil {
|
|
c.AppendToChains(f)
|
|
f.onDialSuccess()
|
|
} else {
|
|
f.onDialFailed()
|
|
}
|
|
|
|
return c, err
|
|
}
|
|
|
|
// ListenPacketContext implements C.ProxyAdapter
|
|
func (f *Fallback) ListenPacketContext(ctx context.Context, metadata *C.Metadata, opts ...dialer.Option) (C.PacketConn, error) {
|
|
proxy := f.findAliveProxy(true)
|
|
pc, err := proxy.ListenPacketContext(ctx, metadata, f.Base.DialOptions(opts...)...)
|
|
if err == nil {
|
|
pc.AppendToChains(f)
|
|
}
|
|
|
|
return pc, err
|
|
}
|
|
|
|
// SupportUDP implements C.ProxyAdapter
|
|
func (f *Fallback) SupportUDP() bool {
|
|
if f.disableUDP {
|
|
return false
|
|
}
|
|
|
|
proxy := f.findAliveProxy(false)
|
|
return proxy.SupportUDP()
|
|
}
|
|
|
|
// MarshalJSON implements C.ProxyAdapter
|
|
func (f *Fallback) MarshalJSON() ([]byte, error) {
|
|
all := []string{}
|
|
for _, proxy := range f.GetProxies(false) {
|
|
all = append(all, proxy.Name())
|
|
}
|
|
return json.Marshal(map[string]any{
|
|
"type": f.Type().String(),
|
|
"now": f.Now(),
|
|
"all": all,
|
|
})
|
|
}
|
|
|
|
// Unwrap implements C.ProxyAdapter
|
|
func (f *Fallback) Unwrap(metadata *C.Metadata) C.Proxy {
|
|
proxy := f.findAliveProxy(true)
|
|
return proxy
|
|
}
|
|
|
|
func (f *Fallback) findAliveProxy(touch bool) C.Proxy {
|
|
proxies := f.GetProxies(touch)
|
|
al := proxies[0]
|
|
for i := len(proxies) - 1; i > -1; i-- {
|
|
proxy := proxies[i]
|
|
if proxy.Name() == f.selected && proxy.Alive() {
|
|
return proxy
|
|
}
|
|
if proxy.Alive() {
|
|
al = proxy
|
|
}
|
|
}
|
|
return al
|
|
}
|
|
|
|
func (f *Fallback) Set(name string) error {
|
|
for _, proxy := range f.GetProxies(false) {
|
|
if proxy.Name() == name {
|
|
f.selected = name
|
|
return nil
|
|
}
|
|
}
|
|
|
|
return errors.New("proxy not exist")
|
|
}
|
|
|
|
func NewFallback(option *GroupCommonOption, providers []provider.ProxyProvider) *Fallback {
|
|
return &Fallback{
|
|
GroupBase: NewGroupBase(GroupBaseOption{
|
|
outbound.BaseOption{
|
|
Name: option.Name,
|
|
Type: C.Fallback,
|
|
Interface: option.Interface,
|
|
RoutingMark: option.RoutingMark,
|
|
},
|
|
option.Filter,
|
|
providers,
|
|
}),
|
|
disableUDP: option.DisableUDP,
|
|
}
|
|
}
|