type cleanup

fireproof-storage · Feb 10, 2024 · 82cc108 · 82cc108
1 parent ad83dc4
commit 82cc108
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/packages/connect/src/connection.ts b/packages/connect/src/connection.ts
@@ -49,12 +49,14 @@ export abstract class Connection {
     await this.loader!.remoteWAL?._process()
   }
 
-  connect({ loader }: { loader: Loader }) {
+  connect({ loader }: { loader: Loader | undefined }) {
+    if (!loader) throw new Error('loader is required')
     this.connectMeta({ loader })
     this.connectStorage({ loader })
   }
 
-  connectMeta({ loader }: { loader: Loader }) {
+  connectMeta({ loader }: { loader: Loader | undefined}) {
+    if (!loader) throw new Error('loader is required')
     this.loader = loader
     this.taskManager = new TaskManager(loader)
     this.onConnect()
@@ -74,7 +76,8 @@ export abstract class Connection {
 
   async onConnect() {}
 
-  connectStorage({ loader }: { loader: Loader }) {
+  connectStorage({ loader }: { loader?: Loader }) {
+    if (!loader) throw new Error('loader is required')
     this.loader = loader
     loader!.remoteCarStore = new RemoteDataStore(this.loader!.name, this)
     loader!.remoteFileStore = new RemoteDataStore(this.loader!.name, this, 'file')

diff --git a/packages/react/src/useFireproof.tsx b/packages/react/src/useFireproof.tsx
@@ -145,7 +145,7 @@ export function useFireproof(name: string | Database = "useFireproof", config: C
         if (!newDoc) return void (opts.reset ? setDoc(initialDoc) : refreshDoc());
         setDoc((d) => (opts.replace ? (newDoc as Doc<T>) : { ...d, ...newDoc }));
       },
-      [refreshDoc]
+      [refreshDoc, initialDoc]
     );
 
     useEffect(() => {