mirror of
https://github.com/nitezs/sub2clash.git
synced 2024-12-24 12:34:43 -05:00
141 lines
3.3 KiB
Go
141 lines
3.3 KiB
Go
package common
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/nitezs/sub2clash/constant"
|
|
"github.com/nitezs/sub2clash/logger"
|
|
"github.com/nitezs/sub2clash/model"
|
|
"github.com/nitezs/sub2clash/parser"
|
|
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func GetContryName(countryKey string) string {
|
|
|
|
countryMaps := []map[string]string{
|
|
model.CountryFlag,
|
|
model.CountryChineseName,
|
|
model.CountryISO,
|
|
model.CountryEnglishName,
|
|
}
|
|
|
|
for i, countryMap := range countryMaps {
|
|
if i == 2 {
|
|
|
|
splitChars := []string{"-", "_", " "}
|
|
key := make([]string, 0)
|
|
for _, splitChar := range splitChars {
|
|
slic := strings.Split(countryKey, splitChar)
|
|
for _, v := range slic {
|
|
if len(v) == 2 {
|
|
key = append(key, v)
|
|
}
|
|
}
|
|
}
|
|
|
|
for _, v := range key {
|
|
|
|
if country, ok := countryMap[strings.ToUpper(v)]; ok {
|
|
return country
|
|
}
|
|
}
|
|
}
|
|
for k, v := range countryMap {
|
|
if strings.Contains(countryKey, k) {
|
|
return v
|
|
}
|
|
}
|
|
}
|
|
return "其他地区"
|
|
}
|
|
|
|
func AddProxy(
|
|
sub *model.Subscription, autotest bool,
|
|
lazy bool, clashType model.ClashType, proxies ...model.Proxy,
|
|
) {
|
|
proxyTypes := model.GetSupportProxyTypes(clashType)
|
|
|
|
for _, proxy := range proxies {
|
|
if !proxyTypes[proxy.Type] {
|
|
continue
|
|
}
|
|
sub.Proxies = append(sub.Proxies, proxy)
|
|
haveProxyGroup := false
|
|
countryName := GetContryName(proxy.Name)
|
|
for i := range sub.ProxyGroups {
|
|
group := &sub.ProxyGroups[i]
|
|
if group.Name == countryName {
|
|
group.Proxies = append(group.Proxies, proxy.Name)
|
|
group.Size++
|
|
haveProxyGroup = true
|
|
}
|
|
}
|
|
if !haveProxyGroup {
|
|
var newGroup model.ProxyGroup
|
|
if !autotest {
|
|
newGroup = model.ProxyGroup{
|
|
Name: countryName,
|
|
Type: "select",
|
|
Proxies: []string{proxy.Name},
|
|
IsCountryGrop: true,
|
|
Size: 1,
|
|
}
|
|
} else {
|
|
newGroup = model.ProxyGroup{
|
|
Name: countryName,
|
|
Type: "url-test",
|
|
Proxies: []string{proxy.Name},
|
|
IsCountryGrop: true,
|
|
Url: "http://www.gstatic.com/generate_204",
|
|
Interval: 300,
|
|
Tolerance: 50,
|
|
Lazy: lazy,
|
|
Size: 1,
|
|
}
|
|
}
|
|
sub.ProxyGroups = append(sub.ProxyGroups, newGroup)
|
|
}
|
|
}
|
|
}
|
|
|
|
func ParseProxy(proxies ...string) []model.Proxy {
|
|
var result []model.Proxy
|
|
for _, proxy := range proxies {
|
|
if proxy != "" {
|
|
var proxyItem model.Proxy
|
|
var err error
|
|
|
|
if strings.HasPrefix(proxy, constant.ShadowsocksPrefix) {
|
|
proxyItem, err = parser.ParseShadowsocks(proxy)
|
|
}
|
|
if strings.HasPrefix(proxy, constant.TrojanPrefix) {
|
|
proxyItem, err = parser.ParseTrojan(proxy)
|
|
}
|
|
if strings.HasPrefix(proxy, constant.VMessPrefix) {
|
|
proxyItem, err = parser.ParseVmess(proxy)
|
|
}
|
|
if strings.HasPrefix(proxy, constant.VLESSPrefix) {
|
|
proxyItem, err = parser.ParseVless(proxy)
|
|
}
|
|
if strings.HasPrefix(proxy, constant.ShadowsocksRPrefix) {
|
|
proxyItem, err = parser.ParseShadowsocksR(proxy)
|
|
}
|
|
if strings.HasPrefix(proxy, constant.Hysteria2Prefix1) || strings.HasPrefix(proxy, constant.Hysteria2Prefix2) {
|
|
proxyItem, err = parser.ParseHysteria2(proxy)
|
|
}
|
|
if strings.HasPrefix(proxy, constant.HysteriaPrefix) {
|
|
proxyItem, err = parser.ParseHysteria(proxy)
|
|
}
|
|
if err == nil {
|
|
result = append(result, proxyItem)
|
|
} else {
|
|
logger.Logger.Debug(
|
|
"parse proxy failed", zap.String("proxy", proxy), zap.Error(err),
|
|
)
|
|
}
|
|
}
|
|
}
|
|
return result
|
|
}
|