Remove document from synchronizer when document is removed from repo

packages/automerge-repo/src/Repo.ts

@@ -110,8 +110,7 @@ export class Repo extends EventEmitter<RepoEvents> {
     this.sharePolicy = sharePolicy ?? this.sharePolicy
 
     this.on("delete-document", ({ documentId }) => {
-      // TODO Pass the delete on to the network
-      // synchronizer.removeDocument(documentId)
+      this.synchronizer.removeDocument(documentId)
 
       if (storageSubsystem) {
         storageSubsystem.removeDoc(documentId).catch(err => {
@@ -798,8 +797,8 @@ export class Repo extends EventEmitter<RepoEvents> {
         )
       }
       delete this.#handleCache[documentId]
-      // TODO: remove document from synchronizer when removeDocument is implemented
-      // this.synchronizer.removeDocument(documentId)
+      delete this.#progressCache[documentId]
+      this.synchronizer.removeDocument(documentId)
     } else {
       this.#log(
         `WARN: removeFromCache called but doc undefined for documentId: ${documentId}`

packages/automerge-repo/src/synchronizer/CollectionSynchronizer.ts

@@ -138,10 +138,15 @@ export class CollectionSynchronizer extends Synchronizer {
     })
   }
 
-  // TODO: implement this
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  /** Removes a document and stops synchronizing them */
   removeDocument(documentId: DocumentId) {
-    throw new Error("not implemented")
+    log(`removing document ${documentId}`)
+    const docSynchronizer = this.docSynchronizers[documentId]
+    if (docSynchronizer !== undefined) {
+      this.peers.forEach(peerId => docSynchronizer.endSync(peerId))
+    }
+    delete this.docSynchronizers[documentId]
+    delete this.#docSetUp[documentId]
   }
 
   /** Adds a peer and maybe starts synchronizing with them */

packages/automerge-repo/test/CollectionSynchronizer.test.ts

@@ -75,4 +75,27 @@ describe("CollectionSynchronizer", () => {
 
       setTimeout(done)
     }))
+
+  it("removes document", () =>
+    new Promise<void>((done, reject) => {
+      const handle = repo.create()
+      synchronizer.addDocument(handle)
+      synchronizer.addPeer("peer1" as PeerId)
+      // starts synchronizing document to peer
+      synchronizer.once("message", event => {
+        const { targetId, documentId } = event as SyncMessage
+        assert(targetId === "peer1")
+        assert(documentId === handle.documentId)
+        done()
+      })
+      // no message should be sent after removing document
+      synchronizer.once("message", () => {
+        reject(new Error("Should not have sent a message"))
+      })
+      assert(synchronizer.docSynchronizers[handle.documentId] !== undefined)
+      synchronizer.removeDocument(handle.documentId)
+      assert(synchronizer.docSynchronizers[handle.documentId] === undefined)
+      // removing document again should not throw an error
+      synchronizer.removeDocument(handle.documentId)
+    }))
 })

packages/automerge-repo/test/Repo.test.ts

@@ -519,13 +519,19 @@ describe("Repo", () => {
         const handle = repo.create({ foo: "bar" })
         await repo.delete(handle.documentId)
         assert(repo.handles[handle.documentId] === undefined)
+        assert(
+          repo.synchronizer.docSynchronizers[handle.documentId] === undefined
+        )
       })
 
       it("removeFromCache removes doc handle", async () => {
         const { repo } = setup()
         const handle = repo.create({ foo: "bar" })
         await repo.removeFromCache(handle.documentId)
         assert(repo.handles[handle.documentId] === undefined)
+        assert(
+          repo.synchronizer.docSynchronizers[handle.documentId] === undefined
+        )
       })
 
       it("removeFromCache for documentId not found", async () => {