78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package handlers
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"html/template"
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/gorilla/sessions"
|
|
)
|
|
|
|
type PageData struct {
|
|
Title string
|
|
Navbar string
|
|
LoggedIn bool
|
|
ShowCookieBanner bool
|
|
BasePath string
|
|
StaticPath string
|
|
CurrentURL string
|
|
}
|
|
|
|
type Config struct {
|
|
DomainName string `json:"domain_name"`
|
|
ThreadrDir string `json:"threadr_dir"`
|
|
DBUsername string `json:"db_username"`
|
|
DBPassword string `json:"db_password"`
|
|
DBDatabase string `json:"db_database"`
|
|
DBServerHost string `json:"db_svr_host"`
|
|
FileStorageDir string `json:"file_storage_dir"`
|
|
}
|
|
|
|
type App struct {
|
|
DB *sql.DB
|
|
Store *sessions.CookieStore
|
|
Config *Config
|
|
Tmpl *template.Template
|
|
}
|
|
|
|
func (app *App) SessionMW(next http.HandlerFunc) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
session, err := app.Store.Get(r, "session-name")
|
|
if err != nil {
|
|
session = sessions.NewSession(app.Store, "session-name")
|
|
session.Options = &sessions.Options{
|
|
Path: "/",
|
|
MaxAge: 86400 * 30, // 30 days
|
|
HttpOnly: true,
|
|
}
|
|
}
|
|
|
|
ctx := context.WithValue(r.Context(), "session", session)
|
|
r = r.WithContext(ctx)
|
|
|
|
next(w, r)
|
|
|
|
if err := session.Save(r, w); err != nil {
|
|
/*
|
|
Ok, so here's the thing
|
|
Errors coming from this function here "can" be ignored.
|
|
They mostly come from errors while setting cookies, so in some
|
|
environments this will trigger a lot, but they are harmless.
|
|
*/
|
|
log.Printf("Error saving session in SessionMW: %v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (app *App) RequireLoginMW(next http.HandlerFunc) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
session := r.Context().Value("session").(*sessions.Session)
|
|
if _, ok := session.Values["user_id"].(int); !ok {
|
|
http.Redirect(w, r, app.Config.ThreadrDir+"/login/?error=session", http.StatusFound)
|
|
return
|
|
}
|
|
next(w, r)
|
|
}
|
|
} |