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"`
	Icon          string   `yaml:"icon,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"`
	Icon       string   `yaml:"icon,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"`
	Icon       string   `yaml:"icon,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"`
	Icon       string   `yaml:"icon,omitempty"`
}

type RelayProxyGroup struct {
	Name    string   `yaml:"name,omitempty"`
	Type    string   `yaml:"type,omitempty"`
	Proxies []string `yaml:"proxies,omitempty"`
	Icon    string   `yaml:"icon,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,
			Icon:       p.Icon,
		}, 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,
			Icon:       p.Icon,
		}, 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,
			Icon:       p.Icon,
		}, nil
	case "relay":
		return RelayProxyGroup{
			Name:    p.Name,
			Type:    p.Type,
			Proxies: p.Proxies,
			Icon:    p.Icon,
		}, nil
	default:
		return SelectProxyGroup{
			Name:    p.Name,
			Type:    p.Type,
			Proxies: p.Proxies,
			Icon:    p.Icon,
		}, 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]
}