clash/rules/logic/or.go

package logic

import (
	"fmt"
	C "github.com/Dreamacro/clash/constant"
	"github.com/Dreamacro/clash/rules/common"
	"strings"
)

type OR struct {
	*common.Base
	rules   []C.Rule
	payload string
	adapter string
	needIP  bool
}

func (or *OR) ShouldFindProcess() bool {
	return false
}

func (or *OR) RuleType() C.RuleType {
	return C.OR
}

func (or *OR) Match(metadata *C.Metadata) (bool, string) {
	for _, rule := range or.rules {
		if m, _ := rule.Match(metadata); m {
			return true, or.adapter
		}
	}

	return false, ""
}

func (or *OR) Adapter() string {
	return or.adapter
}

func (or *OR) Payload() string {
	return or.payload
}

func (or *OR) ShouldResolveIP() bool {
	return or.needIP
}

func NewOR(payload string, adapter string, parse func(tp, payload, target string, params []string, subRules *map[string][]C.Rule) (parsed C.Rule, parseErr error)) (*OR, error) {
	or := &OR{Base: &common.Base{}, payload: payload, adapter: adapter}
	rules, err := ParseRuleByPayload(payload, parse)
	if err != nil {
		return nil, err
	}

	or.rules = rules
	payloads := make([]string, 0, len(rules))
	for _, rule := range rules {
		payloads = append(payloads, fmt.Sprintf("(%s,%s)", rule.RuleType(), rule.Payload()))
		if rule.ShouldResolveIP() {
			or.needIP = true
			break
		}
	}

	or.payload = fmt.Sprintf("(%s)", strings.Join(payloads, " || "))
	return or, nil
}