fix(iroh-docs): Add flush_store and use it to make sure the default author is persisted

iroh-docs/src/actor.rs

@@ -71,6 +71,11 @@ enum Action {
         #[debug("reply")]
         reply: oneshot::Sender<Result<ContentHashesIterator>>,
     },
+    #[display("FlushStore")]
+    FlushStore {
+        #[debug("reply")]
+        reply: oneshot::Sender<Result<()>>,
+    },
     #[display("Replica({}, {})", _0.fmt_short(), _1)]
     Replica(NamespaceId, ReplicaAction),
     #[display("Shutdown")]
@@ -542,6 +547,21 @@ impl SyncHandle {
         rx.await?
     }
 
+    /// Makes sure that all pending database operations are persisted to disk.
+    ///
+    /// Otherwise, database operations are batched into bigger transactions for speed.
+    /// Use this if you need to make sure something is written to the database
+    /// before another operation, e.g. to make sure sudden process exits don't corrupt
+    /// your application state.
+    ///
+    /// It's not necessary to call this function before shutdown, as `shutdown` will
+    /// trigger a flush on its own.
+    pub async fn flush_store(&self) -> Result<()> {
+        let (reply, rx) = oneshot::channel();
+        self.send(Action::FlushStore { reply }).await?;
+        rx.await?
+    }
+
     async fn send(&self, action: Action) -> Result<()> {
         self.tx
             .send_async(action)
@@ -675,6 +695,7 @@ impl Actor {
             Action::ContentHashes { reply } => {
                 send_reply_with(reply, self, |this| this.store.content_hashes())
             }
+            Action::FlushStore { reply } => send_reply(reply, self.store.flush()),
             Action::Replica(namespace, action) => self.on_replica_action(namespace, action),
         }
     }

iroh-docs/src/engine.rs

@@ -351,6 +351,10 @@ impl DefaultAuthorStorage {
                     let author = Author::new(&mut rand::thread_rng());
                     let author_id = author.id();
                     docs_store.import_author(author).await?;
+                    // Make sure to write the default author to the store
+                    // *before* we write the default author ID file.
+                    // Otherwise the default author ID file is effectively a dangling reference.
+                    docs_store.flush_store().await?;
                     self.persist(author_id).await?;
                     Ok(author_id)
                 }