2023-09-12 06:40:24 -04:00
|
|
|
package validator
|
|
|
|
|
2023-09-13 01:47:22 -04:00
|
|
|
import (
|
2023-09-14 12:13:45 -04:00
|
|
|
"crypto/md5"
|
|
|
|
"encoding/hex"
|
2023-09-13 01:47:22 -04:00
|
|
|
"errors"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"net/url"
|
2023-09-14 20:26:48 -04:00
|
|
|
"os"
|
2023-09-13 01:47:22 -04:00
|
|
|
"regexp"
|
|
|
|
"strings"
|
|
|
|
)
|
|
|
|
|
2023-09-12 06:40:24 -04:00
|
|
|
type SubQuery struct {
|
2023-09-13 01:47:22 -04:00
|
|
|
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:""`
|
2023-09-14 12:13:45 -04:00
|
|
|
Sort string `form:"sort" binding:""`
|
2023-09-13 01:47:22 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
type RuleProviderStruct struct {
|
|
|
|
Behavior string
|
|
|
|
Url string
|
|
|
|
Group string
|
|
|
|
Prepend bool
|
2023-09-14 12:13:45 -04:00
|
|
|
Name string
|
2023-09-13 01:47:22 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
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 != "" {
|
2023-09-14 20:26:48 -04:00
|
|
|
uri, err := url.ParseRequestURI(query.Template)
|
2023-09-13 01:47:22 -04:00
|
|
|
if err != nil {
|
2023-09-14 20:26:48 -04:00
|
|
|
if strings.Contains(query.Template, string(os.PathSeparator)) {
|
|
|
|
return SubQuery{}, err
|
|
|
|
}
|
2023-09-13 01:47:22 -04:00
|
|
|
}
|
|
|
|
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], ",")
|
2023-09-14 12:13:45 -04:00
|
|
|
if len(parts) < 4 {
|
2023-09-13 01:47:22 -04:00
|
|
|
return SubQuery{}, errors.New("参数错误: ruleProvider 格式错误")
|
|
|
|
}
|
|
|
|
u := parts[1]
|
|
|
|
uri, err := url.ParseRequestURI(u)
|
|
|
|
if err != nil {
|
|
|
|
return SubQuery{}, errors.New("参数错误: " + err.Error())
|
|
|
|
}
|
2023-09-14 20:26:48 -04:00
|
|
|
u = uri.String()
|
2023-09-14 12:13:45 -04:00
|
|
|
if len(parts) == 4 {
|
|
|
|
hash := md5.Sum([]byte(u))
|
|
|
|
parts = append(parts, hex.EncodeToString(hash[:]))
|
|
|
|
}
|
2023-09-13 01:47:22 -04:00
|
|
|
query.RuleProviders = append(
|
|
|
|
query.RuleProviders, RuleProviderStruct{
|
|
|
|
Behavior: parts[0],
|
|
|
|
Url: u,
|
|
|
|
Group: parts[2],
|
|
|
|
Prepend: parts[3] == "true",
|
2023-09-14 12:13:45 -04:00
|
|
|
Name: parts[4],
|
2023-09-13 01:47:22 -04:00
|
|
|
},
|
|
|
|
)
|
|
|
|
}
|
|
|
|
} 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
|
2023-09-12 06:40:24 -04:00
|
|
|
}
|