cojson-storage-indexeddb 0.6.0 → 0.6.1

package/src/index.ts

@@ -13,6 +13,7 @@ import {
     ReadableStreamDefaultReader,
     WritableStreamDefaultWriter,
 } from "isomorphic-streams";
+import { SyncPromise } from "./syncPromises";
 
 type CoValueRow = {
     id: CojsonInternalTypes.RawCoID;
@@ -64,12 +65,17 @@ export class IDBStorage {
                 done = result.done;
 
                 if (result.value) {
-                    await this.handleSyncMessage(result.value);
                     // console.log(
                     //     "IDB: handling msg",
                     //     result.value.id,
                     //     result.value.action
                     // );
+                    await this.handleSyncMessage(result.value);
+                    // console.log(
+                    //     "IDB: handled msg",
+                    //     result.value.id,
+                    //     result.value.action
+                    // );
                 }
             }
         })();
@@ -144,23 +150,23 @@ export class IDBStorage {
                         keyPath: ["ses", "idx"],
                     });
                 }
-                if (ev.oldVersion !== 0 && ev.oldVersion <= 3) {
-                    // fix embarrassing off-by-one error for transaction indices
-                    console.log("Migration: fixing off-by-one error");
-                    const transaction = (
-                        ev.target as unknown as { transaction: IDBTransaction }
-                    ).transaction;
-
-                    const txsStore = transaction.objectStore("transactions");
-                    const txs = await promised(txsStore.getAll());
-
-                    for (const tx of txs) {
-                        await promised(txsStore.delete([tx.ses, tx.idx]));
-                        tx.idx -= 1;
-                        await promised(txsStore.add(tx));
-                    }
-                    console.log("Migration: fixing off-by-one error - done");
-                }
+                // if (ev.oldVersion !== 0 && ev.oldVersion <= 3) {
+                //     // fix embarrassing off-by-one error for transaction indices
+                //     console.log("Migration: fixing off-by-one error");
+                //     const transaction = (
+                //         ev.target as unknown as { transaction: IDBTransaction }
+                //     ).transaction;
+
+                //     const txsStore = transaction.objectStore("transactions");
+                //     const txs = await promised(txsStore.getAll());
+
+                //     for (const tx of txs) {
+                //         await promised(txsStore.delete([tx.ses, tx.idx]));
+                //         tx.idx -= 1;
+                //         await promised(txsStore.add(tx));
+                //     }
+                //     console.log("Migration: fixing off-by-one error - done");
+                // }
             };
         });
 
@@ -170,418 +176,596 @@ export class IDBStorage {
     async handleSyncMessage(msg: SyncMessage) {
         switch (msg.action) {
             case "load":
-                await this.handleLoad(msg);
+                /*await*/ this.handleLoad(msg);
                 break;
             case "content":
                 await this.handleContent(msg);
                 break;
             case "known":
-                await this.handleKnown(msg);
+                /*await*/ this.handleKnown(msg);
                 break;
             case "done":
-                await this.handleDone(msg);
+                /*await*/ this.handleDone(msg);
                 break;
         }
     }
 
-    async sendNewContentAfter(
-        theirKnown: CojsonInternalTypes.CoValueKnownState,
-        {
-            coValues,
-            sessions,
-            transactions,
-            signatureAfter,
-        }: {
+    currentTx:
+        | {
+              id: number;
+              tx: IDBTransaction;
+              stores: {
+                  coValues: IDBObjectStore;
+                  sessions: IDBObjectStore;
+                  transactions: IDBObjectStore;
+                  signatureAfter: IDBObjectStore;
+              };
+              startedAt: number;
+              pendingRequests: ((txEntry: {
+                  stores: {
+                      coValues: IDBObjectStore;
+                      sessions: IDBObjectStore;
+                      transactions: IDBObjectStore;
+                      signatureAfter: IDBObjectStore;
+                  };
+              }) => void)[];
+          }
+        | undefined;
+    currentTxID = 0;
+
+    makeRequest<T>(
+        handler: (stores: {
             coValues: IDBObjectStore;
             sessions: IDBObjectStore;
             transactions: IDBObjectStore;
             signatureAfter: IDBObjectStore;
-        },
-        asDependencyOf?: CojsonInternalTypes.RawCoID
-    ) {
-        const coValueRow = await promised<StoredCoValueRow | undefined>(
-            coValues.index("coValuesById").get(theirKnown.id)
-        );
+        }) => IDBRequest
+    ): SyncPromise<T> {
+        return new SyncPromise((resolve, reject) => {
+            let txEntry = this.currentTx;
+
+            const requestEntry = ({
+                stores,
+            }: {
+                stores: {
+                    coValues: IDBObjectStore;
+                    sessions: IDBObjectStore;
+                    transactions: IDBObjectStore;
+                    signatureAfter: IDBObjectStore;
+                };
+            }) => {
+                const request = handler(stores);
+                request.onerror = () => {
+                    console.error("Error in request", request.error);
+                    this.currentTx = undefined;
+                    reject(request.error);
+                    // TODO: recover pending requests in new tx
+                };
+                request.onsuccess = () => {
+                    const value = request.result as T;
+                    resolve(value);
 
-        const allOurSessions = coValueRow
-            ? await promised<StoredSessionRow[]>(
-                  sessions.index("sessionsByCoValue").getAll(coValueRow.rowID)
-              )
-            : [];
+                    const next = txEntry!.pendingRequests.shift();
 
-        const ourKnown: CojsonInternalTypes.CoValueKnownState = {
-            id: theirKnown.id,
-            header: !!coValueRow,
-            sessions: {},
-        };
+                    if (next) {
+                        next({ stores });
+                    } else {
+                        if (this.currentTx === txEntry) {
+                            this.currentTx = undefined;
+                        }
+                    }
+                };
+            };
 
-        const newContentPieces: CojsonInternalTypes.NewContentMessage[] = [
-            {
-                action: "content",
-                id: theirKnown.id,
-                header: theirKnown.header ? undefined : coValueRow?.header,
-                new: {},
-            },
-        ];
-
-        for (const sessionRow of allOurSessions) {
-            ourKnown.sessions[sessionRow.sessionID] = sessionRow.lastIdx;
-
-            if (
-                sessionRow.lastIdx >
-                (theirKnown.sessions[sessionRow.sessionID] || 0)
-            ) {
-                const firstNewTxIdx =
-                    theirKnown.sessions[sessionRow.sessionID] || 0;
-
-                const signaturesAndIdxs = await promised<SignatureAfterRow[]>(
-                    signatureAfter.getAll(
-                        IDBKeyRange.bound(
-                            [sessionRow.rowID, firstNewTxIdx],
-                            [sessionRow.rowID, Infinity]
-                        )
-                    )
+            if (!txEntry || performance.now() - txEntry.startedAt > 20) {
+                const tx = this.db.transaction(
+                    ["coValues", "sessions", "transactions", "signatureAfter"],
+                    "readwrite"
                 );
+                txEntry = {
+                    id: this.currentTxID++,
+                    tx,
+                    stores: {
+                        coValues: tx.objectStore("coValues"),
+                        sessions: tx.objectStore("sessions"),
+                        transactions: tx.objectStore("transactions"),
+                        signatureAfter: tx.objectStore("signatureAfter"),
+                    },
+                    startedAt: performance.now(),
+                    pendingRequests: [],
+                };
 
-                // console.log(
-                //     theirKnown.id,
-                //     "signaturesAndIdxs",
-                //     JSON.stringify(signaturesAndIdxs)
-                // );
+                console.time("IndexedDB TX" + txEntry.id);
 
-                const newTxInSession = await promised<TransactionRow[]>(
-                    transactions.getAll(
-                        IDBKeyRange.bound(
-                            [sessionRow.rowID, firstNewTxIdx],
-                            [sessionRow.rowID, Infinity]
-                        )
-                    )
-                );
+                txEntry.tx.oncomplete = () => {
+                    console.timeEnd("IndexedDB TX" + txEntry!.id);
+                };
 
-                let idx = firstNewTxIdx;
+                this.currentTx = txEntry;
 
+                requestEntry(txEntry);
+            } else {
+                txEntry.pendingRequests.push(requestEntry);
                 // console.log(
-                //     theirKnown.id,
-                //     "newTxInSession",
-                //     newTxInSession.length
+                //     "Queued request in TX " + txEntry.id,
+                //     txEntry.pendingRequests.length
                 // );
+            }
+        });
+    }
 
-                for (const tx of newTxInSession) {
-                    let sessionEntry =
-                        newContentPieces[newContentPieces.length - 1]!.new[
-                            sessionRow.sessionID
+    sendNewContentAfter(
+        theirKnown: CojsonInternalTypes.CoValueKnownState,
+        asDependencyOf?: CojsonInternalTypes.RawCoID
+    ): SyncPromise<void> {
+        return this.makeRequest<StoredCoValueRow | undefined>(({ coValues }) =>
+            coValues.index("coValuesById").get(theirKnown.id)
+        )
+            .then((coValueRow) => {
+                return (
+                    coValueRow
+                        ? this.makeRequest<StoredSessionRow[]>(({ sessions }) =>
+                              sessions
+                                  .index("sessionsByCoValue")
+                                  .getAll(coValueRow.rowID)
+                          )
+                        : SyncPromise.resolve([])
+                ).then((allOurSessions) => {
+                    const ourKnown: CojsonInternalTypes.CoValueKnownState = {
+                        id: theirKnown.id,
+                        header: !!coValueRow,
+                        sessions: {},
+                    };
+
+                    const newContentPieces: CojsonInternalTypes.NewContentMessage[] =
+                        [
+                            {
+                                action: "content",
+                                id: theirKnown.id,
+                                header: theirKnown.header
+                                    ? undefined
+                                    : coValueRow?.header,
+                                new: {},
+                            },
                         ];
-                    if (!sessionEntry) {
-                        sessionEntry = {
-                            after: idx,
-                            lastSignature:
-                                "WILL_BE_REPLACED" as CojsonInternalTypes.Signature,
-                            newTransactions: [],
-                        };
-                        newContentPieces[newContentPieces.length - 1]!.new[
-                            sessionRow.sessionID
-                        ] = sessionEntry;
-                    }
 
-                    sessionEntry.newTransactions.push(tx.tx);
-
-                    if (
-                        signaturesAndIdxs[0] &&
-                        idx === signaturesAndIdxs[0].idx
-                    ) {
-                        sessionEntry.lastSignature =
-                            signaturesAndIdxs[0].signature;
-                        signaturesAndIdxs.shift();
-                        newContentPieces.push({
-                            action: "content",
-                            id: theirKnown.id,
-                            new: {},
+                    return SyncPromise.all(
+                        allOurSessions.map((sessionRow) => {
+                            ourKnown.sessions[sessionRow.sessionID] =
+                                sessionRow.lastIdx;
+
+                            if (
+                                sessionRow.lastIdx >
+                                (theirKnown.sessions[sessionRow.sessionID] || 0)
+                            ) {
+                                const firstNewTxIdx =
+                                    theirKnown.sessions[sessionRow.sessionID] ||
+                                    0;
+
+                                return this.makeRequest<SignatureAfterRow[]>(
+                                    ({ signatureAfter }) =>
+                                        signatureAfter.getAll(
+                                            IDBKeyRange.bound(
+                                                [
+                                                    sessionRow.rowID,
+                                                    firstNewTxIdx,
+                                                ],
+                                                [sessionRow.rowID, Infinity]
+                                            )
+                                        )
+                                ).then((signaturesAndIdxs) => {
+                                    // console.log(
+                                    //     theirKnown.id,
+                                    //     "signaturesAndIdxs",
+                                    //     JSON.stringify(signaturesAndIdxs)
+                                    // );
+
+                                    return this.makeRequest<TransactionRow[]>(
+                                        ({ transactions }) =>
+                                            transactions.getAll(
+                                                IDBKeyRange.bound(
+                                                    [
+                                                        sessionRow.rowID,
+                                                        firstNewTxIdx,
+                                                    ],
+                                                    [sessionRow.rowID, Infinity]
+                                                )
+                                            )
+                                    ).then((newTxsInSession) => {
+                                        collectNewTxs(
+                                            newTxsInSession,
+                                            newContentPieces,
+                                            sessionRow,
+                                            signaturesAndIdxs,
+                                            theirKnown,
+                                            firstNewTxIdx
+                                        );
+                                    });
+                                });
+                            } else {
+                                return SyncPromise.resolve();
+                            }
+                        })
+                    ).then(() => {
+                        const dependedOnCoValues = getDependedOnCoValues(
+                            coValueRow,
+                            newContentPieces,
+                            theirKnown
+                        );
+
+                        return SyncPromise.all(
+                            dependedOnCoValues.map((dependedOnCoValue) =>
+                                this.sendNewContentAfter(
+                                    {
+                                        id: dependedOnCoValue,
+                                        header: false,
+                                        sessions: {},
+                                    },
+                                    asDependencyOf || theirKnown.id
+                                )
+                            )
+                        ).then(() => {
+                            // we're done with IndexedDB stuff here so can use native Promises again
+                            setTimeout(async () => {
+                                await this.toLocalNode.write({
+                                    action: "known",
+                                    ...ourKnown,
+                                    asDependencyOf,
+                                });
+
+                                const nonEmptyNewContentPieces =
+                                    newContentPieces.filter(
+                                        (piece) =>
+                                            piece.header ||
+                                            Object.keys(piece.new).length > 0
+                                    );
+
+                                // console.log(theirKnown.id, nonEmptyNewContentPieces);
+
+                                for (const piece of nonEmptyNewContentPieces) {
+                                    await this.toLocalNode.write(piece);
+                                    await new Promise((resolve) =>
+                                        setTimeout(resolve, 0)
+                                    );
+                                }
+                            }, 0);
+
+                            return Promise.resolve();
                         });
-                    } else if (
-                        idx ===
-                        firstNewTxIdx + newTxInSession.length - 1
-                    ) {
-                        sessionEntry.lastSignature = sessionRow.lastSignature;
-                    }
-                    idx += 1;
-                }
-            }
-        }
-
-        const dependedOnCoValues =
-            coValueRow?.header.ruleset.type === "group"
-                ? newContentPieces
-                      .flatMap((piece) => Object.values(piece.new))
-                      .flatMap((sessionEntry) =>
-                          sessionEntry.newTransactions.flatMap((tx) => {
-                              if (tx.privacy !== "trusting") return [];
-                              // TODO: avoid parse here?
-                              return cojsonInternals
-                                  .parseJSON(tx.changes)
-                                  .map(
-                                      (change) =>
-                                          change &&
-                                          typeof change === "object" &&
-                                          "op" in change &&
-                                          change.op === "set" &&
-                                          "key" in change &&
-                                          change.key
-                                  )
-                                  .filter(
-                                      (
-                                          key
-                                      ): key is CojsonInternalTypes.RawCoID =>
-                                          typeof key === "string" &&
-                                          key.startsWith("co_")
-                                  );
-                          })
-                      )
-                : coValueRow?.header.ruleset.type === "ownedByGroup"
-                ? [
-                      coValueRow?.header.ruleset.group,
-                      ...new Set(
-                          newContentPieces.flatMap((piece) =>
-                              Object.keys(piece)
-                                  .map((sessionID) =>
-                                      cojsonInternals.accountOrAgentIDfromSessionID(
-                                          sessionID as SessionID
-                                      )
-                                  )
-                                  .filter(
-                                      (accountID): accountID is AccountID =>
-                                          cojsonInternals.isAccountID(
-                                              accountID
-                                          ) && accountID !== theirKnown.id
-                                  )
-                          )
-                      ),
-                  ]
-                : [];
-
-        for (const dependedOnCoValue of dependedOnCoValues) {
-            await this.sendNewContentAfter(
-                { id: dependedOnCoValue, header: false, sessions: {} },
-                { coValues, sessions, transactions, signatureAfter },
-                asDependencyOf || theirKnown.id
-            );
-        }
+                    });
+                });
+            })
+            .then(() => {});
+    }
 
-        await this.toLocalNode.write({
-            action: "known",
-            ...ourKnown,
-            asDependencyOf,
-        });
+    handleLoad(msg: CojsonInternalTypes.LoadMessage) {
+        return this.sendNewContentAfter(msg);
+    }
 
-        const nonEmptyNewContentPieces = newContentPieces.filter(
-            (piece) => piece.header || Object.keys(piece.new).length > 0
-        );
+    handleContent(
+        msg: CojsonInternalTypes.NewContentMessage
+    ): SyncPromise<void> {
+        return this.makeRequest<StoredCoValueRow | undefined>(({ coValues }) =>
+            coValues.index("coValuesById").get(msg.id)
+        )
+            .then((coValueRow) => {
+                if (coValueRow?.rowID === undefined) {
+                    const header = msg.header;
+                    if (!header) {
+                        console.error("Expected to be sent header first");
+                        this.toLocalNode.write({
+                            action: "known",
+                            id: msg.id,
+                            header: false,
+                            sessions: {},
+                            isCorrection: true,
+                        });
+                        throw new Error("Expected to be sent header first");
+                    }
 
-        // console.log(theirKnown.id, nonEmptyNewContentPieces);
+                    return this.makeRequest<IDBValidKey>(({ coValues }) =>
+                        coValues.put({
+                            id: msg.id,
+                            header: header,
+                        } satisfies CoValueRow)
+                    ) as SyncPromise<number>;
+                } else {
+                    return SyncPromise.resolve(coValueRow.rowID);
+                }
+            })
+            .then((storedCoValueRowID: number) => {
+                this.makeRequest<StoredSessionRow[]>(({ sessions }) =>
+                    sessions
+                        .index("sessionsByCoValue")
+                        .getAll(storedCoValueRowID)
+                ).then((allOurSessionsEntries) => {
+                    const allOurSessions: {
+                        [sessionID: SessionID]: StoredSessionRow;
+                    } = Object.fromEntries(
+                        allOurSessionsEntries.map((row) => [row.sessionID, row])
+                    );
 
-        for (const piece of nonEmptyNewContentPieces) {
-            await this.toLocalNode.write(piece);
-            await new Promise((resolve) => setTimeout(resolve, 0));
-        }
+                    const ourKnown: CojsonInternalTypes.CoValueKnownState = {
+                        id: msg.id,
+                        header: true,
+                        sessions: {},
+                    };
+                    let invalidAssumptions = false;
+
+                    return Promise.all(
+                        (Object.keys(msg.new) as SessionID[]).map(
+                            (sessionID) => {
+                                const sessionRow = allOurSessions[sessionID];
+                                if (sessionRow) {
+                                    ourKnown.sessions[sessionRow.sessionID] =
+                                        sessionRow.lastIdx;
+                                }
+
+                                if (
+                                    (sessionRow?.lastIdx || 0) <
+                                    (msg.new[sessionID]?.after || 0)
+                                ) {
+                                    invalidAssumptions = true;
+                                } else {
+                                    return this.putNewTxs(
+                                        msg,
+                                        sessionID,
+                                        sessionRow,
+                                        storedCoValueRowID
+                                    );
+                                }
+                            }
+                        )
+                    ).then(() => {
+                        if (invalidAssumptions) {
+                            this.toLocalNode.write({
+                                action: "known",
+                                ...ourKnown,
+                                isCorrection: invalidAssumptions,
+                            });
+                        }
+                    });
+                });
+            });
     }
 
-    handleLoad(msg: CojsonInternalTypes.LoadMessage) {
-        return this.sendNewContentAfter(msg, this.inTransaction("readonly"));
-    }
+    private putNewTxs(
+        msg: CojsonInternalTypes.NewContentMessage,
+        sessionID: SessionID,
+        sessionRow: StoredSessionRow | undefined,
+        storedCoValueRowID: number
+    ) {
+        const newTransactions = msg.new[sessionID]?.newTransactions || [];
+
+        const actuallyNewOffset =
+            (sessionRow?.lastIdx || 0) - (msg.new[sessionID]?.after || 0);
+
+        const actuallyNewTransactions =
+            newTransactions.slice(actuallyNewOffset);
+
+        let newBytesSinceLastSignature =
+            (sessionRow?.bytesSinceLastSignature || 0) +
+            actuallyNewTransactions.reduce(
+                (sum, tx) =>
+                    sum +
+                    (tx.privacy === "private"
+                        ? tx.encryptedChanges.length
+                        : tx.changes.length),
+                0
+            );
 
-    async handleContent(msg: CojsonInternalTypes.NewContentMessage) {
-        const { coValues, sessions, transactions, signatureAfter } =
-            this.inTransaction("readwrite");
+        const newLastIdx =
+            (sessionRow?.lastIdx || 0) + actuallyNewTransactions.length;
 
-        let storedCoValueRowID = (
-            await promised<StoredCoValueRow | undefined>(
-                coValues.index("coValuesById").get(msg.id)
-            )
-        )?.rowID;
-
-        if (storedCoValueRowID === undefined) {
-            const header = msg.header;
-            if (!header) {
-                console.error("Expected to be sent header first");
-                await this.toLocalNode.write({
-                    action: "known",
-                    id: msg.id,
-                    header: false,
-                    sessions: {},
-                    isCorrection: true,
-                });
-                return;
-            }
+        let shouldWriteSignature = false;
 
-            storedCoValueRowID = (await promised<IDBValidKey>(
-                coValues.put({
-                    id: msg.id,
-                    header: header,
-                } satisfies CoValueRow)
-            )) as number;
+        if (newBytesSinceLastSignature > MAX_RECOMMENDED_TX_SIZE) {
+            shouldWriteSignature = true;
+            newBytesSinceLastSignature = 0;
         }
 
-        const allOurSessions = await new Promise<{
-            [sessionID: SessionID]: StoredSessionRow;
-        }>((resolve) => {
-            const allOurSessionsRequest = sessions
-                .index("sessionsByCoValue")
-                .getAll(storedCoValueRowID);
-
-            allOurSessionsRequest.onsuccess = () => {
-                resolve(
-                    Object.fromEntries(
-                        (
-                            allOurSessionsRequest.result as StoredSessionRow[]
-                        ).map((row) => [row.sessionID, row])
-                    )
-                );
-            };
-        });
+        const nextIdx = sessionRow?.lastIdx || 0;
 
-        const ourKnown: CojsonInternalTypes.CoValueKnownState = {
-            id: msg.id,
-            header: true,
-            sessions: {},
+        const sessionUpdate = {
+            coValue: storedCoValueRowID,
+            sessionID: sessionID,
+            lastIdx: newLastIdx,
+            lastSignature: msg.new[sessionID]!.lastSignature,
+            bytesSinceLastSignature: newBytesSinceLastSignature,
         };
-        let invalidAssumptions = false;
-
-        for (const sessionID of Object.keys(msg.new) as SessionID[]) {
-            const sessionRow = allOurSessions[sessionID];
-            if (sessionRow) {
-                ourKnown.sessions[sessionRow.sessionID] = sessionRow.lastIdx;
-            }
 
-            if ((sessionRow?.lastIdx || 0) < (msg.new[sessionID]?.after || 0)) {
-                invalidAssumptions = true;
+        return this.makeRequest<number>(({ sessions }) =>
+            sessions.put(
+                sessionRow?.rowID
+                    ? {
+                          rowID: sessionRow.rowID,
+                          ...sessionUpdate,
+                      }
+                    : sessionUpdate
+            )
+        ).then((sessionRowID) => {
+            let maybePutRequest;
+            if (shouldWriteSignature) {
+                maybePutRequest = this.makeRequest(({ signatureAfter }) =>
+                    signatureAfter.put({
+                        ses: sessionRowID,
+                        // TODO: newLastIdx is a misnomer, it's actually more like nextIdx or length
+                        idx: newLastIdx - 1,
+                        signature: msg.new[sessionID]!.lastSignature,
+                    } satisfies SignatureAfterRow)
+                );
             } else {
-                const newTransactions =
-                    msg.new[sessionID]?.newTransactions || [];
-
-                const actuallyNewOffset =
-                    (sessionRow?.lastIdx || 0) -
-                    (msg.new[sessionID]?.after || 0);
-
-                const actuallyNewTransactions =
-                    newTransactions.slice(actuallyNewOffset);
-
-                let newBytesSinceLastSignature =
-                    (sessionRow?.bytesSinceLastSignature || 0) +
-                    actuallyNewTransactions.reduce(
-                        (sum, tx) =>
-                            sum +
-                            (tx.privacy === "private"
-                                ? tx.encryptedChanges.length
-                                : tx.changes.length),
-                        0
-                    );
-
-                const newLastIdx =
-                    (sessionRow?.lastIdx || 0) + actuallyNewTransactions.length;
-
-                let shouldWriteSignature = false;
-
-                if (newBytesSinceLastSignature > MAX_RECOMMENDED_TX_SIZE) {
-                    shouldWriteSignature = true;
-                    newBytesSinceLastSignature = 0;
-                }
-
-                let nextIdx = sessionRow?.lastIdx || 0;
-
-                const sessionUpdate = {
-                    coValue: storedCoValueRowID,
-                    sessionID: sessionID,
-                    lastIdx: newLastIdx,
-                    lastSignature: msg.new[sessionID]!.lastSignature,
-                    bytesSinceLastSignature: newBytesSinceLastSignature,
-                };
-
-                const sessionRowID = (await promised(
-                    sessions.put(
-                        sessionRow?.rowID
-                            ? {
-                                  rowID: sessionRow.rowID,
-                                  ...sessionUpdate,
-                              }
-                            : sessionUpdate
-                    )
-                )) as number;
-
-                if (shouldWriteSignature) {
-                    await promised(
-                        signatureAfter.put({
-                            ses: sessionRowID,
-                            // TODO: newLastIdx is a misnomer, it's actually more like nextIdx or length
-                            idx: newLastIdx - 1,
-                            signature: msg.new[sessionID]!.lastSignature,
-                        } satisfies SignatureAfterRow)
-                    );
-                }
-
-                for (const newTransaction of actuallyNewTransactions) {
-                    await promised(
-                        transactions.add({
-                            ses: sessionRowID,
-                            idx: nextIdx,
-                            tx: newTransaction,
-                        } satisfies TransactionRow)
-                    );
-                    nextIdx++;
-                }
+                maybePutRequest = SyncPromise.resolve();
             }
-        }
 
-        if (invalidAssumptions) {
-            await this.toLocalNode.write({
-                action: "known",
-                ...ourKnown,
-                isCorrection: invalidAssumptions,
-            });
-        }
+            return maybePutRequest.then(() =>
+                Promise.all(
+                    actuallyNewTransactions.map((newTransaction, i) => {
+                        return this.makeRequest(({ transactions }) =>
+                            transactions.add({
+                                ses: sessionRowID,
+                                idx: nextIdx + i,
+                                tx: newTransaction,
+                            } satisfies TransactionRow)
+                        );
+                    })
+                )
+            );
+        });
     }
 
     handleKnown(msg: CojsonInternalTypes.KnownStateMessage) {
-        return this.sendNewContentAfter(msg, this.inTransaction("readonly"));
+        return this.sendNewContentAfter(msg);
     }
 
     handleDone(_msg: CojsonInternalTypes.DoneMessage) {}
 
-    inTransaction(mode: "readwrite" | "readonly"): {
-        coValues: IDBObjectStore;
-        sessions: IDBObjectStore;
-        transactions: IDBObjectStore;
-        signatureAfter: IDBObjectStore;
-    } {
-        const tx = this.db.transaction(
-            ["coValues", "sessions", "transactions", "signatureAfter"],
-            mode
-        );
+    // inTransaction(mode: "readwrite" | "readonly"): {
+    //     coValues: IDBObjectStore;
+    //     sessions: IDBObjectStore;
+    //     transactions: IDBObjectStore;
+    //     signatureAfter: IDBObjectStore;
+    // } {
+    //     const tx = this.db.transaction(
+    //         ["coValues", "sessions", "transactions", "signatureAfter"],
+    //         mode
+    //     );
+
+    //     const txID = lastTx;
+    //     lastTx++;
+    //     console.time("IndexedDB TX" + txID);
+
+    //     tx.onerror = (event) => {
+    //         const target = event.target as unknown as {
+    //             error: DOMException;
+    //             source?: { name: string };
+    //         } | null;
+    //         throw new Error(
+    //             `Error in transaction (${target?.source?.name}): ${target?.error}`,
+    //             { cause: target?.error }
+    //         );
+    //     };
+    //     tx.oncomplete = () => {
+    //         console.timeEnd("IndexedDB TX" + txID);
+    //     }
+    //     const coValues = tx.objectStore("coValues");
+    //     const sessions = tx.objectStore("sessions");
+    //     const transactions = tx.objectStore("transactions");
+    //     const signatureAfter = tx.objectStore("signatureAfter");
+
+    //     return { coValues, sessions, transactions, signatureAfter };
+    // }
+}
 
-        tx.onerror = (event) => {
-            const target = event.target as unknown as {
-                error: DOMException;
-                source?: { name: string };
-            } | null;
-            throw new Error(
-                `Error in transaction (${target?.source?.name}): ${target?.error}`,
-                { cause: target?.error }
-            );
-        };
-        const coValues = tx.objectStore("coValues");
-        const sessions = tx.objectStore("sessions");
-        const transactions = tx.objectStore("transactions");
-        const signatureAfter = tx.objectStore("signatureAfter");
+function collectNewTxs(
+    newTxsInSession: TransactionRow[],
+    newContentPieces: CojsonInternalTypes.NewContentMessage[],
+    sessionRow: StoredSessionRow,
+    signaturesAndIdxs: SignatureAfterRow[],
+    theirKnown: CojsonInternalTypes.CoValueKnownState,
+    firstNewTxIdx: number
+) {
+    let idx = firstNewTxIdx;
+
+    // console.log(
+    //     theirKnown.id,
+    //     "newTxInSession",
+    //     newTxInSession.length
+    // );
+    for (const tx of newTxsInSession) {
+        let sessionEntry =
+            newContentPieces[newContentPieces.length - 1]!.new[
+                sessionRow.sessionID
+            ];
+        if (!sessionEntry) {
+            sessionEntry = {
+                after: idx,
+                lastSignature:
+                    "WILL_BE_REPLACED" as CojsonInternalTypes.Signature,
+                newTransactions: [],
+            };
+            newContentPieces[newContentPieces.length - 1]!.new[
+                sessionRow.sessionID
+            ] = sessionEntry;
+        }
 
-        return { coValues, sessions, transactions, signatureAfter };
+        sessionEntry.newTransactions.push(tx.tx);
+
+        if (signaturesAndIdxs[0] && idx === signaturesAndIdxs[0].idx) {
+            sessionEntry.lastSignature = signaturesAndIdxs[0].signature;
+            signaturesAndIdxs.shift();
+            newContentPieces.push({
+                action: "content",
+                id: theirKnown.id,
+                new: {},
+            });
+        } else if (idx === firstNewTxIdx + newTxsInSession.length - 1) {
+            sessionEntry.lastSignature = sessionRow.lastSignature;
+        }
+        idx += 1;
     }
 }
 
-function promised<T>(request: IDBRequest<T>): Promise<T> {
-    return new Promise<T>((resolve, reject) => {
-        request.onsuccess = () => {
-            resolve(request.result);
-        };
-        request.onerror = () => {
-            reject(request.error);
-        };
-    });
+function getDependedOnCoValues(
+    coValueRow: StoredCoValueRow | undefined,
+    newContentPieces: CojsonInternalTypes.NewContentMessage[],
+    theirKnown: CojsonInternalTypes.CoValueKnownState
+) {
+    return coValueRow?.header.ruleset.type === "group"
+        ? newContentPieces
+              .flatMap((piece) => Object.values(piece.new))
+              .flatMap((sessionEntry) =>
+                  sessionEntry.newTransactions.flatMap((tx) => {
+                      if (tx.privacy !== "trusting") return [];
+                      // TODO: avoid parse here?
+                      return cojsonInternals
+                          .parseJSON(tx.changes)
+                          .map(
+                              (change) =>
+                                  change &&
+                                  typeof change === "object" &&
+                                  "op" in change &&
+                                  change.op === "set" &&
+                                  "key" in change &&
+                                  change.key
+                          )
+                          .filter(
+                              (key): key is CojsonInternalTypes.RawCoID =>
+                                  typeof key === "string" &&
+                                  key.startsWith("co_")
+                          );
+                  })
+              )
+        : coValueRow?.header.ruleset.type === "ownedByGroup"
+        ? [
+              coValueRow?.header.ruleset.group,
+              ...new Set(
+                  newContentPieces.flatMap((piece) =>
+                      Object.keys(piece)
+                          .map((sessionID) =>
+                              cojsonInternals.accountOrAgentIDfromSessionID(
+                                  sessionID as SessionID
+                              )
+                          )
+                          .filter(
+                              (accountID): accountID is AccountID =>
+                                  cojsonInternals.isAccountID(accountID) &&
+                                  accountID !== theirKnown.id
+                          )
+                  )
+              ),
+          ]
+        : [];
 }
+// let lastTx = 0;
+
+// function promised<T>(request: IDBRequest<T>): Promise<T> {
+//     return new Promise<T>((resolve, reject) => {
+//         request.onsuccess = () => {
+//             resolve(request.result);
+//         };
+//         request.onerror = () => {
+//             reject(request.error);
+//         };
+//     });
+// }