chore: new rule
This commit is contained in:
@@ -7,12 +7,7 @@ import (
|
||||
C "github.com/Dreamacro/clash/constant"
|
||||
)
|
||||
|
||||
func ParseRewrite(line string) (C.Rewrite, error) {
|
||||
url, others, found := strings.Cut(strings.TrimSpace(line), "url")
|
||||
if !found {
|
||||
return nil, errInvalid
|
||||
}
|
||||
|
||||
func ParseRewrite(line RawMitmRule) (C.Rewrite, error) {
|
||||
var (
|
||||
urlRegx *regexp.Regexp
|
||||
ruleType *C.RewriteType
|
||||
@@ -22,57 +17,37 @@ func ParseRewrite(line string) (C.Rewrite, error) {
|
||||
err error
|
||||
)
|
||||
|
||||
url := line.Url
|
||||
urlRegx, err = regexp.Compile(strings.Trim(url, " "), regexp.None)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
others = strings.Trim(others, " ")
|
||||
first := strings.Split(others, " ")[0]
|
||||
for k, v := range C.RewriteTypeMapping {
|
||||
if k == others {
|
||||
ruleType = &v
|
||||
ruleType = &line.Action
|
||||
switch *ruleType {
|
||||
case C.Mitm302, C.Mitm307:
|
||||
{
|
||||
rulePayload = line.New
|
||||
break
|
||||
}
|
||||
case C.MitmRequestHeader, C.MitmRequestBody, C.MitmResponseHeader, C.MitmResponseBody:
|
||||
{
|
||||
var old string
|
||||
if line.Old == nil {
|
||||
old = ".*"
|
||||
} else {
|
||||
old = *line.Old
|
||||
}
|
||||
|
||||
if k != first {
|
||||
continue
|
||||
}
|
||||
|
||||
rs := trimArr(strings.Split(others, k))
|
||||
l := len(rs)
|
||||
if l > 2 {
|
||||
continue
|
||||
}
|
||||
|
||||
if l == 1 {
|
||||
ruleType = &v
|
||||
rulePayload = rs[0]
|
||||
break
|
||||
} else {
|
||||
ruleRegx, err = regexp.Compile(rs[0], regexp.None)
|
||||
re, err := regexp.Compile(old, regexp.Singleline)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
ruleRegx = re
|
||||
|
||||
ruleType = &v
|
||||
rulePayload = rs[1]
|
||||
break
|
||||
rulePayload = line.New
|
||||
}
|
||||
}
|
||||
|
||||
if ruleType == nil {
|
||||
return nil, errInvalid
|
||||
}
|
||||
|
||||
return NewRewriteRule(urlRegx, *ruleType, ruleRegx, rulePayload), nil
|
||||
}
|
||||
|
||||
func trimArr(arr []string) (r []string) {
|
||||
for _, e := range arr {
|
||||
if s := strings.Trim(e, " "); s != "" {
|
||||
r = append(r, s)
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user