51 lines
990 B
Go
51 lines
990 B
Go
package common
|
|
|
|
import (
|
|
"golang.org/x/net/idna"
|
|
"strings"
|
|
|
|
C "github.com/Dreamacro/clash/constant"
|
|
)
|
|
|
|
type DomainKeyword struct {
|
|
*Base
|
|
keyword string
|
|
adapter string
|
|
isIDNA bool
|
|
}
|
|
|
|
func (dk *DomainKeyword) RuleType() C.RuleType {
|
|
return C.DomainKeyword
|
|
}
|
|
|
|
func (dk *DomainKeyword) Match(metadata *C.Metadata) bool {
|
|
if metadata.AddrType != C.AtypDomainName {
|
|
return false
|
|
}
|
|
domain := metadata.Host
|
|
return strings.Contains(domain, dk.keyword)
|
|
}
|
|
|
|
func (dk *DomainKeyword) Adapter() string {
|
|
return dk.adapter
|
|
}
|
|
|
|
func (dk *DomainKeyword) Payload() string {
|
|
keyword := dk.keyword
|
|
if dk.isIDNA {
|
|
keyword, _ = idna.ToUnicode(keyword)
|
|
}
|
|
return keyword
|
|
}
|
|
|
|
func NewDomainKeyword(keyword string, adapter string) *DomainKeyword {
|
|
actualDomainKeyword, _ := idna.ToASCII(keyword)
|
|
return &DomainKeyword{
|
|
Base: &Base{},
|
|
keyword: strings.ToLower(actualDomainKeyword),
|
|
adapter: adapter,
|
|
isIDNA: keyword != actualDomainKeyword,
|
|
}
|
|
}
|
|
|
|
var _ C.Rule = (*DomainKeyword)(nil)
|