starred/godoxy-yusing
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2026-03-28 03:51:08 +01:00
Code Issues 18 Packages Projects Releases 10 Wiki Activity

improved middleware variable subsititution

Browse Source
This commit is contained in:
yusing
2024-12-04 01:58:17 +08:00
parent fcfb7a0105
commit eabdd3de00
3 changed files with 109 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
internal/net/http/middleware/modify_request.go
View File

@@ -44,7 +44,7 @@ func NewModifyRequest(optsRaw OptionsRaw) (*Middleware, E.Error) {
func (mr *modifyRequest) checkVarSubstitution() {
for _, m := range []map[string]string{mr.SetHeaders, mr.AddHeaders} {
for _, v := range m {
if strings.Contains(v, "$") {
if strings.ContainsRune(v, '$') {
mr.needVarSubstitution = true
return
}
@@ -53,20 +53,32 @@ func (mr *modifyRequest) checkVarSubstitution() {
}
func (mr *modifyRequest) modifyHeaders(req *Request, resp *Response, headers http.Header) {
replaceVars := varReplacerDummy
if mr.needVarSubstitution {
replaceVars = varReplacer(req, resp)
if !mr.needVarSubstitution {
for k, v := range mr.SetHeaders {
if req != nil && strings.ToLower(k) == "host" {
defer func() {
req.Host = v
}()
}
headers.Set(k, v)
}
for k, v := range mr.AddHeaders {
headers.Add(k, v)
}
} else {
for k, v := range mr.SetHeaders {
if req != nil && strings.ToLower(k) == "host" {
defer func() {
req.Host = varReplace(req, resp, v)
}()
}
headers.Set(k, varReplace(req, resp, v))
}
for k, v := range mr.AddHeaders {
headers.Add(k, varReplace(req, resp, v))
}
}
for k, v := range mr.SetHeaders {
if strings.ToLower(k) == "host" {
req.Host = replaceVars(v)
}
headers.Set(k, replaceVars(v))
}
for k, v := range mr.AddHeaders {
headers.Add(k, replaceVars(v))
}
for _, k := range mr.HideHeaders {
headers.Del(k)
}