mirror of
https://github.com/nitezs/sub2clash.git
synced 2024-12-23 21:24:42 -05:00
141 lines
3.7 KiB
Go
141 lines
3.7 KiB
Go
package model
|
|
|
|
import (
|
|
"golang.org/x/text/collate"
|
|
"golang.org/x/text/language"
|
|
)
|
|
|
|
type ProxyGroup struct {
|
|
Name string `yaml:"name,omitempty"`
|
|
Type string `yaml:"type,omitempty"`
|
|
Proxies []string `yaml:"proxies,omitempty"`
|
|
IsCountryGrop bool `yaml:"-"`
|
|
Url string `yaml:"url,omitempty"`
|
|
Interval int `yaml:"interval,omitempty"`
|
|
Tolerance int `yaml:"tolerance,omitempty"`
|
|
Lazy bool `yaml:"lazy"`
|
|
Size int `yaml:"-"`
|
|
DisableUDP bool `yaml:"disable-udp,omitempty"`
|
|
Strategy string `yaml:"strategy,omitempty"`
|
|
}
|
|
|
|
type SelectProxyGroup struct {
|
|
Name string `yaml:"name,omitempty"`
|
|
Type string `yaml:"type,omitempty"`
|
|
Proxies []string `yaml:"proxies,omitempty"`
|
|
DisableUDP bool `yaml:"disable-udp,omitempty"`
|
|
}
|
|
|
|
type UrlTestProxyGroup struct {
|
|
Name string `yaml:"name,omitempty"`
|
|
Type string `yaml:"type,omitempty"`
|
|
Proxies []string `yaml:"proxies,omitempty"`
|
|
Url string `yaml:"url,omitempty"`
|
|
Interval int `yaml:"interval,omitempty"`
|
|
Tolerance int `yaml:"tolerance,omitempty"`
|
|
Lazy bool `yaml:"lazy"`
|
|
DisableUDP bool `yaml:"disable-udp,omitempty"`
|
|
}
|
|
|
|
type LoadBalanceProxyGroup struct {
|
|
Name string `yaml:"name,omitempty"`
|
|
Type string `yaml:"type,omitempty"`
|
|
Proxies []string `yaml:"proxies,omitempty"`
|
|
DisableUDP bool `yaml:"disable-udp,omitempty"`
|
|
Url string `yaml:"url,omitempty"`
|
|
Interval int `yaml:"interval,omitempty"`
|
|
Lazy bool `yaml:"lazy"`
|
|
Strategy string `yaml:"strategy,omitempty"`
|
|
}
|
|
|
|
type RelayProxyGroup struct {
|
|
Name string `yaml:"name,omitempty"`
|
|
Type string `yaml:"type,omitempty"`
|
|
Proxies []string `yaml:"proxies,omitempty"`
|
|
}
|
|
|
|
func (p ProxyGroup) MarshalYAML() (interface{}, error) {
|
|
switch p.Type {
|
|
case "select":
|
|
return SelectProxyGroup{
|
|
Name: p.Name,
|
|
Type: p.Type,
|
|
Proxies: p.Proxies,
|
|
DisableUDP: p.DisableUDP,
|
|
}, nil
|
|
case "url-test", "fallback":
|
|
return UrlTestProxyGroup{
|
|
Name: p.Name,
|
|
Type: p.Type,
|
|
Proxies: p.Proxies,
|
|
Url: p.Url,
|
|
Interval: p.Interval,
|
|
Tolerance: p.Tolerance,
|
|
Lazy: p.Lazy,
|
|
DisableUDP: p.DisableUDP,
|
|
}, nil
|
|
case "load-balance":
|
|
return LoadBalanceProxyGroup{
|
|
Name: p.Name,
|
|
Type: p.Type,
|
|
Proxies: p.Proxies,
|
|
DisableUDP: p.DisableUDP,
|
|
Url: p.Url,
|
|
Interval: p.Interval,
|
|
Lazy: p.Lazy,
|
|
Strategy: p.Strategy,
|
|
}, nil
|
|
case "relay":
|
|
return RelayProxyGroup{
|
|
Name: p.Name,
|
|
Type: p.Type,
|
|
Proxies: p.Proxies,
|
|
}, nil
|
|
default:
|
|
return SelectProxyGroup{
|
|
Name: p.Name,
|
|
Type: p.Type,
|
|
Proxies: p.Proxies,
|
|
}, nil
|
|
}
|
|
}
|
|
|
|
type ProxyGroupsSortByName []ProxyGroup
|
|
type ProxyGroupsSortBySize []ProxyGroup
|
|
|
|
func (p ProxyGroupsSortByName) Len() int {
|
|
return len(p)
|
|
}
|
|
func (p ProxyGroupsSortBySize) Len() int {
|
|
return len(p)
|
|
}
|
|
|
|
func (p ProxyGroupsSortByName) Less(i, j int) bool {
|
|
// 定义一组备选语言:首选英语,其次中文
|
|
tags := []language.Tag{
|
|
language.English,
|
|
language.Chinese,
|
|
}
|
|
matcher := language.NewMatcher(tags)
|
|
|
|
// 假设我们的请求语言是 "zh"(中文),则使用匹配器找到最佳匹配的语言
|
|
bestMatch, _, _ := matcher.Match(language.Make("zh"))
|
|
// 使用最佳匹配的语言进行排序
|
|
c := collate.New(bestMatch)
|
|
return c.CompareString(p[i].Name, p[j].Name) < 0
|
|
}
|
|
|
|
func (p ProxyGroupsSortBySize) Less(i, j int) bool {
|
|
if p[i].Size == p[j].Size {
|
|
return p[i].Name < p[j].Name
|
|
}
|
|
return p[i].Size < p[j].Size
|
|
}
|
|
|
|
func (p ProxyGroupsSortByName) Swap(i, j int) {
|
|
p[i], p[j] = p[j], p[i]
|
|
}
|
|
func (p ProxyGroupsSortBySize) Swap(i, j int) {
|
|
p[i], p[j] = p[j], p[i]
|
|
}
|