sub2clash/validator/sub.go

package validator

import (
	"crypto/md5"
	"encoding/hex"
	"errors"
	"github.com/gin-gonic/gin"
	"net/url"
	"os"
	"regexp"
	"strings"
)

type SubQuery struct {
	Sub           string               `form:"sub" binding:""`
	Subs          []string             `form:"-" binding:""`
	Proxy         string               `form:"proxy" binding:""`
	Proxies       []string             `form:"-" binding:""`
	Refresh       bool                 `form:"refresh,default=false" binding:""`
	Template      string               `form:"template" binding:""`
	RuleProvider  string               `form:"ruleProvider" binding:""`
	RuleProviders []RuleProviderStruct `form:"-" binding:""`
	Rule          string               `form:"rule" binding:""`
	Rules         []RuleStruct         `form:"-" binding:""`
	AutoTest      bool                 `form:"autoTest,default=false" binding:""`
	Lazy          bool                 `form:"lazy,default=false" binding:""`
	Sort          string               `form:"sort" binding:""`
}

type RuleProviderStruct struct {
	Behavior string
	Url      string
	Group    string
	Prepend  bool
	Name     string
}

type RuleStruct struct {
	Rule    string
	Prepend bool
}

func ParseQuery(c *gin.Context) (SubQuery, error) {
	var query SubQuery
	if err := c.ShouldBind(&query); err != nil {
		return SubQuery{}, errors.New("参数错误: " + err.Error())
	}
	if query.Sub == "" && query.Proxy == "" {
		return SubQuery{}, errors.New("参数错误: sub 和 proxy 不能同时为空")
	}
	if query.Sub != "" {
		query.Subs = strings.Split(query.Sub, ",")
		for i := range query.Subs {
			if _, err := url.ParseRequestURI(query.Subs[i]); err != nil {
				return SubQuery{}, errors.New("参数错误: " + err.Error())
			}
		}
	} else {
		query.Subs = nil
	}
	if query.Proxy != "" {
		query.Proxies = strings.Split(query.Proxy, ",")
	} else {
		query.Proxies = nil
	}
	if query.Template != "" {
		uri, err := url.ParseRequestURI(query.Template)
		if err != nil {
			if strings.Contains(query.Template, string(os.PathSeparator)) {
				return SubQuery{}, err
			}
		}
		query.Template = uri.String()
	}
	if query.RuleProvider != "" {
		reg := regexp.MustCompile(`\[(.*?)\]`)
		ruleProviders := reg.FindAllStringSubmatch(query.RuleProvider, -1)
		for i := range ruleProviders {
			length := len(ruleProviders)
			parts := strings.Split(ruleProviders[length-i-1][1], ",")
			if len(parts) < 4 {
				return SubQuery{}, errors.New("参数错误: ruleProvider 格式错误")
			}
			u := parts[1]
			uri, err := url.ParseRequestURI(u)
			if err != nil {
				return SubQuery{}, errors.New("参数错误: " + err.Error())
			}
			u = uri.String()
			if len(parts) == 4 {
				hash := md5.Sum([]byte(u))
				parts = append(parts, hex.EncodeToString(hash[:]))
			}
			query.RuleProviders = append(
				query.RuleProviders, RuleProviderStruct{
					Behavior: parts[0],
					Url:      u,
					Group:    parts[2],
					Prepend:  parts[3] == "true",
					Name:     parts[4],
				},
			)
		}
		// 校验 Rule-Provider 是否有重名
		names := make(map[string]bool)
		for _, ruleProvider := range query.RuleProviders {
			if _, ok := names[ruleProvider.Name]; ok {
				return SubQuery{}, errors.New("参数错误: Rule-Provider 名称重复")
			}
			names[ruleProvider.Name] = true
		}
	} else {
		query.RuleProviders = nil
	}
	if query.Rule != "" {
		reg := regexp.MustCompile(`\[(.*?)\]`)
		rules := reg.FindAllStringSubmatch(query.Rule, -1)
		for i := range rules {
			length := len(rules)
			r := rules[length-1-i][1]
			strings.LastIndex(r, ",")
			parts := [2]string{}
			parts[0] = r[:strings.LastIndex(r, ",")]
			parts[1] = r[strings.LastIndex(r, ",")+1:]
			query.Rules = append(
				query.Rules, RuleStruct{
					Rule:    parts[0],
					Prepend: parts[1] == "true",
				},
			)
		}
	} else {
		query.Rules = nil
	}
	return query, nil
}