feat: _internalRenderSubMenuItem and _internalRenderMenuItem support selected (#432)

* _internalRenderSubMenuItem and _internalRenderMenuItem support selected

* use stateProps

Co-authored-by: 期贤 <[email protected]>

Author: chenshuai2144

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;
 }
 

src/MenuItem.tsx

@@ -244,7 +244,7 @@ const InternalMenuItem = (props: MenuItemProps) => {
   );
 
   if (_internalRenderMenuItem) {
-    renderNode = _internalRenderMenuItem(renderNode, props);
+    renderNode = _internalRenderMenuItem(renderNode, props, { selected });
   }
 
   return renderNode;

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