initialv0.0.1

1 files changed, 34 insertions, 0 deletions

diff --git a/middleware/user.go b/middleware/user.go
new file mode 100644
index 0000000..eaca6e9
--- /dev/null
+++ b/middleware/user.go
@@ -0,0 +1,34 @@
+package middleware
+
+import (
+	"context"
+
+	"github.com/labstack/echo-contrib/session"
+	"github.com/labstack/echo/v4"
+	"gitrepo.ru/neonxp/gorum/contextlib"
+	"gitrepo.ru/neonxp/gorum/models"
+)
+
+func UserMiddleware() echo.MiddlewareFunc {
+	return func(next echo.HandlerFunc) echo.HandlerFunc {
+		return func(c echo.Context) error {
+			sess, err := session.Get("session", c)
+			if err != nil {
+				return err
+			}
+			u, okSess := sess.Values["user"]
+			if !okSess {
+				return next(c)
+			}
+			user, okUser := u.(models.User)
+			if !okUser {
+				return next(c)
+			}
+			ctx := context.WithValue(c.Request().Context(), contextlib.UserKey, user)
+			req := c.Request().WithContext(ctx)
+			c.SetRequest(req)
+
+			return next(c)
+		}
+	}
+}