update invitations in real-time

Author: nikgraf

apps/events/src/routes/playground.tsx

@@ -270,6 +270,12 @@ const App = ({
           }
           const message: EventMessage = { type: 'event', event: spaceEvent.value, spaceId: invitation.spaceId };
           websocketConnection?.send(JSON.stringify(message));
+
+          // temporary until we have define a strategy for accepting invitations response
+          setTimeout(() => {
+            const message2: RequestListInvitations = { type: 'list-invitations' };
+            websocketConnection?.send(JSON.stringify(message2));
+          }, 1000);
         }}
       />
       <h2 className="text-lg">Spaces</h2>

apps/server/src/index.ts

@@ -1,20 +1,23 @@
 import cors from 'cors';
 import 'dotenv/config';
-import { parse } from 'node:url';
 import { Effect, Exit, Schema } from 'effect';
 import express from 'express';
 import type { ResponseListInvitations, ResponseListSpaces, ResponseSpace } from 'graph-framework-messages';
 import { RequestMessage } from 'graph-framework-messages';
 import { applyEvent } from 'graph-framework-space-events';
-import type WebSocket from 'ws';
-import { WebSocketServer } from 'ws';
+import { parse } from 'node:url';
+import WebSocket, { WebSocketServer } from 'ws';
 import { applySpaceEvent } from './handlers/applySpaceEvent.js';
 import { createSpace } from './handlers/createSpace.js';
 import { getSpace } from './handlers/getSpace.js';
 import { listInvitations } from './handlers/listInvitations.js';
 import { listSpaces } from './handlers/listSpaces.js';
 import { tmpInitAccount } from './handlers/tmpInitAccount.js';
 import { assertExhaustive } from './utils/assertExhaustive.js';
+interface CustomWebSocket extends WebSocket {
+  accountId: string;
+  subscribedSpaces: Set<string>;
+}
 
 const decodeRequestMessage = Schema.decodeUnknownEither(RequestMessage);
 
@@ -38,13 +41,15 @@ const server = app.listen(PORT, () => {
   console.log(`Listening on port ${PORT}`);
 });
 
-webSocketServer.on('connection', async (webSocket: WebSocket, request: Request) => {
+webSocketServer.on('connection', async (webSocket: CustomWebSocket, request: Request) => {
   const params = parse(request.url, true);
   if (!params.query.accountId || typeof params.query.accountId !== 'string') {
     webSocket.close();
     return;
   }
   const accountId = params.query.accountId;
+  webSocket.accountId = accountId;
+  webSocket.subscribedSpaces = new Set();
 
   console.log('Connection established', accountId);
   webSocket.on('message', async (message) => {
@@ -101,6 +106,18 @@ webSocketServer.on('connection', async (webSocket: WebSocket, request: Request)
             type: 'space',
           };
           webSocket.send(JSON.stringify(outgoingMessage));
+          for (const client of webSocketServer.clients as Set<CustomWebSocket>) {
+            if (
+              client.readyState === WebSocket.OPEN &&
+              client.accountId === data.event.transaction.signaturePublicKey
+            ) {
+              const invitations = await listInvitations({ accountId: client.accountId });
+              const outgoingMessage: ResponseListInvitations = { type: 'list-invitations', invitations: invitations };
+              // for now sending the entire list of invitations to the client - we could send only a single one
+              client.send(JSON.stringify(outgoingMessage));
+            }
+          }
+
           break;
         }
         case 'event': {