File: subrouter.go

package info (click to toggle)
golang-github-zenazn-goji 1.0.1-2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, forky, sid, trixie
  • size: 464 kB
  • sloc: makefile: 3
file content (65 lines) | stat: -rw-r--r-- 1,693 bytes parent folder | download | duplicates (3)
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
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
			oldmatch := web.GetMatch(*s.c)
			r.URL.Path = path
			if oldmatch.Handler != nil {
				delete(s.c.Env, web.MatchKey)
			}

			defer func() {
				r.URL.Path = oldpath

				if s.c.Env == nil {
					return
				}
				if oldmatch.Handler != nil {
					s.c.Env[web.MatchKey] = oldmatch
				} else {
					delete(s.c.Env, web.MatchKey)
				}
			}()
		}
	}
	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<_>/.*)$".

This middleware is Match-aware: it will un-set any explicit routing information
contained in the Goji context in order to prevent routing loops when using
explicit routing with sub-routers. See the documentation for Mux.Router for
more.
*/
func SubRouter(c *web.C, h http.Handler) http.Handler {
	return subrouter{c, h}
}