Jayden/AFFiNE-Mirror
1
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-02-13 21:05:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(editor): transform to draftmodel first when get snapshot (#10477)

Browse Source
This commit is contained in:
Saul-Mirone
2025-02-27 07:52:18 +00:00
parent 1fd3d618be
commit d57ef5c5b3
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
blocksuite/framework/store/src/transformer/base.ts
View File

@@ -1,5 +1,5 @@
import type { BlockModel } from '../model/block/block-model';
import type { DraftModel } from '../model/block/draft';
import { BlockModel } from '../model/block/block-model';
import { type DraftModel, toDraftModel } from '../model/block/draft';
import {
type InternalPrimitives,
internalPrimitives,
@@ -20,7 +20,7 @@ export type FromSnapshotPayload = {
};
export type ToSnapshotPayload<Props extends object> = {
model: DraftModel<BlockModel<Props>>;
model: DraftModel<BlockModel<Props>> | BlockModel<Props>;
assets: AssetsManager;
};
@@ -42,10 +42,16 @@ export class BaseBlockTransformer<Props extends object = object> {
) as Props;
}
protected _propsToSnapshot(model: DraftModel) {
protected _propsToSnapshot(model: DraftModel | BlockModel) {
let draftModel: DraftModel;
if (model instanceof BlockModel) {
draftModel = toDraftModel(model);
} else {
draftModel = model;
}
return Object.fromEntries(
model.keys.map(key => {
const value = model[key as keyof typeof model];
draftModel.keys.map(key => {
const value = draftModel[key as keyof typeof draftModel];
return [key, toJSON(value)];
})
);