sub2clash/utils/proxy.go

package utils

import (
	"strings"
	"sub2clash/model"
	"sub2clash/parser"
)

func GetContryName(proxy model.Proxy) string {
	// 创建一个切片包含所有的国家映射
	countryMaps := []map[string]string{
		model.CountryFlag,
		model.CountryChineseName,
		model.CountryISO,
		model.CountryEnglishName,
	}

	// 对每一个映射进行检查
	for _, countryMap := range countryMaps {
		for k, v := range countryMap {
			if strings.Contains(proxy.Name, k) {
				return v
			}
		}
	}

	return "其他地区"
}

var skipGroups = map[string]bool{
	"手动切换": true,
	"全球直连": true,
	"广告拦截": true,
	"应用进化": true,
}

func AddProxy(sub *model.Subscription, proxies ...model.Proxy) {
	newCountryGroupNames := make([]string, 0)

	for _, proxy := range proxies {
		sub.Proxies = append(sub.Proxies, proxy)

		haveProxyGroup := false
		countryName := GetContryName(proxy)

		for i := range sub.ProxyGroups {
			group := &sub.ProxyGroups[i]

			if group.Name == countryName {
				group.Proxies = append(group.Proxies, proxy.Name)
				haveProxyGroup = true
			}

			if group.Name == "手动切换" {
				group.Proxies = append(group.Proxies, proxy.Name)
			}
		}

		if !haveProxyGroup {
			newGroup := model.ProxyGroup{
				Name:          countryName,
				Type:          "select",
				Proxies:       []string{proxy.Name},
				IsCountryGrop: true,
			}
			sub.ProxyGroups = append(sub.ProxyGroups, newGroup)
			newCountryGroupNames = append(newCountryGroupNames, countryName)
		}
	}

	for i := range sub.ProxyGroups {
		if sub.ProxyGroups[i].IsCountryGrop {
			continue
		}
		if !skipGroups[sub.ProxyGroups[i].Name] {
			sub.ProxyGroups[i].Proxies = append(newCountryGroupNames, sub.ProxyGroups[i].Proxies...)
		}
	}
}

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, "ss://") {
				proxyItem, err = parser.ParseSS(proxy)
			}
			if strings.HasPrefix(proxy, "trojan://") {
				proxyItem, err = parser.ParseTrojan(proxy)
			}
			if strings.HasPrefix(proxy, "vmess://") {
				proxyItem, err = parser.ParseVmess(proxy)
			}
			if strings.HasPrefix(proxy, "vless://") {
				proxyItem, err = parser.ParseVless(proxy)
			}
			if strings.HasPrefix(proxy, "ssr://") {
				proxyItem, err = parser.ParseShadowsocksR(proxy)
			}
			if err == nil {
				result = append(result, proxyItem)
			}
		}
	}
	return result
}