Bootstrap the db on connect if required

Author: avinassh

libsql/src/database.rs

@@ -675,6 +675,18 @@ impl Database {
                 };
                 use tokio::sync::Mutex;
 
+                let _ = tokio::task::block_in_place(move || {
+                    let rt = tokio::runtime::Builder::new_current_thread()
+                        .enable_all()
+                        .build()
+                        .unwrap();
+                    rt.block_on(async {
+                        // we will ignore if any errors occurred during the bootstrapping the db,
+                        // because the client could be offline when trying to connect.
+                        let _ = db.bootstrap_db().await;
+                    })
+                });
+
                 let local = db.connect()?;
 
                 if *remote_writes {

libsql/src/local/database.rs

@@ -473,6 +473,13 @@ impl Database {
         crate::sync::sync_offline(&mut sync_ctx, &conn).await
     }
 
+    #[cfg(feature = "sync")]
+    /// Brings the .db file from server, if required.
+    pub async fn bootstrap_db(&self) -> Result<()> {
+        let mut sync_ctx = self.sync_ctx.as_ref().unwrap().lock().await;
+        crate::sync::bootstrap_db(&mut sync_ctx).await
+    }
+
     pub(crate) fn path(&self) -> &str {
         &self.db_path
     }