HTTP Header：实现请求方法、域名、状态码等限制，实现内容替换功能

pkg/serverconfigs/shared/http_header_config.go

@@ -2,27 +2,81 @@ package shared
 
 import (
 	"github.com/TeaOSLab/EdgeCommon/pkg/configutils"
+	"regexp"
+	"strings"
 )
 
-// 头部信息定义
+// HTTPHeaderReplaceValue 值替换定义
+type HTTPHeaderReplaceValue struct {
+	Pattern           string `yaml:"pattern" json:"pattern"`
+	Replacement       string `yaml:"replacement" json:"replacement"`
+	IsCaseInsensitive bool   `yaml:"isCaseInsensitive" json:"isCaseInsensitive"` // TODO
+	IsRegexp          bool   `yaml:"isRegexp" json:"isRegexp"`                   // TODO
+
+	patternReg *regexp.Regexp
+}
+
+func (this *HTTPHeaderReplaceValue) Init() error {
+	if this.IsRegexp {
+		var pattern = this.Pattern
+		if this.IsCaseInsensitive && !strings.HasPrefix(pattern, "(?i)") {
+			pattern = "(?i)" + pattern
+		}
+
+		reg, err := regexp.Compile(pattern)
+		if err != nil {
+			return err
+		}
+
+		// TODO 支持匹配名（${name}）和反向引用${1}。。。
+		this.patternReg = reg
+	} else {
+		if this.IsCaseInsensitive {
+			var pattern = "(?i)" + regexp.QuoteMeta(this.Pattern)
+			reg, err := regexp.Compile(pattern)
+			if err != nil {
+				return err
+			}
+			this.patternReg = reg
+		}
+	}
+	return nil
+}
+
+func (this *HTTPHeaderReplaceValue) Replace(value string) string {
+	if this.patternReg != nil {
+		return this.patternReg.ReplaceAllString(value, this.Replacement)
+	} else {
+		return strings.ReplaceAll(value, this.Pattern, this.Replacement)
+	}
+}
+
+// HTTPHeaderConfig 头部信息定义
 type HTTPHeaderConfig struct {
-	Id     int64             `yaml:"id" json:"id"`         // ID
-	IsOn   bool              `yaml:"isOn" json:"isOn"`     // 是否开启
-	Name   string            `yaml:"name" json:"name"`     // Name
-	Value  string            `yaml:"value" json:"value"`   // Value
-	Status *HTTPStatusConfig `yaml:"status" json:"status"` // 支持的状态码 TODO
+	Id    int64  `yaml:"id" json:"id"`       // ID
+	IsOn  bool   `yaml:"isOn" json:"isOn"`   // 是否开启
+	Name  string `yaml:"name" json:"name"`   // Name
+	Value string `yaml:"value" json:"value"` // Value
+
+	Status          *HTTPStatusConfig         `yaml:"status" json:"status"`                   // 支持的状态码
+	DisableRedirect bool                      `yaml:"disableRedirect" json:"disableRedirect"` // 在跳转时不调用
+	ShouldAppend    bool                      `yaml:"shouldAppend" json:"shouldAppend"`       // 是否为附加
+	ShouldReplace   bool                      `yaml:"shouldReplace" json:"shouldReplace"`     // 是否替换值
+	ReplaceValues   []*HTTPHeaderReplaceValue `yaml:"replaceValues" json:"replaceValues"`     // 替换值
+	Methods         []string                  `yaml:"methods" json:"methods"`                 // 请求方法
+	Domains         []string                  `yaml:"domains" json:"domains"`                 // 专属域名
 
 	hasVariables bool
 }
 
-// 获取新Header对象
+// NewHeaderConfig 获取新Header对象
 func NewHeaderConfig() *HTTPHeaderConfig {
 	return &HTTPHeaderConfig{
 		IsOn: true,
 	}
 }
 
-// 校验
+// Init 校验
 func (this *HTTPHeaderConfig) Init() error {
 	this.hasVariables = configutils.HasVariables(this.Value)
 
@@ -33,23 +87,19 @@ func (this *HTTPHeaderConfig) Init() error {
 		}
 	}
 
+	if this.ShouldReplace {
+		for _, v := range this.ReplaceValues {
+			err := v.Init()
+			if err != nil {
+				return err
+			}
+		}
+	}
+
 	return nil
 }
 
-// 判断是否匹配状态码
-func (this *HTTPHeaderConfig) Match(statusCode int) bool {
-	if !this.IsOn {
-		return false
-	}
-
-	if this.Status == nil {
-		return false
-	}
-
-	return this.Status.Match(statusCode)
-}
-
-// 是否有变量
+// HasVariables 是否有变量
 func (this *HTTPHeaderConfig) HasVariables() bool {
 	return this.hasVariables
 }