okay, tests passing. still more to do

automerge · Jan 5, 2025 · f548d88 · f548d88
1 parent c61e7e4
commit f548d88
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/packages/automerge-repo-react-hooks/src/useDocument.ts b/packages/automerge-repo-react-hooks/src/useDocument.ts
@@ -29,9 +29,15 @@ export function useDocument<T>(
   id: AnyDocumentId
 ): [Doc<T>, (changeFn: ChangeFn<T>, options?: ChangeOptions<T>) => void] {
   const handle = useDocHandle<T>(id)
+  // Initialize with current doc state
   const [doc, setDoc] = useState<Doc<T>>(() => handle.docSync())
   const [deleteError, setDeleteError] = useState<Error>()
 
+  // Reinitialize doc when handle changes
+  useEffect(() => {
+    setDoc(handle.docSync())
+  }, [handle])
+
   useEffect(() => {
     const onChange = () => setDoc(handle.docSync())
     const onDelete = () => {
@@ -54,7 +60,6 @@ export function useDocument<T>(
     [handle]
   )
 
-  // Throw delete error during render
   if (deleteError) {
     throw deleteError
   }