51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package server
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"dynatron.me/x/stillbox/pkg/gordio/database"
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/chi/v5/middleware"
|
|
"github.com/go-chi/httprate"
|
|
"github.com/go-chi/render"
|
|
)
|
|
|
|
func (s *Server) setupRoutes() {
|
|
r := s.r
|
|
r.Use(middleware.WithValue(database.DBCTXKeyValue, s.db))
|
|
|
|
r.Group(func(r chi.Router) {
|
|
// authenticated routes
|
|
s.auth.InstallVerifyMiddleware(r)
|
|
s.auth.InstallAuthMiddleware(r)
|
|
})
|
|
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(rateLimiter())
|
|
r.Use(render.SetContentType(render.ContentTypeJSON))
|
|
// public routes
|
|
s.auth.InstallRoutes(r)
|
|
s.sources.InstallPublicRoutes(r)
|
|
})
|
|
|
|
r.Group(func(r chi.Router) {
|
|
r.Use(rateLimiter())
|
|
s.auth.InstallVerifyMiddleware(r)
|
|
|
|
// optional auth routes
|
|
|
|
r.Get("/", s.routeIndex)
|
|
})
|
|
}
|
|
|
|
func rateLimiter() func(http.Handler) http.Handler {
|
|
return httprate.LimitByRealIP(100, 1*time.Minute)
|
|
}
|
|
|
|
func (s *Server) routeIndex(w http.ResponseWriter, r *http.Request) {
|
|
if cl, authenticated := s.auth.Authenticated(r); authenticated {
|
|
w.Write([]byte("Hello " + cl["user"].(string) + "\n"))
|
|
}
|
|
w.Write([]byte("Welcome to gordio\n"))
|
|
}
|