gophertype/pkg/routes.go

54 lines
1.3 KiB
Go

package gophertype
import (
"fmt"
"log"
"net/http"
"git.kirsle.net/apps/gophertype/pkg/controllers"
"git.kirsle.net/apps/gophertype/pkg/glue"
"git.kirsle.net/apps/gophertype/pkg/middleware"
"github.com/gorilla/mux"
)
// SetupRouter sets up the HTTP router.
func (s *Site) SetupRouter() error {
router := mux.NewRouter()
router.Use(middleware.CSRF)
for _, route := range glue.GetControllers() {
log.Printf("Register: %+v", route)
if len(route.Methods) == 0 {
route.Methods = []string{"GET"}
}
route.Methods = append(route.Methods, "HEAD")
if len(route.Middleware) > 0 {
log.Printf("%+v has middlewares!", route)
handler := route.Middleware[0](http.HandlerFunc(route.Handler))
router.Handle(route.Path, handler).Methods(route.Methods...)
} else {
router.HandleFunc(route.Path, route.Handler).Methods(route.Methods...)
}
}
router.PathPrefix("/").HandlerFunc(controllers.CatchAllHandler)
router.NotFoundHandler = http.HandlerFunc(controllers.CatchAllHandler)
log.Println("Walk the mux.Router:")
router.Walk(func(route *mux.Route, router *mux.Router, ancestors []*mux.Route) error {
tpl, err1 := route.GetPathTemplate()
met, err2 := route.GetMethods()
fmt.Println(tpl, err1, met, err2)
return nil
})
s.mux = router
s.n.UseHandler(router)
return nil
}