Modified replace operation to use regexes in caddyfile

Author: armadi1809

modules/caddyhttp/rewrite/caddyfile.go

@@ -214,7 +214,7 @@ func applyQueryOps(h httpcaddyfile.Helper, qo *queryOps, args []string) error {
 		qo.Rename = append(qo.Rename, queryOpsArguments{Key: renameValKey[0], Val: renameValKey[1]})
 
 	case len(args) == 3:
-		qo.Replace = append(qo.Replace, queryOpsReplacement{Key: key, Search: args[1], Replace: args[2]})
+		qo.Replace = append(qo.Replace, queryOpsReplacement{Key: key, SearchRegexp: args[1], Replace: args[2]})
 
 	default:
 		if len(args) != 2 {

modules/caddyhttp/rewrite/rewrite.go

@@ -497,9 +497,20 @@ type queryOps struct {
 	Delete []string `json:"delete,omitempty"`
 }
 
+// Provision to set up the Replace operation regex compiler.
+func (replacement *queryOpsReplacement) Provision() error {
+	if replacement.SearchRegexp != "" {
+		re, err := regexp.Compile(replacement.SearchRegexp)
+		if err != nil {
+			return fmt.Errorf("replacement for query field '%s': %v", replacement.Key, err)
+		}
+		replacement.re = re
+	}
+	return nil
+}
+
 func (q *queryOps) do(r *http.Request, repl *caddy.Replacer) {
 	query := r.URL.Query()
-
 	for _, renameParam := range q.Rename {
 		key := repl.ReplaceAll(renameParam.Key, "")
 		val := repl.ReplaceAll(renameParam.Val, "")
@@ -529,6 +540,7 @@ func (q *queryOps) do(r *http.Request, repl *caddy.Replacer) {
 	}
 
 	for _, replaceParam := range q.Replace {
+		replaceParam.Provision()
 		key := repl.ReplaceAll(replaceParam.Key, "")
 		search := repl.ReplaceKnown(replaceParam.Search, "")
 		replace := repl.ReplaceKnown(replaceParam.Replace, "")