useDocument listen to delete events

automerge · Mar 10, 2024 · ef122f7 · ef122f7
1 parent 5977b9d
commit ef122f7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/packages/automerge-repo-react-hooks/src/useDocument.ts b/packages/automerge-repo-react-hooks/src/useDocument.ts
@@ -44,8 +44,11 @@ export function useDocument<T>(id?: AnyDocumentId) {
 
     const onChange = (h: DocHandleChangePayload<T>) => setDoc(h.doc)
     handle.on("change", onChange)
+    const onDelete = () => setDoc(undefined)
+    handle.on("delete", onDelete)
     const cleanup = () => {
       handle.removeListener("change", onChange)
+      handle.removeListener("delete", onDelete)
     }
 
     return cleanup