Jayden/AFFiNE-Mirror
0
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-07-19 19:16:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: mindmap ai preview (#10222)

Browse Source
This commit is contained in:
doouding
2025-02-17 12:45:20 +00:00
parent 5748591bc5
commit eed00e0b26
3 changed files with 26 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
blocksuite/framework/block-std/src/gfx/view/view-manager.ts
+16 -15
View File
@@ -5,6 +5,7 @@ import type { GfxController } from '../controller.js';
import { GfxExtension, GfxExtensionIdentifier } from '../extension.js';
import { GfxBlockElementModel } from '../model/gfx-block-model.js';
import type { GfxModel } from '../model/model.js';
import type { GfxPrimitiveElementModel } from '../model/surface/element-model.js';
import type { GfxLocalElementModel } from '../model/surface/local-element-model.js';
import type { SurfaceBlockModel } from '../model/surface/surface-model.js';
import {
@@ -57,15 +58,21 @@ export class ViewManager extends GfxExtension {
});
const updateViewOnElementChange = (surface: SurfaceBlockModel) => {
const createView = (
model: GfxPrimitiveElementModel | GfxLocalElementModel
) => {
const ViewCtor =
this._viewCtorMap.get(model.type) ?? GfxElementModelView;
const view = new ViewCtor(model, this.gfx);
this._viewMap.set(model.id, view);
view.onCreated();
};
this._disposable.add(
surface.elementAdded.on(payload => {
const model = surface.getElementById(payload.id)!;
const ViewCtor =
this._viewCtorMap.get(model.type) ?? GfxElementModelView;
const view = new ViewCtor(model, this.gfx);
this._viewMap.set(model.id, view);
view.onCreated();
createView(model);
})
);
@@ -79,9 +86,7 @@ export class ViewManager extends GfxExtension {
this._disposable.add(
surface.localElementAdded.on(model => {
const View = this._viewCtorMap.get(model.type) ?? GfxElementModelView;
this._viewMap.set(model.id, new View(model, this.gfx));
createView(model);
})
);
@@ -94,15 +99,11 @@ export class ViewManager extends GfxExtension {
);
surface.localElementModels.forEach(model => {
const View = this._viewCtorMap.get(model.type) ?? GfxElementModelView;
this._viewMap.set(model.id, new View(model, this.gfx));
createView(model);
});
surface.elementModels.forEach(model => {
const View = this._viewCtorMap.get(model.type) ?? GfxElementModelView;
this._viewMap.set(model.id, new View(model, this.gfx));
createView(model);
});
};