Skip to content

Commit

Permalink
Feature: add V2Ray subscription support to proxy provider
Browse files Browse the repository at this point in the history
  • Loading branch information
yaling888 authored and MetaCubeX committed Jun 5, 2022
1 parent edc73f3 commit 2188fad
Show file tree
Hide file tree
Showing 3 changed files with 714 additions and 69 deletions.
126 changes: 57 additions & 69 deletions adapter/provider/provider.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,15 +5,15 @@ import (
"errors"
"fmt"
"github.com/dlclark/regexp2"
"math"
"runtime"
"time"

"github.com/Dreamacro/clash/adapter"
"github.com/Dreamacro/clash/common/convert"
C "github.com/Dreamacro/clash/constant"
types "github.com/Dreamacro/clash/constant/provider"

"gopkg.in/yaml.v2"
"gopkg.in/yaml.v3"
)

const (
Expand All @@ -24,13 +24,13 @@ type ProxySchema struct {
Proxies []map[string]any `yaml:"proxies"`
}

// for auto gc
// ProxySetProvider for auto gc
type ProxySetProvider struct {
*proxySetProvider
}

type proxySetProvider struct {
*fetcher
*fetcher[[]C.Proxy]
proxies []C.Proxy
healthCheck *HealthCheck
version uint
Expand All @@ -46,8 +46,7 @@ func (pp *proxySetProvider) MarshalJSON() ([]byte, error) {
"type": pp.Type().String(),
"vehicleType": pp.VehicleType().String(),
"proxies": pp.Proxies(),
//TODO maybe error because year value overflow
"updatedAt": pp.updatedAt,
"updatedAt": pp.updatedAt,
})
}

Expand All @@ -74,10 +73,6 @@ func (pp *proxySetProvider) Initial() error {
}

pp.onUpdate(elm)
if pp.healthCheck.auto() {
defer func() { go pp.healthCheck.process() }()
}

return nil
}

Expand All @@ -100,80 +95,37 @@ func (pp *proxySetProvider) setProxies(proxies []C.Proxy) {
if pp.healthCheck.auto() {
go pp.healthCheck.check()
}

}

func stopProxyProvider(pd *ProxySetProvider) {
pd.healthCheck.close()
pd.fetcher.Destroy()
_ = pd.fetcher.Destroy()
}

func NewProxySetProvider(name string, interval time.Duration, filter string, vehicle types.Vehicle, hc *HealthCheck) (*ProxySetProvider, error) {
//filterReg, err := regexp.Compile(filter)
filterReg, err := regexp2.Compile(filter, 0)
if err != nil {
return nil, fmt.Errorf("invalid filter regex: %w", err)
}

if hc.auto() {
go hc.process()
}

pd := &proxySetProvider{
proxies: []C.Proxy{},
healthCheck: hc,
}

onUpdate := func(elm any) {
ret := elm.([]C.Proxy)
pd.setProxies(ret)
if pd.version == math.MaxUint {
pd.version = 0
} else {
pd.version++
}
}

proxiesParseAndFilter := func(buf []byte) (any, error) {
schema := &ProxySchema{}

if err := yaml.Unmarshal(buf, schema); err != nil {
return nil, err
}

if schema.Proxies == nil {
return nil, errors.New("file must have a `proxies` field")
}

proxies := []C.Proxy{}
for idx, mapping := range schema.Proxies {
name, ok := mapping["name"]
mat, _ := filterReg.FindStringMatch(name.(string))
if ok && len(filter) > 0 && mat == nil {
continue
}
proxy, err := adapter.ParseProxy(mapping)
if err != nil {
return nil, fmt.Errorf("proxy %d error: %w", idx, err)
}
proxies = append(proxies, proxy)
}

if len(proxies) == 0 {
if len(filter) > 0 {
return nil, errors.New("doesn't match any proxy, please check your filter")
}
return nil, errors.New("file doesn't have any proxy")
}

return proxies, nil
}

fetcher := newFetcher(name, interval, vehicle, proxiesParseAndFilter, onUpdate)
fetcher := newFetcher[[]C.Proxy](name, interval, vehicle, proxiesParseAndFilter(filter, filterReg), proxiesOnUpdate(pd))
pd.fetcher = fetcher

wrapper := &ProxySetProvider{pd}
runtime.SetFinalizer(wrapper, stopProxyProvider)
return wrapper, nil
}

// for auto gc
// CompatibleProvider for auto gc
type CompatibleProvider struct {
*compatibleProvider
}
Expand All @@ -182,11 +134,6 @@ type compatibleProvider struct {
name string
healthCheck *HealthCheck
proxies []C.Proxy
version uint
}

func (cp *compatibleProvider) Version() uint {
return cp.version
}

func (cp *compatibleProvider) MarshalJSON() ([]byte, error) {
Expand All @@ -211,10 +158,6 @@ func (cp *compatibleProvider) Update() error {
}

func (cp *compatibleProvider) Initial() error {
if cp.healthCheck.auto() {
go cp.healthCheck.process()
}

return nil
}

Expand Down Expand Up @@ -244,6 +187,10 @@ func NewCompatibleProvider(name string, proxies []C.Proxy, hc *HealthCheck) (*Co
return nil, errors.New("provider need one proxy at least")
}

if hc.auto() {
go hc.process()
}

pd := &compatibleProvider{
name: name,
proxies: proxies,
Expand All @@ -254,3 +201,44 @@ func NewCompatibleProvider(name string, proxies []C.Proxy, hc *HealthCheck) (*Co
runtime.SetFinalizer(wrapper, stopCompatibleProvider)
return wrapper, nil
}

func proxiesOnUpdate(pd *proxySetProvider) func([]C.Proxy) {
return func(elm []C.Proxy) {
pd.setProxies(elm)
}
}

func proxiesParseAndFilter(filter string, filterReg *regexp2.Regexp) parser[[]C.Proxy] {
return func(buf []byte) ([]C.Proxy, error) {
schema := &ProxySchema{}

if err := yaml.Unmarshal(buf, schema); err != nil {
return nil, err
}

if schema.Proxies == nil {
return nil, errors.New("file must have a `proxies` field")
}

proxies := []C.Proxy{}
for idx, mapping := range schema.Proxies {
if name, ok := mapping["name"]; ok && len(filter) > 0 && !filterReg.MatchString(name.(string)) {
continue
}
proxy, err := adapter.ParseProxy(mapping)
if err != nil {
return nil, fmt.Errorf("proxy %d error: %w", idx, err)
}
proxies = append(proxies, proxy)
}

if len(proxies) == 0 {
if len(filter) > 0 {
return nil, errors.New("doesn't match any proxy, please check your filter")
}
return nil, errors.New("file doesn't have any proxy")
}

return proxies, nil
}
}
Loading

0 comments on commit 2188fad

Please sign in to comment.