bottable · alanwu4321 · Jun 13, 2020 · Jun 13, 2020
diff --git a/docs/component/menu.mdx b/docs/component/menu.mdx
@@ -4,7 +4,7 @@ route: /component/menu
 menu: Component
 ---
 
-import { Menu, Text } from 'fiber'
+import { Menu, Text, Divider, Heading } from 'fiber'
 import { Playground } from 'docz'
 
 # Menu
@@ -14,16 +14,35 @@ import { Playground } from 'docz'
 <Playground>
   {() => {
     return (
-      <div>
-        <Menu style={{ width: 300 }}>
-          <Menu.Item>
-            <Text> Item 1 </Text>
-          </Menu.Item>
+      <div style={{ width: 300 }}>
+        <Menu>
+          <Menu.Sub title={  <span> Sub Menu 1 </span> }>
+            <Menu.Sub title={ <span> Sub Menu 1.1 </span> }>
+              <Menu.Item>
+                Item 1
+              </Menu.Item>
+              <Menu.Item>
+                Item 2
+              </Menu.Item>
+            </Menu.Sub>
+            <Menu.Item>
+                Item 3
+            </Menu.Item>
+          </Menu.Sub>
+           <Menu.Sub title={ <span> Sub Menu 2 </span>  }>
+            <Menu.Item>
+                Item 4
+            </Menu.Item>
+            <Menu.Item>
+                Item 5
+            </Menu.Item>
+            <Menu.Item>
+                Item 6
+            </Menu.Item>
+          </Menu.Sub>
+          <Divider />
           <Menu.Item>
-            <Text> Item 2 </Text>
-          </Menu.Item>
-          <Menu.Item>
-            <Text> Item 3 </Text>
+             Item 3
           </Menu.Item>
         </Menu>
       </div>

diff --git a/src/Dropdown/Dropdown.tsx b/src/Dropdown/Dropdown.tsx
@@ -38,7 +38,7 @@ const DropdownWrapper = styled.div<any>`
 const useDropdownStatus = (trigger: TriggerType) => {
   const node = useRef<
     HTMLDivElement & { contains: (e: EventTarget) => Boolean }
-    // eslint-disable-next-line indent
+  // eslint-disable-next-line indent
   >(null)
   const [expand, setExpand] = useState(false)
 

diff --git a/src/Dropdown/__tests__/__snapshots__/Dropdown.spec.tsx.snap b/src/Dropdown/__tests__/__snapshots__/Dropdown.spec.tsx.snap
@@ -2,7 +2,7 @@
 
 exports[`Menu & Menu Items renders correctly 1`] = `
 <div
-  className="sc-fzozJi cvaiIV"
+  className="sc-fznyAO hWrpzu"
   onMouseEnter={[Function]}
   onMouseLeave={[Function]}
 >
@@ -13,7 +13,7 @@ exports[`Menu & Menu Items renders correctly 1`] = `
      Hover Me 
   </span>
   <div
-    className="sc-fzoLsD iJVoRQ"
+    className="sc-fznKkj cBTFzs"
   />
 </div>
 `;
diff --git a/src/Menu/Menu.tsx b/src/Menu/Menu.tsx
@@ -1,5 +1,6 @@
 import { styleComposition } from '../utils/styles'
 
+import { SubMenu } from './SubMenu'
 import { MenuItem } from './MenuItem'
 import { MenuProps, MenuContext } from './useMenu'
 
@@ -8,6 +9,7 @@ import styled from 'styled-components'
 
 type MenuFC<P> = FC<P> & {
   Item: FC
+  Sub: FC<any>
 }
 
 const MenuWrapper = styled.ul<MenuProps>`
@@ -18,6 +20,7 @@ const MenuWrapper = styled.ul<MenuProps>`
   padding: 0;
   overflow: auto;
   list-style-type: none;
+  border-right: solid 1px ${(props) => props.theme.colors.gray3};
 `
 
 const Menu: MenuFC<MenuProps> = ({ children, inline }) => {
@@ -29,5 +32,6 @@ const Menu: MenuFC<MenuProps> = ({ children, inline }) => {
 }
 
 Menu.Item = MenuItem
+Menu.Sub = SubMenu
 
 export { Menu }
diff --git a/src/Menu/MenuItem.tsx b/src/Menu/MenuItem.tsx
@@ -2,39 +2,46 @@ import { StyleProps, styleComposition } from '../utils/styles'
 
 import { useMenu } from './useMenu'
 
-import styled from 'styled-components'
-import React, { FC } from 'react'
+import styled, { css } from 'styled-components'
+import React, { FC, useState } from 'react'
 
-export interface MenuItemProps extends StyleProps {}
+export interface MenuItemProps extends StyleProps { }
 
+const selectedItem = css`
+  color: white;
+  background: ${(props) => props.theme.colors.lightest};
+  border-right: solid 3.5px ${(props) => props.theme.colors.light};
+`
 const BlockMenuItem = styled.li<any>`
   ${styleComposition}
 
   display: block;
   padding: 8px 10px;
   float: none;
-  border-right: solid ${(props) => props.theme.colors.gray3} 0.5px;
   color: ${(props) => props.color};
   text-align: left;
   text-decoration: none;
 
   &:hover {
-    background: ${(props) => props.theme.colors.gray3};
+    color: ${(props) => props.theme.colors.light};
     cursor: pointer;
   }
+
+  ${(props) => props.selected ? selectedItem : null};
 `
 
-const InlineMenuItem = styled(BlockMenuItem)<any>`
+const InlineMenuItem = styled(BlockMenuItem) <any>`
   display: inline-block;
   float: left;
 `
 
 const MenuItem: FC<MenuItemProps> = ({ children, ...props }) => {
   const { inline } = useMenu()
+  const [selected, setSelected] = useState<boolean>(false)
 
   const Item = inline ? InlineMenuItem : BlockMenuItem
 
-  return <Item {...props}>{children}</Item>
+  return <Item {...props} selected={selected} onClick={() => setSelected(!selected)}>{children}</Item>
 }
 
 export { MenuItem }
diff --git a/src/Menu/SubMenu.tsx b/src/Menu/SubMenu.tsx
@@ -0,0 +1,31 @@
+import { StyleProps } from '../utils/styles'
+
+import { Arrow, TitleWrapper, SubMenuWrapper, ItemWrapper } from './styles'
+
+import React, { FC, ReactElement, useState } from 'react'
+
+export type TitleFunc = () => React.ReactElement
+
+export interface SubMenuProps extends StyleProps {
+  title: ReactElement | TitleFunc;
+}
+
+const SubMenu: FC<SubMenuProps> = ({ children, title }) => {
+  const [expand, setExpand] = useState<boolean>(false)
+
+  const titleNode = typeof title === 'function' ? (title as TitleFunc)() : title
+
+  return (
+    <SubMenuWrapper>
+      <TitleWrapper expand={expand} onClick={() => setExpand(!expand)}>
+        {titleNode}
+        <Arrow expand={expand} />
+      </TitleWrapper>
+      <ItemWrapper>
+        {expand ? children : null}
+      </ItemWrapper>
+    </SubMenuWrapper>
+  )
+}
+
+export { SubMenu }
diff --git a/src/Menu/__tests__/Menu.spec.tsx b/src/Menu/__tests__/Menu.spec.tsx
@@ -1,4 +1,4 @@
-import { Menu, Text } from '../../'
+import { Menu, Text, Divider } from '../../'
 
 import React from 'react'
 import { render } from 'test/utils'
@@ -7,14 +7,33 @@ describe('Block Menu & Menu Items', () => {
   it('renders correctly', () => {
     const block = render(
       <Menu>
+        <Menu.Sub title={<span> Sub Menu 1 </span>}>
+          <Menu.Sub title={<span> Sub Menu 1.1 </span>}>
+            <Menu.Item>
+              Item 1
+            </Menu.Item>
+            <Menu.Item>
+              Item 2
+            </Menu.Item>
+          </Menu.Sub>
+          <Menu.Item>
+            Item 3
+          </Menu.Item>
+        </Menu.Sub>
+        <Menu.Sub title={<span> Sub Menu 2 </span>}>
+          <Menu.Item>
+            Item 4
+          </Menu.Item>
+          <Menu.Item>
+            Item 5
+          </Menu.Item>
+          <Menu.Item>
+            Item 6
+          </Menu.Item>
+        </Menu.Sub>
+        <Divider />
         <Menu.Item>
-          <Text> Item 1 </Text>
-        </Menu.Item>
-        <Menu.Item>
-          <Text> Item 2 </Text>
-        </Menu.Item>
-        <Menu.Item>
-          <Text> Item 3 </Text>
+          Item 3
         </Menu.Item>
       </Menu>
     )

diff --git a/src/Menu/__tests__/__snapshots__/Menu.spec.tsx.snap b/src/Menu/__tests__/__snapshots__/Menu.spec.tsx.snap
@@ -2,47 +2,70 @@
 
 exports[`Block Menu & Menu Items renders correctly 1`] = `
 <ul
-  className="sc-AxmLO enoELj"
+  className="sc-fzplWN kCYiUw"
 >
-  <li
-    className="sc-AxheI bBCLtC"
+  <div
+    className="sc-Axmtr icUvKO"
   >
-    <span
-      className="sc-AxiKw ioUFXk"
-      size={4}
+    <div
+      className="sc-AxheI kokSYy"
+      onClick={[Function]}
     >
-       Item 1 
-    </span>
-  </li>
-  <li
-    className="sc-AxheI bBCLtC"
+      <span>
+         Sub Menu 1 
+      </span>
+      <i
+        className="sc-fzozJi grUWqN"
+      />
+    </div>
+    <ul
+      className="sc-AxmLO hAqXzN"
+    />
+  </div>
+  <div
+    className="sc-Axmtr icUvKO"
   >
-    <span
-      className="sc-AxiKw ioUFXk"
-      size={4}
+    <div
+      className="sc-AxheI kokSYy"
+      onClick={[Function]}
     >
-       Item 2 
-    </span>
-  </li>
-  <li
-    className="sc-AxheI bBCLtC"
+      <span>
+         Sub Menu 2 
+      </span>
+      <i
+        className="sc-fzozJi grUWqN"
+      />
+    </div>
+    <ul
+      className="sc-AxmLO hAqXzN"
+    />
+  </div>
+  <div
+    className="sc-fznZeY bCvFKa"
   >
     <span
       className="sc-AxiKw ioUFXk"
       size={4}
-    >
-       Item 3 
-    </span>
+    />
+  </div>
+  <li
+    className="sc-fzoLsD ePTHqD"
+    onClick={[Function]}
+    selected={false}
+  >
+    Item 3
   </li>
 </ul>
 `;
 
 exports[`Inline Menu & Menu Items renders correctly 1`] = `
 <ul
-  className="sc-AxmLO enoELj"
+  className="sc-fzplWN kCYiUw"
 >
   <li
-    className="sc-AxheI sc-Axmtr gCqhiy"
+    className="sc-fzoLsD sc-fzpans cPjyk"
+    onClick={[Function]}
+    selected={false}
   >
     <span
       className="sc-AxiKw ioUFXk"
@@ -52,7 +75,9 @@ exports[`Inline Menu & Menu Items renders correctly 1`] = `
     </span>
   </li>
   <li
-    className="sc-AxheI sc-Axmtr gCqhiy"
+    className="sc-fzoLsD sc-fzpans cPjyk"
+    onClick={[Function]}
+    selected={false}
   >
     <span
       className="sc-AxiKw ioUFXk"
@@ -62,7 +87,9 @@ exports[`Inline Menu & Menu Items renders correctly 1`] = `
     </span>
   </li>
   <li
-    className="sc-AxheI sc-Axmtr gCqhiy"
+    className="sc-fzoLsD sc-fzpans cPjyk"
+    onClick={[Function]}
+    selected={false}
   >
     <span
       className="sc-AxiKw ioUFXk"