diff options
Diffstat (limited to 'backend/router.go')
| -rw-r--r-- | backend/router.go | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/backend/router.go b/backend/router.go index b7403cf..7f9de5c 100644 --- a/backend/router.go +++ b/backend/router.go @@ -2,6 +2,7 @@ package backend import ( "context" + "database/sql" "embed" "fmt" "io/fs" @@ -45,7 +46,7 @@ func SetupLogger(debug bool) { slog.SetDefault(logger) } -func NewRouter(debug bool, cfg *Config, assets fs.FS) *chi.Mux { +func NewRouter(debug bool, cfg *Config, db *sql.DB, assets fs.FS) *chi.Mux { r := chi.NewRouter() logLevel := slog.LevelWarn @@ -70,7 +71,7 @@ func NewRouter(debug bool, cfg *Config, assets fs.FS) *chi.Mux { r.Use(func(next http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { // prevent tracking - w.Header().Add("Referrer-Policy", "no-referrer") + w.Header().Add("Referrer-Policy", "same-origin") // prevent iframe w.Header().Add("X-Frame-Options", "deny") // prevent bad content being parsed @@ -94,6 +95,15 @@ func NewRouter(debug bool, cfg *Config, assets fs.FS) *chi.Mux { next.ServeHTTP(w, r.WithContext(ctx)) }) }) + // stats + r.Use(func(next http.Handler) http.Handler { + return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + if err := UpdateStats(r.Context(), db, r); err != nil { + slog.Error("updating stats", "error", err) + } + next.ServeHTTP(w, r) + }) + }) return r } |