feat: _internalRenderSubMenuItem and _internalRenderMenuItem support …

…selected (#432) * _internalRenderSubMenuItem and _internalRenderMenuItem support selected * use stateProps Co-authored-by: 期贤 <qixian.cs@antgroup.com>
react-component · Jan 17, 2022 · 0256d8b · 0256d8b
1 parent d6b1eb0
commit 0256d8b
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/src/Menu.tsx b/src/Menu.tsx
@@ -120,6 +120,9 @@ export interface MenuProps
   _internalRenderMenuItem?: (
     originNode: React.ReactElement,
     menuItemProps: any,
+    stateProps: {
+      selected: boolean;
+    },
   ) => React.ReactElement;
   /***
    * @private Only used for `pro-layout`. Do not use in your prod directly
@@ -128,6 +131,9 @@ export interface MenuProps
   _internalRenderSubMenuItem?: (
     originNode: React.ReactElement,
     subMenuItemProps: any,
+    stateProps: {
+      selected: boolean;
+    },
   ) => React.ReactElement;
 }
 

diff --git a/src/MenuItem.tsx b/src/MenuItem.tsx
@@ -244,7 +244,7 @@ const InternalMenuItem = (props: MenuItemProps) => {
   );
 
   if (_internalRenderMenuItem) {
-    renderNode = _internalRenderMenuItem(renderNode, props);
+    renderNode = _internalRenderMenuItem(renderNode, props, { selected });
   }
 
   return renderNode;

diff --git a/src/SubMenu/index.tsx b/src/SubMenu/index.tsx
@@ -355,7 +355,9 @@ const InternalSubMenu = (props: SubMenuProps) => {
   );
 
   if (_internalRenderSubMenuItem) {
-    listNode = _internalRenderSubMenuItem(listNode, props);
+    listNode = _internalRenderSubMenuItem(listNode, props, {
+      selected: childrenSelected,
+    });
   }
 
   // >>>>> Render