feat: add actual user for trpc wss-dev-server (#261)

* feat: add actual user for trpc wss-dev-server #233

* chore: address pull request feedback

* fix: deepsource issue

packages/api/src/wssDevServer.ts

@@ -1,6 +1,9 @@
 import { applyWSSHandler } from "@trpc/server/adapters/ws";
 import { WebSocketServer } from "ws";
 
+import { getSessionFromToken, sessionTokenCookieName } from "@homarr/auth";
+import { parseCookies } from "@homarr/common";
+import { db } from "@homarr/db";
 import { logger } from "@homarr/log";
 
 import { appRouter } from "./root";
@@ -12,33 +15,41 @@ const wss = new WebSocketServer({
 const handler = applyWSSHandler({
   wss,
   router: appRouter,
-  createContext: ({ req }) => {
-    return createTRPCContext({
-      headers: {
-        ...req.headers,
-        get(key: string) {
-          const item = req.headers[key];
-          return typeof item === "string" ? item ?? null : item?.at(0) ?? null;
-        },
-      } as Headers,
-      session: {
-        // TODO: replace with actual session
-        user: {
-          id: "1",
-          name: "Test User",
-          email: "",
-        },
-        expires: new Date().toISOString(),
-      },
-    });
+  createContext: async ({ req }) => {
+    try {
+      const headers = Object.entries(req.headers).map(
+        ([key, value]) =>
+          [key, typeof value === "string" ? value : value?.[0]] as [
+            string,
+            string,
+          ],
+      );
+      const nextHeaders = new Headers(headers);
+
+      const store = parseCookies(nextHeaders.get("cookie") ?? "");
+      const sessionToken = store[sessionTokenCookieName];
+
+      const session = await getSessionFromToken(db, sessionToken);
+
+      return createTRPCContext({
+        headers: nextHeaders,
+        session,
+      });
+    } catch (error) {
+      logger.error(error);
+      return createTRPCContext({
+        headers: new Headers(),
+        session: null,
+      });
+    }
   },
 });
 
-wss.on("connection", (ws, incomingMessage) => {
+wss.on("connection", (websocket, incomingMessage) => {
   logger.info(
     `➕ Connection (${wss.clients.size}) ${incomingMessage.method} ${incomingMessage.url}`,
   );
-  ws.once("close", (code, reason) => {
+  websocket.once("close", (code, reason) => {
     logger.info(
       `➖ Connection (${wss.clients.size}) ${code} ${reason.toString()}`,
     );