optional tenant gate and event stream in options

decentralized-identity · Feb 12, 2024 · 8d20cd6 · 8d20cd6
1 parent c93180a
commit 8d20cd6
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 9 deletions.
diff --git a/src/dwn-server.ts b/src/dwn-server.ts
@@ -1,4 +1,5 @@
-import { Dwn } from '@tbd54566975/dwn-sdk-js';
+import type { EventStream } from '@tbd54566975/dwn-sdk-js';
+import { Dwn, EventEmitterStream } from '@tbd54566975/dwn-sdk-js';
 
 import type { Server } from 'http';
 import log from 'loglevel';
@@ -61,7 +62,17 @@ export class DwnServer {
         proofOfWorkInitialMaximumAllowedHash: this.config.registrationProofOfWorkInitialMaxHash,
       });
 
-      this.dwn = await Dwn.create(getDWNConfig(this.config, registrationManager));
+      let eventStream: EventStream | undefined;
+      if (this.config.webSocketServerEnabled) {
+        // setting `EventEmitterStream` as default the default `EventStream
+        // if an alternate implementation is needed instantiate a `Dwn` with a custom `EventStream` and add it to server options. 
+        eventStream = new EventEmitterStream();
+      }
+
+      this.dwn = await Dwn.create(getDWNConfig(this.config, {
+        tenantGate: registrationManager,
+        eventStream,
+      }));
     }
 
     this.#httpApi = new HttpApi(this.config, this.dwn, registrationManager);
@@ -76,8 +87,7 @@ export class DwnServer {
 
     if (this.config.webSocketServerEnabled) {
       this.#wsApi = new WsApi(this.#httpApi.server, this.dwn);
-      this.#wsApi.start();
-      log.info('WebSocketServer ready...');
+      this.#wsApi.start(() => log.info('WebSocketServer ready...'));
     }
   }
 

diff --git a/src/storage.ts b/src/storage.ts
@@ -2,14 +2,14 @@ import * as fs from 'fs';
 
 import {
   DataStoreLevel,
-  EventEmitterStream,
   EventLogLevel,
   MessageStoreLevel,
 } from '@tbd54566975/dwn-sdk-js';
 import type {
   DataStore,
   DwnConfig,
   EventLog,
+  EventStream,
   MessageStore,
   TenantGate,
 } from '@tbd54566975/dwn-sdk-js';
@@ -46,17 +46,20 @@ export enum BackendTypes {
 export type StoreType = DataStore | EventLog | MessageStore;
 
 export function getDWNConfig(
-  config: DwnServerConfig,
-  tenantGate: TenantGate,
+  config  : DwnServerConfig,
+  options : {
+    tenantGate?  : TenantGate,
+    eventStream? : EventStream,
+  }
 ): DwnConfig {
+  const { tenantGate, eventStream } = options;
   const dataStore: DataStore = getStore(config.dataStore, EStoreType.DataStore);
   const eventLog: EventLog = getStore(config.eventLog, EStoreType.EventLog);
   const messageStore: MessageStore = getStore(
     config.messageStore,
     EStoreType.MessageStore,
   );
 
-  const eventStream = config.webSocketServerEnabled ? new EventEmitterStream() : undefined;
   return { eventStream, eventLog, dataStore, messageStore, tenantGate };
 }
 

diff --git a/src/ws-api.ts b/src/ws-api.ts
@@ -268,8 +268,9 @@ export class WsApi {
     });
   }
 
-  start(): WebSocketServer {
+  start(callback?: () => void): WebSocketServer {
     this.#setupWebSocket();
+    callback?.();
     return this.#wsServer;
   }