fix(electron): export and import (#9767)

packages/frontend/native/nbstore/src/lib.rs

@@ -9,6 +9,7 @@ use chrono::NaiveDateTime;
 use napi::bindgen_prelude::*;
 use napi_derive::napi;
 use pool::SqliteDocStoragePool;
+use storage::SqliteDocStorage;
 
 #[cfg(feature = "use-as-lib")]
 type Result<T> = anyhow::Result<T>;
@@ -99,12 +100,6 @@ impl DocStoragePool {
     Ok(())
   }
 
-  #[napi]
-  pub async fn disconnect(&self, universal_id: String) -> Result<()> {
-    self.pool.disconnect(universal_id).await?;
-    Ok(())
-  }
-
   #[napi]
   pub async fn set_space_id(&self, universal_id: String, space_id: String) -> Result<()> {
     self
@@ -115,6 +110,18 @@ impl DocStoragePool {
     Ok(())
   }
 
+  #[napi]
+  pub async fn disconnect(&self, universal_id: String) -> Result<()> {
+    self.pool.disconnect(universal_id).await?;
+    Ok(())
+  }
+
+  #[napi]
+  pub async fn checkpoint(&self, universal_id: String) -> Result<()> {
+    self.pool.ensure_storage(universal_id)?.checkpoint().await?;
+    Ok(())
+  }
+
   #[napi]
   pub async fn push_update(
     &self,
@@ -430,3 +437,32 @@ impl DocStoragePool {
     Ok(())
   }
 }
+
+#[napi]
+pub struct DocStorage {
+  storage: SqliteDocStorage,
+}
+
+#[napi]
+impl DocStorage {
+  #[napi(constructor, async_runtime)]
+  pub fn new(path: String) -> Self {
+    Self {
+      storage: SqliteDocStorage::new(path),
+    }
+  }
+
+  #[napi]
+  pub async fn validate(&self) -> Result<bool> {
+    Ok(self.storage.validate().await?)
+  }
+
+  #[napi]
+  pub async fn set_space_id(&self, space_id: String) -> Result<()> {
+    self.storage.connect().await?;
+    self.storage.set_space_id(space_id).await?;
+    println!("clocks {:?}", self.storage.get_doc_clocks(None).await?);
+    self.storage.close().await;
+    Ok(())
+  }
+}

packages/frontend/native/nbstore/src/pool.rs

@@ -46,9 +46,13 @@ impl SqliteDocStoragePool {
   }
 
   pub async fn disconnect(&self, universal_id: String) -> Result<()> {
-    let storage = self.ensure_storage(universal_id.to_owned())?;
-    storage.close().await;
-    self.inner.remove(&universal_id);
+    let entry = self.inner.entry(universal_id);
+
+    if let Entry::Occupied(entry) = entry {
+      let storage = entry.remove();
+      storage.close().await;
+    }
+
     Ok(())
   }
 }