forked from zenazn/goji
/
subrouter.go
45 lines (40 loc) · 1.18 KB
/
subrouter.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
package middleware
import (
"net/http"
"github.com/zenazn/goji/web"
)
type subrouter struct {
c *web.C
h http.Handler
}
func (s subrouter) ServeHTTP(w http.ResponseWriter, r *http.Request) {
if s.c.URLParams != nil {
path, ok := s.c.URLParams["*"]
if !ok {
path, ok = s.c.URLParams["_"]
}
if ok {
oldpath := r.URL.Path
r.URL.Path = path
defer func() {
r.URL.Path = oldpath
}()
}
}
s.h.ServeHTTP(w, r)
}
// SubRouter is a helper middleware that makes writing sub-routers easier.
//
// If you register a sub-router under a key like "/admin/*", Goji's router will
// automatically set c.URLParams["*"] to the unmatched path suffix. This
// middleware will help you set the request URL's Path to this unmatched suffix,
// allowing you to write sub-routers with no knowledge of what routes the parent
// router matches.
//
// Since Go's regular expressions do not allow you to create a capturing group
// named "*", SubRouter also accepts the string "_". For instance, to duplicate
// the semantics of the string pattern "/foo/*", you might use the regular
// expression "^/foo(?P<_>/.*)$".
func SubRouter(c *web.C, h http.Handler) http.Handler {
return subrouter{c, h}
}