pkg/utils/user.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

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())
}