fix: fix document.onMountNode parameters

alibaba · Oct 26, 2023 · 055ab68 · 055ab68
1 parent 11ee97c
commit 055ab68
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/packages/designer/src/document/document-model.ts b/packages/designer/src/document/document-model.ts
@@ -74,7 +74,6 @@ export interface IDocumentModel extends Omit<IPublicModelDocumentModel<
   'onRemoveNode' |
   'onChangeDetecting' |
   'onChangeSelection' |
-  'onMountNode' |
   'onChangeNodeProp' |
   'onImportSchema' |
   'isDetectingNode' |
@@ -414,6 +413,14 @@ export class DocumentModel implements IDocumentModel {
     return node ? !node.isPurged : false;
   }
 
+  onMountNode(fn: (payload: { node: INode }) => void) {
+    this.designer.editor.eventBus.on('node.add', fn as any);
+
+    return () => {
+      this.designer.editor.eventBus.off('node.add', fn as any);
+    };
+  }
+
   /**
    * 根据 schema 创建一个节点
    */

diff --git a/packages/shell/src/model/document-model.ts b/packages/shell/src/model/document-model.ts
@@ -252,10 +252,11 @@ export class DocumentModel implements IPublicModelDocumentModel {
    * 当前 document 新增节点事件，此时节点已经挂载到 document 上
    */
   onMountNode(fn: (payload: { node: IPublicModelNode }) => void): IPublicTypeDisposable {
-    this[editorSymbol].eventBus.on('node.add', fn as any);
-    return () => {
-      this[editorSymbol].eventBus.off('node.add', fn as any);
-    };
+    return this[documentSymbol].onMountNode(({
+      node,
+    }) => {
+      fn({ node: ShellNode.create(node)! });
+    });
   }
 
   /**