chore: Add rule type name constants

Author: layou233

config/root.go

@@ -147,11 +147,11 @@ func LoadConfigFromFile(ctx context.Context, filePath string, watch bool, logger
 				Router: Router{
 					Rules: []*Rule{
 						{
-							Type:  "always",
+							Type:  RuleTypeAlways,
 							Sniff: jsonx.Listable[string]{"minecraft"},
 						},
 						{
-							Type:      "ServiceName",
+							Type:      RuleTypeServiceName,
 							Parameter: json.RawMessage("\"Hypixel-in\""),
 							Rewrite: RuleRewrite{
 								Minecraft: &ruleRewriteMinecraft{

config/rule_type.go

@@ -0,0 +1,16 @@
+package config
+
+const (
+	RuleTypeAlways = "always"
+	RuleTypeAnd    = "and"
+	RuleTypeOr     = "or"
+
+	RuleTypeServiceName         = "ServiceName"
+	RuleTypeSourceIPVersion     = "SourceIPVersion"
+	RuleTypeSourceIP            = "SourceIP"
+	RuleTypeSourcePort          = "SourcePort"
+	RuleTypeMinecraftHostname   = "MinecraftHostname"
+	RuleTypeMinecraftPlayerName = "MinecraftPlayerName"
+	RuleTypeMinecraftStatus     = "MinecraftStatus"
+	RuleTypeMinecraftTransfer   = "MinecraftTransfer"
+)

route/rule.go

@@ -25,38 +25,38 @@ type Rule interface {
 	Match(metadata *adapter.Metadata) bool
 }
 
-func NewRule(logger *log.Logger, config *config.Rule, listMap map[string]set.StringSet, ruleRegistry map[string]CustomRuleInitializer) (Rule, error) {
-	switch config.Type {
-	case "always":
-		return &RuleAlways{config}, nil
-	case "and":
-		return NewLogicalAndRule(logger, config, listMap, ruleRegistry)
-	case "or":
-		return NewLogicalOrRule(logger, config, listMap, ruleRegistry)
-	case "ServiceName":
-		return NewServiceNameRule(config, listMap)
-	case "SourceIPVersion":
-		return NewSourceIPVersionRule(config)
-	case "SourceIP":
-		return NewSourceIPRule(config, listMap)
-	case "SourcePort":
-		return NewSourcePortRule(config)
-	case "MinecraftHostname":
-		return NewMinecraftHostnameRule(config, listMap)
-	case "MinecraftPlayerName":
-		return NewMinecraftPlayerNameRule(config, listMap)
-	case "MinecraftStatus":
-		return NewMinecraftStatusRule(config)
-	case "MinecraftTransfer":
-		return NewMinecraftTransferRule(config)
+func NewRule(logger *log.Logger, ruleConfig *config.Rule, listMap map[string]set.StringSet, ruleRegistry map[string]CustomRuleInitializer) (Rule, error) {
+	switch ruleConfig.Type {
+	case config.RuleTypeAlways:
+		return &RuleAlways{ruleConfig}, nil
+	case config.RuleTypeAnd:
+		return NewLogicalAndRule(logger, ruleConfig, listMap, ruleRegistry)
+	case config.RuleTypeOr:
+		return NewLogicalOrRule(logger, ruleConfig, listMap, ruleRegistry)
+	case config.RuleTypeServiceName:
+		return NewServiceNameRule(ruleConfig, listMap)
+	case config.RuleTypeSourceIPVersion:
+		return NewSourceIPVersionRule(ruleConfig)
+	case config.RuleTypeSourceIP:
+		return NewSourceIPRule(ruleConfig, listMap)
+	case config.RuleTypeSourcePort:
+		return NewSourcePortRule(ruleConfig)
+	case config.RuleTypeMinecraftHostname:
+		return NewMinecraftHostnameRule(ruleConfig, listMap)
+	case config.RuleTypeMinecraftPlayerName:
+		return NewMinecraftPlayerNameRule(ruleConfig, listMap)
+	case config.RuleTypeMinecraftStatus:
+		return NewMinecraftStatusRule(ruleConfig)
+	case config.RuleTypeMinecraftTransfer:
+		return NewMinecraftTransferRule(ruleConfig)
 	}
-	if len(ruleRegistry) > 0 && strings.HasPrefix(config.Type, typeCustomPrefix) {
-		typeName := strings.TrimPrefix(config.Type, typeCustomPrefix)
+	if len(ruleRegistry) > 0 && strings.HasPrefix(ruleConfig.Type, typeCustomPrefix) {
+		typeName := strings.TrimPrefix(ruleConfig.Type, typeCustomPrefix)
 		ruleInitializer, found := ruleRegistry[typeName]
 		if !found {
 			return nil, fmt.Errorf("unknown custom rule type: %s", typeName)
 		}
-		return ruleInitializer(logger, config, listMap)
+		return ruleInitializer(logger, ruleConfig, listMap)
 	}
-	return nil, common.Cause("type ["+config.Type+"]: ", ErrRuleTypeNotFound)
+	return nil, common.Cause("type ["+ruleConfig.Type+"]: ", ErrRuleTypeNotFound)
 }