70 lines
2.1 KiB
Go
70 lines
2.1 KiB
Go
package handlers
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"html/template"
|
|
"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"`
|
|
}
|
|
|
|
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")
|
|
}
|
|
if _, ok := session.Values["user_id"].(int); ok {
|
|
if session.Values["user_ip"] != r.RemoteAddr || session.Values["user_agent"] != r.UserAgent() {
|
|
session.Values = make(map[interface{}]interface{})
|
|
session.Options.MaxAge = -1
|
|
session.Save(r, w)
|
|
http.Redirect(w, r, app.Config.ThreadrDir+"/login/?error=session", http.StatusFound)
|
|
return
|
|
}
|
|
ctx := context.WithValue(r.Context(), "session", session)
|
|
r = r.WithContext(ctx)
|
|
} else {
|
|
ctx := context.WithValue(r.Context(), "session", session)
|
|
r = r.WithContext(ctx)
|
|
}
|
|
next(w, r)
|
|
}
|
|
}
|
|
|
|
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)
|
|
}
|
|
} |