feat: add DOMAIN-REGEX rule

This commit is contained in:
xishang0128 2024-03-07 23:32:07 +08:00
parent fad1a08378
commit 234a4bfc93
No known key found for this signature in database
GPG key ID: 44A1E10B5ADF68CB
3 changed files with 48 additions and 1 deletions

View file

@ -5,6 +5,7 @@ const (
Domain RuleType = iota
DomainSuffix
DomainKeyword
DomainRegex
GEOSITE
GEOIP
IPCIDR
@ -40,6 +41,8 @@ func (rt RuleType) String() string {
return "DomainSuffix"
case DomainKeyword:
return "DomainKeyword"
case DomainRegex:
return "DomainRegex"
case GEOSITE:
return "GeoSite"
case GEOIP:

View file

@ -0,0 +1,42 @@
package common
import (
"regexp"
"strings"
C "github.com/metacubex/mihomo/constant"
)
type DomainRegex struct {
*Base
regex string
adapter string
}
func (dr *DomainRegex) RuleType() C.RuleType {
return C.DomainRegex
}
func (dr *DomainRegex) Match(metadata *C.Metadata) (bool, string) {
domain := metadata.RuleHost()
match, _ := regexp.MatchString(dr.regex, domain)
return match, dr.adapter
}
func (dr *DomainRegex) Adapter() string {
return dr.adapter
}
func (dr *DomainRegex) Payload() string {
return dr.regex
}
func NewDomainRegex(regex string, adapter string) *DomainRegex {
return &DomainRegex{
Base: &Base{},
regex: strings.ToLower(regex),
adapter: adapter,
}
}
//var _ C.Rule = (*DomainRegex)(nil)

View file

@ -17,6 +17,8 @@ func ParseRule(tp, payload, target string, params []string, subRules map[string]
parsed = RC.NewDomainSuffix(payload, target)
case "DOMAIN-KEYWORD":
parsed = RC.NewDomainKeyword(payload, target)
case "DOMAIN-REGEX":
parsed = RC.NewDomainRegex(payload, target)
case "GEOSITE":
parsed, parseErr = RC.NewGEOSITE(payload, target)
case "GEOIP":