feat(main): 增加StripPrefix中间件

BadKid90s · Feb 1, 2024 · 4c320ce · 4c320ce
1 parent d80afc8
commit 4c320ce
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()
+	}
+}