85 lines
1.6 KiB
Go
85 lines
1.6 KiB
Go
package router
|
|
|
|
import (
|
|
"net/http"
|
|
"regexp"
|
|
)
|
|
|
|
type Handler func(*Context)
|
|
|
|
type Router struct {
|
|
middle []Handler
|
|
routes []Route
|
|
}
|
|
|
|
type Route struct {
|
|
regex *regexp.Regexp
|
|
chain []Handler
|
|
}
|
|
|
|
func NewRouter() *Router {
|
|
router := &Router{}
|
|
router.middle = make([]Handler, 0)
|
|
router.routes = make([]Route, 0)
|
|
return router
|
|
}
|
|
|
|
func (r *Router) Use(h Handler) {
|
|
r.middle = append(r.middle, h)
|
|
}
|
|
|
|
func (r *Router) For(path string, handler Handler) {
|
|
x := Route{}
|
|
x.regex = routeRegexp(path)
|
|
x.chain = append(r.middle, handler)
|
|
|
|
r.routes = append(r.routes, x)
|
|
}
|
|
|
|
func (r *Router) resolve(path string) *Route {
|
|
for _, r := range r.routes {
|
|
if r.regex.MatchString(path) {
|
|
return &r
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *Router) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
|
|
path := req.Url.Path
|
|
|
|
route := r.resolve(path)
|
|
if route == nil {
|
|
rw.WriteHeader(http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
context := &Context{}
|
|
context.Req = req
|
|
context.Out = rw
|
|
context.vars = regexGroups(path, route.regex)
|
|
context.index = -1
|
|
context.chain = route.chain
|
|
context.Next()
|
|
}
|
|
|
|
func regexGroups(input string, regex *regexp.Regexp) map[string]string {
|
|
groups := make(map[string]string)
|
|
matches := regex.FindStringSubmatchIndex(input)
|
|
for i, key := range regex.SubexpNames()[1:] {
|
|
groups[key] = input[matches[i*2+2]:matches[i*2+3]]
|
|
}
|
|
return groups
|
|
}
|
|
|
|
func routeRegexp(route string) *regexp.Regexp {
|
|
chunks := regexp.MustCompile(`[\*\:]\w+`)
|
|
output := chunks.ReplaceAllStringFunc(route, func(m string) string {
|
|
if m[0:1] == `*` {
|
|
return "(?P<" + m[1:] + ">.+)"
|
|
}
|
|
return "(?P<" + m[1:] + ">[^/]+)"
|
|
})
|
|
output = "^" + output + "$"
|
|
return regexp.MustCompile(output)
|
|
}
|