Merge pull request #17 from BadKid90s/strip_prefix

feat(main): 增加StripPrefix中间件
BadKid90s · Apr 24, 2024 · 362cef3 · 362cef3
2 parents 07e70da + 4c320ce
commit 362cef3
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/knife/middleware/strip_prefix.go b/knife/middleware/strip_prefix.go
@@ -0,0 +1,21 @@
+package middleware
+
+import (
+	"knife"
+	"net/url"
+	"strings"
+)
+
+func StripPrefix(prefix string) knife.MiddlewareFunc {
+	return func(context *knife.Context) {
+		r := context.Req
+		p := strings.TrimPrefix(r.URL.Path, prefix)
+		rp := strings.TrimPrefix(r.URL.RawPath, prefix)
+		if len(p) < len(r.URL.Path) && (r.URL.RawPath == "" || len(rp) < len(r.URL.RawPath)) {
+			r.URL = new(url.URL)
+			r.URL.Path = p
+			r.URL.RawPath = rp
+		}
+		context.Next()
+	}
+}