fix: from_snapshot with shallow snapshot err

Author: zxch3n

crates/loro-internal/src/encoding/fast_snapshot.rs

@@ -94,6 +94,14 @@ pub(crate) fn decode_snapshot(doc: &LoroDoc, bytes: Bytes) -> LoroResult<()> {
 }
 
 pub(crate) fn decode_snapshot_inner(snapshot: Snapshot, doc: &LoroDoc) -> Result<(), LoroError> {
+    let (options, guard) = doc.commit_then_stop();
+    _decode_snapshot_inner(snapshot, doc)?;
+    drop(guard);
+    doc.renew_txn_if_auto_commit(options);
+    Ok(())
+}
+
+fn _decode_snapshot_inner(snapshot: Snapshot, doc: &LoroDoc) -> Result<(), LoroError> {
     let Snapshot {
         oplog_bytes,
         state_bytes,

crates/loro-wasm/tests/basic.test.ts

@@ -1622,3 +1622,15 @@ it("hideEmptyRootContainers", () => {
   doc.setHideEmptyRootContainers(true);
   expect(doc.toJSON()).toStrictEqual({});
 });
+
+it("fromShallowSnapshot", () => {
+  const doc = new LoroDoc();
+  doc.setPeerId("1");
+  doc.getText("text").insert(0, "Hello");
+  doc.commit();
+  const snapshot = doc.export({ mode: "shallow-snapshot", frontiers: doc.frontiers() });
+  const newDoc = LoroDoc.fromSnapshot(snapshot);
+  expect(newDoc.toJSON()).toStrictEqual({
+    text: "Hello"
+  });
+})

crates/loro/src/lib.rs

@@ -472,6 +472,12 @@ impl LoroDoc {
         self.doc.is_detached()
     }
 
+    /// Create a new `LoroDoc` from a snapshot.
+    pub fn from_snapshot(bytes: &[u8]) -> LoroResult<Self> {
+        let inner = InnerLoroDoc::from_snapshot(bytes)?;
+        Ok(Self::_new(inner))
+    }
+
     /// Import updates/snapshot exported by [`LoroDoc::export_snapshot`] or [`LoroDoc::export_from`].
     #[inline]
     pub fn import(&self, bytes: &[u8]) -> Result<ImportStatus, LoroError> {

crates/loro/tests/loro_rust_test.rs

@@ -17,7 +17,7 @@ use loro::{
     event::{Diff, DiffBatch, ListDiffItem},
     loro_value, CommitOptions, ContainerID, ContainerTrait, ContainerType, ExportMode, Frontiers,
     FrontiersNotIncluded, IdSpan, Index, LoroDoc, LoroError, LoroList, LoroMap, LoroMapValue,
-    LoroStringValue, LoroText, LoroValue, ToJson, TreeParentId, UpdateOptions,
+    LoroStringValue, LoroText, LoroValue, ToJson, TreeParentId,
 };
 use loro_internal::{
     encoding::EncodedBlobMode, fx_map, handler::TextDelta, id::ID, version_range, vv, LoroResult,
@@ -3341,3 +3341,20 @@ fn test_hide_empty_root_containers() {
         LoroValue::Map(LoroMapValue::default())
     );
 }
+
+#[test]
+fn test_from_shallow_snapshot() {
+    let doc = LoroDoc::new();
+    doc.set_peer_id(1).unwrap();
+    doc.get_text("text").insert(0, "Hello").unwrap();
+    doc.commit();
+    let snapshot = doc
+        .export(ExportMode::shallow_snapshot_owned(doc.state_frontiers()))
+        .unwrap();
+    let new_doc = LoroDoc::from_snapshot(&snapshot).unwrap();
+    let mut expected = LoroMapValue::default();
+    expected
+        .make_mut()
+        .insert("text".into(), LoroValue::String("Hello".into()));
+    assert_eq!(new_doc.get_deep_value(), LoroValue::Map(expected));
+}