Refactor rules
This commit is contained in:
43
route/rule_item_outbound.go
Normal file
43
route/rule_item_outbound.go
Normal file
@@ -0,0 +1,43 @@
|
||||
package route
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"github.com/sagernet/sing-box/adapter"
|
||||
F "github.com/sagernet/sing/common/format"
|
||||
)
|
||||
|
||||
var _ RuleItem = (*OutboundItem)(nil)
|
||||
|
||||
type OutboundItem struct {
|
||||
outbounds []string
|
||||
outboundMap map[string]bool
|
||||
matchAny bool
|
||||
}
|
||||
|
||||
func NewOutboundRule(outbounds []string) *OutboundItem {
|
||||
rule := &OutboundItem{outbounds: outbounds, outboundMap: make(map[string]bool)}
|
||||
for _, outbound := range outbounds {
|
||||
if outbound == "any" {
|
||||
rule.matchAny = true
|
||||
} else {
|
||||
rule.outboundMap[outbound] = true
|
||||
}
|
||||
}
|
||||
return rule
|
||||
}
|
||||
|
||||
func (r *OutboundItem) Match(metadata *adapter.InboundContext) bool {
|
||||
if r.matchAny && metadata.Outbound != "" {
|
||||
return true
|
||||
}
|
||||
return r.outboundMap[metadata.Outbound]
|
||||
}
|
||||
|
||||
func (r *OutboundItem) String() string {
|
||||
if len(r.outbounds) == 1 {
|
||||
return F.ToString("outbound=", r.outbounds[0])
|
||||
} else {
|
||||
return F.ToString("outbound=[", strings.Join(r.outbounds, " "), "]")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user