Release catalyst

database/busdb/context.go

@@ -0,0 +1,34 @@
+package busdb
+
+import (
+	"context"
+
+	"github.com/gin-gonic/gin"
+
+	"github.com/SecurityBrewery/catalyst/generated/models"
+	"github.com/SecurityBrewery/catalyst/role"
+)
+
+const (
+	userContextKey  = "user"
+	groupContextKey = "groups"
+)
+
+func SetContext(ctx *gin.Context, user *models.UserResponse) {
+	user.Roles = role.Strings(role.Explodes(user.Roles))
+	ctx.Set(userContextKey, user)
+}
+
+func SetGroupContext(ctx *gin.Context, groups []string) {
+	ctx.Set(groupContextKey, groups)
+}
+
+func UserContext(ctx context.Context, user *models.UserResponse) context.Context {
+	user.Roles = role.Strings(role.Explodes(user.Roles))
+	return context.WithValue(ctx, userContextKey, user)
+}
+
+func UserFromContext(ctx context.Context) (*models.UserResponse, bool) {
+	u, ok := ctx.Value(userContextKey).(*models.UserResponse)
+	return u, ok
+}