package utils

import (
	"context"

	"github.com/gorilla/sessions"
	"github.com/labstack/echo-contrib/session"
	"github.com/labstack/echo/v4"
	"go.neonxp.ru/framework/pkg/middleware"
	"go.neonxp.ru/framework/pkg/model"
)

func GetUserCtx(ctx context.Context) *model.User {
	u := ctx.Value(middleware.ContextKey("user"))
	if u == nil {
		return nil
	}

	if u, ok := u.(model.User); ok {
		return &u
	}

	return nil
}

func SetUser(c echo.Context, u *model.User, maxage int) error {
	sess, err := session.Get("user", c)
	if err != nil {
		return err
	}

	sess.Values["user"] = u
	sess.Options = &sessions.Options{
		Path:     "/",
		MaxAge:   maxage,
		HttpOnly: true,
		Secure:   true,
	}

	return sess.Save(c.Request(), c.Response())
}