From 7fc53be9308c920436af9ee814ee0fa19cfc9db8 Mon Sep 17 00:00:00 2001
From: liujuping <liujup@foxmail.com>
Date: Wed, 17 Jan 2024 11:08:28 +0800
Subject: [PATCH] fix(context-menu): fix context menu component instance
 changed

---
 packages/shell/src/api/commonUI.tsx | 35 ++++++++++++++++-------------
 1 file changed, 20 insertions(+), 15 deletions(-)

diff --git a/packages/shell/src/api/commonUI.tsx b/packages/shell/src/api/commonUI.tsx
index f25cd232c..69dd104b2 100644
--- a/packages/shell/src/api/commonUI.tsx
+++ b/packages/shell/src/api/commonUI.tsx
@@ -8,6 +8,7 @@ import {
 import { Balloon, Breadcrumb, Button, Card, Checkbox, DatePicker, Dialog, Dropdown, Form, Icon, Input, Loading, Message, Overlay, Pagination, Radio, Search, Select, SplitButton, Step, Switch, Tab, Table, Tree, TreeSelect, Upload, Divider } from '@alifd/next';
 import { ContextMenu } from '../components/context-menu';
 import { editorSymbol } from '../symbols';
+import { ReactElement } from 'react';
 
 export class CommonUI implements IPublicApiCommonUI {
   [editorSymbol]: IEditor;
@@ -40,8 +41,27 @@ export class CommonUI implements IPublicApiCommonUI {
   Upload = Upload;
   Divider = Divider;
 
+  ContextMenu: ((props: {
+    menus: IPublicTypeContextMenuAction[];
+    children: React.ReactElement[] | React.ReactElement;
+  }) => ReactElement) & {
+    create(menus: IPublicTypeContextMenuAction[], event: MouseEvent | React.MouseEvent): void;
+  };
+
   constructor(editor: IEditor) {
     this[editorSymbol] = editor;
+
+    const innerContextMenu = (props: any) => {
+      const pluginContext: IPublicModelPluginContext = editor.get('pluginContext') as IPublicModelPluginContext;
+      return <ContextMenu {...props} pluginContext={pluginContext} />;
+    };
+
+    innerContextMenu.create = (menus: IPublicTypeContextMenuAction[], event: MouseEvent) => {
+      const pluginContext: IPublicModelPluginContext = editor.get('pluginContext') as IPublicModelPluginContext;
+      return ContextMenu.create(pluginContext, menus, event);
+    };
+
+    this.ContextMenu = innerContextMenu;
   }
 
   get Tip() {
@@ -55,19 +75,4 @@ export class CommonUI implements IPublicApiCommonUI {
   get Title() {
     return InnerTitle;
   }
-
-  get ContextMenu() {
-    const editor = this[editorSymbol];
-    const innerContextMenu = (props: any) => {
-      const pluginContext: IPublicModelPluginContext = editor.get('pluginContext') as IPublicModelPluginContext;
-      return <ContextMenu {...props} pluginContext={pluginContext} />;
-    };
-
-    innerContextMenu.create = (menus: IPublicTypeContextMenuAction[], event: MouseEvent) => {
-      const pluginContext: IPublicModelPluginContext = editor.get('pluginContext') as IPublicModelPluginContext;
-      return ContextMenu.create(pluginContext, menus, event);
-    };
-
-    return innerContextMenu;
-  }
 }