package app

import (
	"git.optclblast.xyz/draincloud/draincloud-core/internal/reqcontext"
	"git.optclblast.xyz/draincloud/draincloud-core/internal/storage/models"
	"github.com/gin-gonic/gin"
)

type authorizer interface {
	authorize(ctx *gin.Context) (*models.Session, error)
}

func WithAuth(handler gin.HandlerFunc, auth authorizer) gin.HandlerFunc {
	return func(ctx *gin.Context) {
		sess, err := auth.authorize(ctx)
		if err != nil {
			writeError(ctx, err)
			ctx.Abort()
			return
		}

		authCtx := reqcontext.WithSession(ctx.Request.Context(), sess)
		authCtx = reqcontext.WithUserID(authCtx, sess.UserID)
		ctx.Request = ctx.Request.WithContext(authCtx)

		handler(ctx)
	}
}