package controller

import (
	"crypto/md5"
	"encoding/hex"
	"errors"
	"gopkg.in/yaml.v3"
	"regexp"
	"strings"
	"sub2clash/model"
	"sub2clash/parser"
	"sub2clash/utils"
	"sub2clash/validator"
)

func BuildSub(query validator.SubQuery, template string) (
	*model.Subscription, error,
) {
	// 定义变量
	var externalTemplate = query.Template != ""
	var temp *model.Subscription
	var sub *model.Subscription
	var err error
	var templateBytes []byte
	// 加载模板
	if !externalTemplate {
		templateBytes, err = utils.LoadTemplate(template)
		if err != nil {
			return nil, errors.New("加载模板失败: " + err.Error())
		}
	} else {
		templateBytes, err = utils.LoadSubscription(template, query.Refresh)
		if err != nil {
			return nil, errors.New("加载模板失败: " + err.Error())
		}
	}
	// 解析模板
	err = yaml.Unmarshal(templateBytes, &temp)
	if err != nil {
		return nil, errors.New("解析模板失败: " + err.Error())
	}
	// 加载订阅
	for i := range query.Subs {
		data, err := utils.LoadSubscription(query.Subs[i], query.Refresh)
		if err != nil {
			return nil, errors.New("加载订阅失败: " + err.Error())
		}
		// 解析订阅
		var proxyList []model.Proxy
		err = yaml.Unmarshal(data, &sub)
		if err != nil {
			reg, _ := regexp.Compile("(ssr|ss|vmess|trojan|http|https)://")
			if reg.Match(data) {
				proxyList = utils.ParseProxy(strings.Split(string(data), "\n")...)
			} else {
				// 如果无法直接解析，尝试Base64解码
				base64, err := parser.DecodeBase64(string(data))
				if err != nil {
					return nil, errors.New("加载订阅失败: " + err.Error())
				}
				proxyList = utils.ParseProxy(strings.Split(base64, "\n")...)
			}
		} else {
			proxyList = sub.Proxies
		}
		utils.AddProxy(temp, query.AutoTest, query.Lazy, proxyList...)
	}
	// 处理自定义代理
	utils.AddProxy(temp, query.AutoTest, query.Lazy, utils.ParseProxy(query.Proxies...)...)
	// 处理自定义规则
	for _, v := range query.Rules {
		if v.Prepend {
			utils.PrependRules(temp, v.Rule)
		} else {
			utils.AppendRules(temp, v.Rule)
		}
	}
	// 处理自定义 ruleProvider
	for _, v := range query.RuleProviders {
		hash := md5.Sum([]byte(v.Url))
		name := hex.EncodeToString(hash[:])
		provider := model.RuleProvider{
			Type:     "http",
			Behavior: v.Behavior,
			Url:      v.Url,
			Path:     "./" + name + ".yaml",
			Interval: 3600,
		}
		if v.Prepend {
			utils.PrependRuleProvider(
				temp, name, v.Group, provider,
			)
		} else {
			utils.AppenddRuleProvider(
				temp, name, v.Group, provider,
			)
		}
	}
	return temp, nil
}