mirror of
https://github.com/caddyserver/caddy.git
synced 2025-06-04 19:24:42 +08:00
Moved rewrite middleware into its own package
This commit is contained in:
48
middleware/rewrite/rewrite.go
Normal file
48
middleware/rewrite/rewrite.go
Normal file
@ -0,0 +1,48 @@
|
||||
// Package rewrite is middleware for rewriting requests internally to
|
||||
// a different path.
|
||||
package rewrite
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/mholt/caddy/middleware"
|
||||
)
|
||||
|
||||
// New instantiates a new rewrite middleware.
|
||||
func New(c middleware.Controller) (middleware.Middleware, error) {
|
||||
var rewrites []rewrite
|
||||
|
||||
for c.Next() {
|
||||
var rule rewrite
|
||||
|
||||
if !c.NextArg() {
|
||||
return nil, c.ArgErr()
|
||||
}
|
||||
rule.From = c.Val()
|
||||
|
||||
if !c.NextArg() {
|
||||
return nil, c.ArgErr()
|
||||
}
|
||||
rule.To = c.Val()
|
||||
|
||||
rewrites = append(rewrites, rule)
|
||||
}
|
||||
|
||||
return func(next http.HandlerFunc) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
for _, rule := range rewrites {
|
||||
if r.URL.Path == rule.From {
|
||||
r.URL.Path = rule.To
|
||||
break
|
||||
}
|
||||
}
|
||||
next(w, r)
|
||||
}
|
||||
}, nil
|
||||
}
|
||||
|
||||
// rewrite describes an internal location rewrite rule.
|
||||
type rewrite struct {
|
||||
From string
|
||||
To string
|
||||
}
|
Reference in New Issue
Block a user