feat: add procedure to relay

api/libraries/server/context.ts

@@ -1,4 +1,6 @@
-import { ServerContext, UnauthorizedError } from "@spec/relay";
+import { ServerContext } from "@spec/relay";
+
+import type { Sockets } from "~libraries/socket/sockets.ts";
 
 import { Session } from "../auth/auth.ts";
 import { req } from "./request.ts";
@@ -11,15 +13,25 @@ declare module "@spec/relay" {
     request: Request;
 
     /**
-     * Get request session instance.
+     * Is the request authenticated.
      */
-    session: Session;
+    isAuthenticated: boolean;
 
     /**
      * Get account id from session, throws an error if the request
      * does not have a valid session.
      */
     accountId: string;
+
+    /**
+     * Get request session instance.
+     */
+    session: Session;
+
+    /**
+     * Sockets instance attached to the server.
+     */
+    sockets: Sockets;
   }
 }
 
@@ -27,15 +39,20 @@ export function getRequestContext(request: Request): ServerContext {
   return {
     request,
 
-    get session(): Session {
-      if (req.session === undefined) {
-        throw new UnauthorizedError();
-      }
-      return req.session;
+    get isAuthenticated(): boolean {
+      return req.isAuthenticated;
     },
 
     get accountId() {
       return this.session.accountId;
     },
+
+    get session(): Session {
+      return req.session;
+    },
+
+    get sockets(): Sockets {
+      return req.sockets;
+    },
   };
 }

api/libraries/server/request.ts

@@ -1,3 +1,5 @@
+import { InternalServerError, UnauthorizedError } from "@spec/relay";
+
 import { Session } from "../auth/auth.ts";
 import { asyncLocalStorage } from "./storage.ts";
 
@@ -5,21 +7,16 @@ export const req = {
   get store() {
     const store = asyncLocalStorage.getStore();
     if (store === undefined) {
-      throw new Error("Request > AsyncLocalStorage not defined.");
+      throw new InternalServerError("AsyncLocalStorage not defined.");
     }
     return store;
   },
 
-  get socket() {
-    return this.store.socket;
-  },
-
-  /**
-   * Get store that is potentially undefined.
-   * Typically used when utility functions might run in and out of request scope.
-   */
-  get unsafeStore() {
-    return asyncLocalStorage.getStore();
+  get sockets() {
+    if (this.store.sockets === undefined) {
+      throw new InternalServerError("Sockets not defined.");
+    }
+    return this.store.sockets;
   },
 
   /**
@@ -32,7 +29,10 @@ export const req = {
   /**
    * Get current session.
    */
-  get session(): Session | undefined {
+  get session(): Session {
+    if (this.store.session === undefined) {
+      throw new UnauthorizedError();
+    }
     return this.store.session;
   },
 
@@ -44,14 +44,18 @@ export const req = {
   },
 
   /**
-   * Sends a JSON-RPC 2.0 notification to the request if sent through a
-   * WebSocket connection.
-   *
-   * @param method - Method to send notification to.
-   * @param params - Params to pass to the method.
+   * Get current session.
    */
-  notify(method: string, params: any): void {
-    this.socket?.send(JSON.stringify({ jsonrpc: "2.0", method, params }));
+  getSession(): Session | undefined {
+    return this.store.session;
+  },
+
+  /**
+   * Get store that is potentially undefined.
+   * Typically used when utility functions might run in and out of request scope.
+   */
+  getStore() {
+    return asyncLocalStorage.getStore();
   },
 } as const;
 

api/libraries/server/storage.ts

@@ -1,6 +1,7 @@
 import { AsyncLocalStorage } from "node:async_hooks";
 
 import type { Session } from "~libraries/auth/mod.ts";
+import type { Sockets } from "~libraries/socket/sockets.ts";
 
 export const asyncLocalStorage = new AsyncLocalStorage<{
   session?: Session;
@@ -9,7 +10,7 @@ export const asyncLocalStorage = new AsyncLocalStorage<{
     start: number;
     end?: number;
   };
-  socket?: WebSocket;
+  sockets?: Sockets;
   response: {
     headers: Headers;
   };