106 lines
2.2 KiB
Go
106 lines
2.2 KiB
Go
package httpx
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
type LogWrapper struct {
|
|
statusCode *int
|
|
|
|
http.ResponseWriter
|
|
}
|
|
|
|
func (w LogWrapper) WriteHeader(statusCode int) {
|
|
*w.statusCode = statusCode
|
|
|
|
w.ResponseWriter.WriteHeader(statusCode)
|
|
}
|
|
|
|
func Log(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
statusCode := 200
|
|
wrapper := LogWrapper{statusCode: &statusCode, ResponseWriter: w}
|
|
start := time.Now()
|
|
next.ServeHTTP(wrapper, r)
|
|
|
|
log.Println(r.Method, r.URL.Path, statusCode, time.Since(start))
|
|
})
|
|
}
|
|
|
|
type AuthInfo struct {
|
|
IsAdmin bool
|
|
SessionId string
|
|
}
|
|
|
|
func queryCheckAuth(db *sql.DB, sessionId string) (AuthInfo, error) {
|
|
sessionCount := 0
|
|
err := db.QueryRow("select count(*) from session where session_id = ?", sessionId).Scan(&sessionCount)
|
|
if err != nil {
|
|
return AuthInfo{}, err
|
|
}
|
|
|
|
return AuthInfo{IsAdmin: sessionCount == 1, SessionId: sessionId}, nil
|
|
}
|
|
|
|
func checkAuthed(db *sql.DB, r *http.Request) (AuthInfo, error) {
|
|
cookie, err := r.Cookie("SHELVES_OWNER_SESSION_ID")
|
|
if err == http.ErrNoCookie {
|
|
return AuthInfo{}, nil
|
|
}
|
|
if err != nil {
|
|
return AuthInfo{}, err
|
|
}
|
|
|
|
sessionId := cookie.Value
|
|
return queryCheckAuth(db, sessionId)
|
|
}
|
|
|
|
func queryNeedsOwnerSetup(db *sql.DB) (bool, error) {
|
|
count := 0
|
|
err := db.QueryRow("select count(*) from owner_settings").Scan(&count)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
return count != 1, nil
|
|
}
|
|
|
|
type Ctx struct {
|
|
DB *sql.DB
|
|
Auth AuthInfo
|
|
NeedsOwnerSetup bool
|
|
}
|
|
|
|
func GetCtx(r *http.Request) Ctx {
|
|
ctx := r.Context()
|
|
|
|
return ctx.Value("__ctx").(Ctx)
|
|
}
|
|
|
|
func WithCtx(db *sql.DB, next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
needsOwnerSetup, err := queryNeedsOwnerSetup(db)
|
|
if err != nil {
|
|
log.Printf("Error while querying owner_settings: %v\n", err)
|
|
}
|
|
|
|
ctx := r.Context()
|
|
auth, err := checkAuthed(db, r)
|
|
if err != nil {
|
|
log.Printf("Error while querying auth info: %v\n", err)
|
|
}
|
|
ctx = context.WithValue(ctx, "__ctx", Ctx{
|
|
DB: db,
|
|
Auth: auth,
|
|
NeedsOwnerSetup: needsOwnerSetup,
|
|
})
|
|
r = r.WithContext(ctx)
|
|
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|