mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-12-25 19:55:38 +08:00
83 lines
1.7 KiB
Go
83 lines
1.7 KiB
Go
package constant
|
|
|
|
import (
|
|
"regexp"
|
|
)
|
|
|
|
var RewriteTypeMapping = map[string]RewriteType{
|
|
MitmReject.String(): MitmReject,
|
|
MitmReject200.String(): MitmReject200,
|
|
MitmRejectImg.String(): MitmRejectImg,
|
|
MitmRejectDict.String(): MitmRejectDict,
|
|
MitmRejectArray.String(): MitmRejectArray,
|
|
Mitm302.String(): Mitm302,
|
|
Mitm307.String(): Mitm307,
|
|
MitmRequestHeader.String(): MitmRequestHeader,
|
|
MitmRequestBody.String(): MitmRequestBody,
|
|
MitmResponseHeader.String(): MitmResponseHeader,
|
|
MitmResponseBody.String(): MitmResponseBody,
|
|
}
|
|
|
|
const (
|
|
MitmReject RewriteType = iota
|
|
MitmReject200
|
|
MitmRejectImg
|
|
MitmRejectDict
|
|
MitmRejectArray
|
|
|
|
Mitm302
|
|
Mitm307
|
|
|
|
MitmRequestHeader
|
|
MitmRequestBody
|
|
|
|
MitmResponseHeader
|
|
MitmResponseBody
|
|
)
|
|
|
|
type RewriteType int
|
|
|
|
func (rt RewriteType) String() string {
|
|
switch rt {
|
|
case MitmReject:
|
|
return "reject" // 404
|
|
case MitmReject200:
|
|
return "reject-200"
|
|
case MitmRejectImg:
|
|
return "reject-img"
|
|
case MitmRejectDict:
|
|
return "reject-dict"
|
|
case MitmRejectArray:
|
|
return "reject-array"
|
|
case Mitm302:
|
|
return "302"
|
|
case Mitm307:
|
|
return "307"
|
|
case MitmRequestHeader:
|
|
return "request-header"
|
|
case MitmRequestBody:
|
|
return "request-body"
|
|
case MitmResponseHeader:
|
|
return "response-header"
|
|
case MitmResponseBody:
|
|
return "response-body"
|
|
default:
|
|
return "Unknown"
|
|
}
|
|
}
|
|
|
|
type Rewrite interface {
|
|
ID() string
|
|
URLRegx() *regexp.Regexp
|
|
RuleType() RewriteType
|
|
RuleRegx() *regexp.Regexp
|
|
RulePayload() string
|
|
ReplaceURLPayload([]string) string
|
|
ReplaceSubPayload(string) string
|
|
}
|
|
|
|
type RewriteRule interface {
|
|
SearchInRequest(func(Rewrite) bool) bool
|
|
SearchInResponse(func(Rewrite) bool) bool
|
|
}
|