document-drive 1.8.4 → 1.9.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "document-drive",
-  "version": "1.8.4",
+  "version": "1.9.0",
   "license": "AGPL-3.0-only",
   "type": "module",
   "module": "./src/index.ts",
@@ -33,9 +33,9 @@
     "nanoevents": "^9.0.0",
     "sanitize-filename": "^1.6.3",
     "uuid": "^9.0.1",
-    "document-model": "2.10.0",
-    "@powerhousedao/scalars": "1.12.0",
-    "document-model-libs": "1.120.3"
+    "@powerhousedao/scalars": "1.13.0",
+    "document-model": "2.11.0",
+    "document-model-libs": "1.121.0"
   },
   "optionalDependencies": {
     "@prisma/client": "^5.18.0",
@@ -44,7 +44,7 @@
     "redis": "^4.6.15",
     "sequelize": "^6.37.3",
     "sqlite3": "^5.1.7",
-    "@powerhousedao/scalars": "1.12.0"
+    "@powerhousedao/scalars": "1.13.0"
   },
   "devDependencies": {
     "@prisma/client": "5.17.0",
@@ -66,8 +66,8 @@
     "nanoevents": "^9.0.0",
     "sanitize-filename": "^1.6.3",
     "uuid": "^9.0.1",
-    "document-model": "2.10.0",
-    "document-model-libs": "1.120.3"
+    "document-model": "2.11.0",
+    "document-model-libs": "1.121.0"
   },
   "scripts": {
     "check-types": "tsc --build",
@@ -79,6 +79,6 @@
     "test:watch": "vitest watch",
     "clean": "rimraf dist",
     "clean:node_modules": "rimraf node_modules",
-    "build": "prisma generate"
+    "postinstall": "prisma generate"
   }
 }

package/src/index.ts

@@ -2,5 +2,3 @@ export * from "./server";
 export * from "./server/error";
 export * from "./storage";
 export * from "./utils";
-
-export const test = "test";

package/src/server/listener/manager.ts

@@ -102,6 +102,8 @@ export class ListenerManager extends BaseListenerManager {
     const driveTransmitters = this.transmitters[drive] || {};
     driveTransmitters[listener.listenerId] = transmitter;
     this.transmitters[drive] = driveTransmitters;
+
+    this.triggerUpdate(true, { type: "local" });
     return Promise.resolve(transmitter);
   }
 

package/src/server/listener/transmitter/internal.ts

@@ -1,30 +1,40 @@
-import { Document, OperationScope } from "document-model/document";
+import { Document, Operation, OperationScope } from "document-model/document";
 import { logger } from "../../../utils/logger";
 import {
   IBaseDocumentDriveServer,
   Listener,
   ListenerRevision,
-  OperationUpdate,
   StrandUpdate,
 } from "../../types";
-import { buildRevisionsFilter } from "../../utils";
-import { ITransmitter } from "./types";
+import { ITransmitter, StrandUpdateSource } from "./types";
+import { InferDocumentOperation } from "../../../read-mode/types";
 
-export interface IReceiver {
-  transmit: (strands: InternalTransmitterUpdate[]) => Promise<void>;
-  disconnect: () => Promise<void>;
+export interface IReceiver<
+  T extends Document = Document,
+  S extends OperationScope = OperationScope,
+> {
+  onStrands: (strands: InternalTransmitterUpdate<T, S>[]) => Promise<void>;
+  onDisconnect: () => Promise<void>;
 }
 
+export type InternalOperationUpdate<
+  D extends Document = Document,
+  S extends OperationScope = OperationScope,
+> = Omit<Operation<InferDocumentOperation<D>>, "scope"> & {
+  state: D["state"][S];
+  previousState: D["state"][S];
+};
+
 export type InternalTransmitterUpdate<
-  T extends Document = Document,
+  D extends Document = Document,
   S extends OperationScope = OperationScope,
 > = {
   driveId: string;
   documentId: string;
   scope: S;
   branch: string;
-  operations: OperationUpdate[];
-  state: T["state"][S];
+  operations: InternalOperationUpdate<D, S>[];
+  state: D["state"][S];
 };
 
 export interface IInternalTransmitter extends ITransmitter {
@@ -32,52 +42,75 @@ export interface IInternalTransmitter extends ITransmitter {
 }
 
 export class InternalTransmitter implements ITransmitter {
-  private drive: IBaseDocumentDriveServer;
-  private listener: Listener;
-  private receiver: IReceiver | undefined;
+  protected drive: IBaseDocumentDriveServer;
+  protected listener: Listener;
+  protected receiver: IReceiver | undefined;
 
   constructor(listener: Listener, drive: IBaseDocumentDriveServer) {
     this.listener = listener;
     this.drive = drive;
   }
 
+  async #buildInternalOperationUpdate(strand: StrandUpdate) {
+    const operations: InternalOperationUpdate[] = [];
+    const stateByIndex = new Map<number, unknown>();
+    const getStateByIndex = async (index: number) => {
+      const state = stateByIndex.get(index);
+      if (state) {
+        return state;
+      }
+      const document = await this.drive.getDocument(
+        strand.driveId,
+        strand.documentId,
+        {
+          revisions: {
+            [strand.scope]: index,
+          },
+          checkHashes: false,
+        },
+      );
+      if (index < 0) {
+        stateByIndex.set(index, document.initialState.state[strand.scope]);
+      } else {
+        stateByIndex.set(index, document.state[strand.scope]);
+      }
+      return stateByIndex.get(index);
+    };
+    for (const operation of strand.operations) {
+      operations.push({
+        ...operation,
+        state: await getStateByIndex(operation.index),
+        previousState: await getStateByIndex(operation.index - 1),
+      });
+    }
+    return operations;
+  }
+
   async transmit(
-    strands: InternalTransmitterUpdate[],
+    strands: StrandUpdate[],
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
+    _source: StrandUpdateSource,
   ): Promise<ListenerRevision[]> {
     if (!this.receiver) {
       return [];
     }
 
-    const retrievedDocuments = new Map<string, Document>();
     const updates: InternalTransmitterUpdate[] = [];
     for (const strand of strands) {
-      let document = retrievedDocuments.get(
-        `${strand.driveId}:${strand.documentId}`,
-      );
-      if (!document) {
-        const revisions = buildRevisionsFilter(
-          strands,
-          strand.driveId,
-          strand.documentId,
-        );
-        document = await (strand.documentId
-          ? this.drive.getDocument(strand.driveId, strand.documentId, {
-              revisions,
-            })
-          : this.drive.getDrive(strand.driveId, { revisions }));
-        retrievedDocuments.set(
-          `${strand.driveId}:${strand.documentId}`,
-          document,
-        );
-      }
-      updates.push({ ...strand, state: document.state[strand.scope] });
+      const operations = await this.#buildInternalOperationUpdate(strand);
+      const state = operations.at(-1)?.state ?? {};
+      updates.push({
+        ...strand,
+        operations,
+        state,
+      });
     }
     try {
-      await this.receiver.transmit(updates);
+      await this.receiver.onStrands(updates);
       return strands.map(({ operations, ...s }) => ({
         ...s,
         status: "SUCCESS",
-        revision: operations[operations.length - 1]?.index ?? -1,
+        revision: operations.at(operations.length - 1)?.index ?? -1,
       }));
     } catch (error) {
       logger.error(error);
@@ -85,7 +118,7 @@ export class InternalTransmitter implements ITransmitter {
       return strands.map(({ operations, ...s }) => ({
         ...s,
         status: "ERROR",
-        revision: (operations[0]?.index ?? 0) - 1,
+        revision: (operations.at(0)?.index ?? 0) - 1,
       }));
     }
   }
@@ -95,6 +128,10 @@ export class InternalTransmitter implements ITransmitter {
   }
 
   async disconnect(): Promise<void> {
-    await this.receiver?.disconnect();
+    await this.receiver?.onDisconnect();
+  }
+
+  getListener(): Listener {
+    return this.listener;
   }
 }