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, autotest bool, lazy bool, 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 {
			var newGroup model.ProxyGroup
			if !autotest {
				newGroup = model.ProxyGroup{
					Name:          countryName,
					Type:          "select",
					Proxies:       []string{proxy.Name},
					IsCountryGrop: true,
				}
			} 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,
				}
			}
			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] {
			combined := make([]string, len(newCountryGroupNames)+len(sub.ProxyGroups[i].Proxies))
			copy(combined, newCountryGroupNames)
			copy(combined[len(newCountryGroupNames):], sub.ProxyGroups[i].Proxies)
			sub.ProxyGroups[i].Proxies = combined
		}
	}
}

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
}