From 18dfad28d7484435959b343d1a5c4b59124bdabc Mon Sep 17 00:00:00 2001
From: zzj3720 <17165520+zzj3720@users.noreply.github.com>
Date: Thu, 27 Feb 2025 05:16:10 +0000
Subject: [PATCH] fix(editor): toDraftModal supports flat data structures
 (#10466)

fix: PD-2374
fix: BS-2703
---
 blocksuite/framework/store/src/model/block/draft.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/blocksuite/framework/store/src/model/block/draft.ts b/blocksuite/framework/store/src/model/block/draft.ts
index e9e764c60..48baa54a9 100644
--- a/blocksuite/framework/store/src/model/block/draft.ts
+++ b/blocksuite/framework/store/src/model/block/draft.ts
@@ -15,10 +15,14 @@ export function toDraftModel<Model extends BlockModel = BlockModel>(
   origin: Model
 ): DraftModel<Model> {
   const { id, version, flavour, role, keys, text, children } = origin;
+
+  const isFlatData = origin.schema.model.isFlatData;
   const props = origin.keys.reduce((acc, key) => {
+    const target = isFlatData ? origin.props : origin;
+    const value = target[key as keyof typeof target];
     return {
       ...acc,
-      [key]: origin[key as keyof Model],
+      [key]: value,
     };
   }, {} as ModelProps<Model>);