mirror of
https://github.com/nitezs/sub2clash.git
synced 2024-12-25 08:50:54 -05:00
101 lines
2.5 KiB
Go
101 lines
2.5 KiB
Go
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
|
||
}
|