From 16689f6c312d55af06c5197552b46675f1905b26 Mon Sep 17 00:00:00 2001
From: Derrick Hammer <derrick@derrickhammer.com>
Date: Tue, 13 Feb 2024 23:22:36 -0500
Subject: [PATCH] feat: add GetUserFromContext

---
 api/middleware/middleware.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/api/middleware/middleware.go b/api/middleware/middleware.go
index 2281bc6..3036c27 100644
--- a/api/middleware/middleware.go
+++ b/api/middleware/middleware.go
@@ -156,3 +156,23 @@ func AuthMiddleware(options AuthMiddlewareOptions) func(http.Handler) http.Handl
 		})
 	}
 }
+
+func GetUserFromContext(ctx context.Context, key ...string) uint {
+	realKey := ""
+
+	if len(key) > 0 {
+		realKey = key[0]
+	}
+
+	if realKey == "" {
+		realKey = DEFAULT_AUTH_CONTEXT_KEY
+	}
+
+	userId, ok := ctx.Value(key).(uint)
+
+	if !ok {
+		panic("user id stored in context is not of type uint")
+	}
+
+	return userId
+}