From c9b4db79ec2b4360f23f42d3ab49f265a56e9447 Mon Sep 17 00:00:00 2001
From: Van Anderson <van.m.anderson@gmail.com>
Date: Wed, 5 May 2021 14:18:58 -0500
Subject: [PATCH] Action Menu can have its open state be controlled externally
 (#1199)

* Action Menu can be a controlled component with open state

* update docs for ActionMenu

* Create pink-lions-suffer.md

* Handle onAction for DropdownMenu Items (#1194)

* useProvidedStateOrCreate

* linter fixes

* update hook doc comment

* Update src/hooks/useProvidedStateOrCreate.ts

Co-authored-by: Dusty Greif <dgreif@users.noreply.github.com>

* useProvidedOrCreateState has better implementation for generics

* useProvidedOrCreateState has better implementation for generics

* make string type explicit

Co-authored-by: Dusty Greif <dgreif@users.noreply.github.com>
---
 .changeset/pink-lions-suffer.md               |  5 +++
 docs/content/ActionMenu.mdx                   | 18 +++++----
 src/ActionMenu.tsx                            | 19 +++++----
 .../hooks/useProvidedStateOrCreate.tsx        | 39 ++++++++++++++++++
 src/hooks/useProvidedStateOrCreate.ts         | 27 +++++++++++++
 src/stories/ActionMenu.stories.tsx            | 40 +++++++++++++++++++
 6 files changed, 133 insertions(+), 15 deletions(-)
 create mode 100644 .changeset/pink-lions-suffer.md
 create mode 100644 src/__tests__/hooks/useProvidedStateOrCreate.tsx
 create mode 100644 src/hooks/useProvidedStateOrCreate.ts

diff --git a/.changeset/pink-lions-suffer.md b/.changeset/pink-lions-suffer.md
new file mode 100644
index 00000000000..54b3857da98
--- /dev/null
+++ b/.changeset/pink-lions-suffer.md
@@ -0,0 +1,5 @@
+---
+"@primer/components": patch
+---
+
+Action Menu can have its open state be controlled externally.
diff --git a/docs/content/ActionMenu.mdx b/docs/content/ActionMenu.mdx
index fef5d6e07fb..85a098b7423 100644
--- a/docs/content/ActionMenu.mdx
+++ b/docs/content/ActionMenu.mdx
@@ -68,11 +68,13 @@ An `ActionMenu` is an ActionList-based component for creating a menu of actions
 
 ## Component props
 
-| Name          | Type                                  |      Default      | Description                                                                                                                                                          |
-| :------------ | :------------------------------------ | :---------------: | :------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
-| items         | `ItemProps[]`                         |    `undefined`    | Required. A list of item objects conforming to the `ActionList.Item` props interface.                                                                                |
-| renderItem    | `(props: ItemProps) => JSX.Element`   | `ActionList.Item` | Optional. If defined, each item in `items` will be passed to this function, allowing for `ActionList`-wide custom item rendering.                                    |
-| groupMetadata | `GroupProps[]`                        |    `undefined`    | Optional. If defined, `ActionList` will group `items` into `ActionList.Group`s separated by `ActionList.Divider` according to their `groupId` property.              |
-| renderAnchor  | `(props: ButtonProps) => JSX.Element` |     `Button`      | Optional. If defined, provided component will be used to render the menu anchor. Will receive the selected `Item` text as `children` prop when an item is activated. |
-| anchorContent | React.ReactNode                       |    `undefined`    | Optional. If defined, it will be passed to the trigger as the elements child.                                                                                        |
-| onAction      | (props: ItemProps) => void            |    `undefined`    | Optional. If defined, this function will be called when a menu item is activated either by a click or a keyboard press.                                              |
+| Name          | Type                                  |      Default      | Description                                                                                                                                                                                        |
+| :------------ | :------------------------------------ | :---------------: | :------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
+| items         | `ItemProps[]`                         |    `undefined`    | Required. A list of item objects conforming to the `ActionList.Item` props interface.                                                                                                              |
+| renderItem    | `(props: ItemProps) => JSX.Element`   | `ActionList.Item` | Optional. If defined, each item in `items` will be passed to this function, allowing for `ActionList`-wide custom item rendering.                                                                  |
+| groupMetadata | `GroupProps[]`                        |    `undefined`    | Optional. If defined, `ActionList` will group `items` into `ActionList.Group`s separated by `ActionList.Divider` according to their `groupId` property.                                            |
+| renderAnchor  | `(props: ButtonProps) => JSX.Element` |     `Button`      | Optional. If defined, provided component will be used to render the menu anchor. Will receive the selected `Item` text as `children` prop when an item is activated.                               |
+| anchorContent | React.ReactNode                       |    `undefined`    | Optional. If defined, it will be passed to the trigger as the elements child.                                                                                                                      |
+| onAction      | (props: ItemProps) => void            |    `undefined`    | Optional. If defined, this function will be called when a menu item is activated either by a click or a keyboard press.                                                                            |
+| open          | boolean                               |    `undefined`    | Optional. If defined, ActionMenu will use this to control the open/closed state of the Overlay instead of controlling the state internally. Should be used in conjunction with the `setOpen` prop. |
+| setOpen       | (state: boolean) => void              |    `undefined`    | Optional. If defined, ActionMenu will use this to control the open/closed state of the Overlay instead of controlling the state internally. Should be used in conjunction with the `open` prop.    |
diff --git a/src/ActionMenu.tsx b/src/ActionMenu.tsx
index 8985e00a796..1b06bbeded5 100644
--- a/src/ActionMenu.tsx
+++ b/src/ActionMenu.tsx
@@ -2,14 +2,16 @@ import {GroupedListProps, List, ListPropsBase} from './ActionList/List'
 import {Item, ItemProps} from './ActionList/Item'
 import {Divider} from './ActionList/Divider'
 import Button, {ButtonProps} from './Button'
-import React, {useCallback, useEffect, useRef, useState} from 'react'
+import React, {useCallback, useEffect, useRef} from 'react'
 import {AnchoredOverlay} from './AnchoredOverlay'
-
+import {useProvidedStateOrCreate} from './hooks/useProvidedStateOrCreate'
 export interface ActionMenuProps extends Partial<Omit<GroupedListProps, keyof ListPropsBase>>, ListPropsBase {
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
   renderAnchor?: (props: any) => JSX.Element
   anchorContent?: React.ReactNode
-  onAction?: (props: ItemProps, event: React.MouseEvent<HTMLDivElement> | React.KeyboardEvent<HTMLDivElement>) => void
+  onAction?: (props: ItemProps, event?: React.MouseEvent<HTMLDivElement> | React.KeyboardEvent<HTMLDivElement>) => void
+  open?: boolean
+  setOpen?: (s: boolean) => void
 }
 
 const ActionMenuItem = (props: ItemProps) => <Item role="menuitem" {...props} />
@@ -21,12 +23,14 @@ const ActionMenuBase = ({
   renderAnchor = <T extends ButtonProps>(props: T) => <Button {...props} />,
   renderItem = Item,
   onAction,
+  open,
+  setOpen,
   ...listProps
 }: ActionMenuProps): JSX.Element => {
-  const [open, setOpen] = useState(false)
-  const onOpen = useCallback(() => setOpen(true), [])
-  const onClose = useCallback(() => setOpen(false), [])
   const pendingActionRef = useRef<() => unknown>()
+  const [combinedOpenState, setCombinedOpenState] = useProvidedStateOrCreate(open, setOpen, false)
+  const onOpen = useCallback(() => setCombinedOpenState(true), [setCombinedOpenState])
+  const onClose = useCallback(() => setCombinedOpenState(false), [setCombinedOpenState])
 
   const renderMenuAnchor = useCallback(
     <T extends React.HTMLAttributes<HTMLElement>>(props: T) => {
@@ -47,6 +51,7 @@ const ActionMenuBase = ({
         onAction: (props, event) => {
           const actionCallback = itemOnAction ?? onAction
           pendingActionRef.current = () => actionCallback?.(props as ItemProps, event)
+          actionCallback?.(props as ItemProps, event)
           onClose()
         }
       })
@@ -64,7 +69,7 @@ const ActionMenuBase = ({
   }, [open])
 
   return (
-    <AnchoredOverlay renderAnchor={renderMenuAnchor} open={open} onOpen={onOpen} onClose={onClose}>
+    <AnchoredOverlay renderAnchor={renderMenuAnchor} open={combinedOpenState} onOpen={onOpen} onClose={onClose}>
       <List {...listProps} role="menu" renderItem={renderMenuItem} />
     </AnchoredOverlay>
   )
diff --git a/src/__tests__/hooks/useProvidedStateOrCreate.tsx b/src/__tests__/hooks/useProvidedStateOrCreate.tsx
new file mode 100644
index 00000000000..cedc663c82d
--- /dev/null
+++ b/src/__tests__/hooks/useProvidedStateOrCreate.tsx
@@ -0,0 +1,39 @@
+import {useProvidedStateOrCreate} from '../../hooks/useProvidedStateOrCreate'
+import {render} from '@testing-library/react'
+import React, {useState} from 'react'
+
+it('will use the provided state', () => {
+  const Component = () => {
+    const [state, setState] = useState('foo')
+    const [combinedState] = useProvidedStateOrCreate(state, setState, 'bar')
+    return <div>{combinedState}</div>
+  }
+
+  const doc = render(<Component />)
+  expect(doc.baseElement.textContent).toEqual('foo')
+})
+
+it('will set state correctly when provided a set state method', () => {
+  const Component = () => {
+    const [state, setState] = useState('foo')
+    const [combinedState, setCombinedState] = useProvidedStateOrCreate(state, setState, 'bar')
+    if (combinedState !== 'baz') setCombinedState('baz')
+    return <div>{combinedState}</div>
+  }
+
+  const doc = render(<Component />)
+  expect(doc.baseElement.textContent).toEqual('baz')
+})
+
+it('if not provided a state, will use an internal state', () => {
+  const Component = () => {
+    const state = undefined
+    const setState = undefined
+    const [combinedState, setCombinedState] = useProvidedStateOrCreate(state, setState, '')
+    if (combinedState !== 'bar') setCombinedState('bar')
+    return <div>{combinedState}</div>
+  }
+
+  const doc = render(<Component />)
+  expect(doc.baseElement.textContent).toEqual('bar')
+})
diff --git a/src/hooks/useProvidedStateOrCreate.ts b/src/hooks/useProvidedStateOrCreate.ts
new file mode 100644
index 00000000000..49508e98880
--- /dev/null
+++ b/src/hooks/useProvidedStateOrCreate.ts
@@ -0,0 +1,27 @@
+import {useCallback, useState} from 'react'
+
+/**
+ * There are some situations where we want to give users the option to control state externally with their own state handlers
+ * or default to using internal state handlers. Because of the 'rules-of-hooks', we cannot conditionally make a call to `React.useState`
+ * only in the situations where the state is not provided as a prop.
+ * This hook aims to encapsulate that logic, so the consumer doesn't need to be concerned with violating `rules-of-hooks`.
+ * @param externalState The state to use - if undefined, will use the state from a call to React.useState
+ * @param setExternalState The setState to use - if undefined, will use the setState from a call to React.useState
+ * @param defaultState The defaultState to use, if using internal state.
+ */
+export function useProvidedStateOrCreate<T>(
+  externalState: T | undefined,
+  setExternalState: ((s: T) => void) | undefined,
+  defaultState: T
+) {
+  const [internalState, setInternalState] = useState<T>(defaultState)
+  const state = externalState ?? internalState
+  const setState = useCallback(
+    (s: T) => {
+      setInternalState(s)
+      if (setExternalState) setExternalState(s)
+    },
+    [setExternalState]
+  )
+  return [state, setState] as const
+}
diff --git a/src/stories/ActionMenu.stories.tsx b/src/stories/ActionMenu.stories.tsx
index 45555283f3d..31b50b251ab 100644
--- a/src/stories/ActionMenu.stories.tsx
+++ b/src/stories/ActionMenu.stories.tsx
@@ -16,6 +16,7 @@ import styled from 'styled-components'
 import {ThemeProvider} from '..'
 import {ActionMenu} from '../ActionMenu'
 import Link, {LinkProps} from '../Link'
+import Button from '../Button'
 import {ActionList, ItemProps} from '../ActionList'
 import BaseStyles from '../BaseStyles'
 
@@ -113,6 +114,45 @@ export function SimpleListStory(): JSX.Element {
 }
 SimpleListStory.storyName = 'Simple List'
 
+export function ExternalOpenState(): JSX.Element {
+  const [option, setOption] = useState('Select an option')
+  const [open, setOpen] = useState(false)
+  const onAction = (itemProps: ItemProps) => {
+    setOption(itemProps.text as string)
+  }
+  return (
+    <>
+      <h1>Simple List</h1>
+      <h2>Last option activated: {option}</h2>
+      <h2>External Open State: {open ? 'Open' : 'Closed'}</h2>
+      <div>
+        <Button onClick={() => setOpen(!open)}>Toggle External State</Button>
+      </div>
+      <br />
+      <ErsatzOverlay>
+        <ActionMenu
+          onAction={onAction}
+          anchorContent="Menu"
+          open={open}
+          setOpen={setOpen}
+          items={[
+            {text: 'New file', trailingText: '⌘O', disabled: true, leadingVisual: ProjectIcon},
+            ActionList.Divider,
+            {text: 'Copy link', trailingText: 'ctrl+C'},
+            {text: 'Edit file', trailingText: '⌘E'},
+            {
+              text: 'Delete file',
+              variant: 'danger',
+              trailingText: '⌘D'
+            }
+          ]}
+        />
+      </ErsatzOverlay>
+    </>
+  )
+}
+ExternalOpenState.storyName = 'External Open State'
+
 export function ComplexListStory(): JSX.Element {
   const [option, setOption] = useState('Select an option')
   const onAction = (itemProps: ItemProps) => {