package main import ( "context" "fmt" "html/template" "net/http" "nilfm.cc/git/quartzgun/indentalUserDB" "nilfm.cc/git/quartzgun/middleware" "nilfm.cc/git/quartzgun/renderer" "nilfm.cc/git/quartzgun/router" "testing" ) func AddContent(next http.Handler) http.Handler { handlerFunc := func(w http.ResponseWriter, req *http.Request) { if !req.Form.Has("Content") { req.Form.Add("Content", "Yesssssss") } next.ServeHTTP(w, req) } return http.HandlerFunc(handlerFunc) } func ApiSomething(next http.Handler) http.Handler { handlerFunc := func(w http.ResponseWriter, req *http.Request) { *req = *req.WithContext(context.WithValue(req.Context(), "apiData", "something")) next.ServeHTTP(w, req) } return http.HandlerFunc(handlerFunc) } func TestMain(m *testing.M) { udb := indentalUserDB.CreateIndentalUserDB("testData/userDB.ndtl") udb.AddUser("nilix", "questing") sesh, _ := udb.InitiateSession("nilix", "questing", 60) fmt.Printf("%s // %s\n", sesh, sesh) rtr := &router.Router{ StaticPaths: map[string]string{ "/static": "testData/static", }, Fallback: *template.Must(template.ParseFiles("testData/templates/error.html", "testData/templates/footer.html")), } rtr.Get("/login", renderer.Template( "testData/templates/login.html")) rtr.Post("/login", middleware.Authorize("/", udb, "/login?tryagain=1", 120)) rtr.Post("/provision", middleware.Provision(udb, 60)) rtr.Get("/protected", middleware.Validate(renderer.Template("testData/templates/test.html"), udb, map[string]string{})) rtr.Get("/", middleware.Protected( renderer.Template( "testData/templates/test.html"), http.MethodGet, udb, "/login")) rtr.Get("/json", ApiSomething(renderer.JSON("apiData"))) rtr.Get(`/thing/(?P\w+)`, renderer.Template("testData/templates/paramTest.html")) http.ListenAndServe(":8080", rtr) }