From 978b71bdbebec295865499fcbc749b4ab6e8a6c3 Mon Sep 17 00:00:00 2001
From: dariaoldenburg <daria.oldenburg.ext@wire.com>
Date: Fri, 31 Jan 2025 16:27:27 +0100
Subject: [PATCH] feat(react-ui-kit): add menuListHeading inside Select
 [WPB-3307] (#6714)

* feat(react-ui-kit): add menuListHeading inside Select [WPB-3307]

* feat(react-ui-kit/SelectComponents/MenuList): use separate file for styles and add aria-label for btn [WPB-3307]
---
 packages/react-ui-kit/src/Form/Select.md      | 39 ++++++++++++++++--
 packages/react-ui-kit/src/Form/Select.tsx     |  4 ++
 .../src/Form/SelectComponents.styles.ts       | 40 +++++++++++++++++++
 .../src/Form/SelectComponents.tsx             | 35 +++++++++++++++-
 4 files changed, 114 insertions(+), 4 deletions(-)
 create mode 100644 packages/react-ui-kit/src/Form/SelectComponents.styles.ts

diff --git a/packages/react-ui-kit/src/Form/Select.md b/packages/react-ui-kit/src/Form/Select.md
index 06ae497e05..878b095d48 100644
--- a/packages/react-ui-kit/src/Form/Select.md
+++ b/packages/react-ui-kit/src/Form/Select.md
@@ -31,7 +31,8 @@ const groupOptions = [
     ],
   },
 ];
-const [isOpen, setIsOpen] = React.useState(false);
+const [isGroupSelectOpen, setIsGroupSelectOpen] = React.useState(false);
+const [isGroupSelectWithHeadingOpen, setIsGroupSelectWithHeadingOpen] = React.useState(false);
 
 <Container>
   <Columns>
@@ -93,12 +94,12 @@ const [isOpen, setIsOpen] = React.useState(false);
       <button
         className="device-toggle-button"
         onClick={() => {
-          setIsOpen(prev => !prev);
+          setIsGroupSelectOpen(prev => !prev);
         }}
         style={{width: '100px', height: '30px'}}
       >
         click me
-        {isOpen && (
+        {isGroupSelectOpen && (
           <Select
             id="groupSelect"
             menuPlacement="top"
@@ -117,6 +118,38 @@ const [isOpen, setIsOpen] = React.useState(false);
     </Column>
   </Columns>
 
+  <Columns>
+    <Column>Select With Groups and heading for all groups</Column>
+
+    <Column>
+      <button
+        className="device-toggle-button"
+        onClick={() => {
+          setIsGroupSelectWithHeadingOpen(prev => !prev);
+        }}
+        style={{width: '100px', height: '30px'}}
+      >
+        click me
+        {isGroupSelectWithHeadingOpen && (
+          <Select
+            id="groupSelectWithHeading"
+            menuPlacement="top"
+            menuIsOpen
+            controlShouldRenderValue={false}
+            isClearable={false}
+            backspaceRemovesValue={false}
+            hideSelectedOptions={false}
+            options={groupOptions}
+            onChange={selectedOption => console.log('Selected option', selectedOption)}
+            dataUieName="group-select-with-heading"
+            defaultValue={[groupOptions[0].options[0], groupOptions[1].options[3]]}
+            menuListHeading="Heading"
+          />
+        )}
+      </button>
+    </Column>
+  </Columns>
+
   <Columns>
     <Column>Invalid Select</Column>
 
diff --git a/packages/react-ui-kit/src/Form/Select.tsx b/packages/react-ui-kit/src/Form/Select.tsx
index d95ff9c0f5..67b7623020 100644
--- a/packages/react-ui-kit/src/Form/Select.tsx
+++ b/packages/react-ui-kit/src/Form/Select.tsx
@@ -29,6 +29,7 @@ import {
   DropdownIndicator,
   IndicatorsContainer,
   Menu,
+  MenuList,
   SelectContainer,
   ValueContainer,
   isGroup,
@@ -61,6 +62,7 @@ interface SelectProps<IsMulti extends boolean, Group extends GroupBase<Option>>
   isMulti?: IsMulti;
   isSearchable?: boolean;
   overlayMenu?: boolean;
+  menuListHeading?: string;
 }
 
 export const Select = <IsMulti extends boolean = false, Group extends GroupBase<Option> = GroupBase<Option>>({
@@ -78,6 +80,7 @@ export const Select = <IsMulti extends boolean = false, Group extends GroupBase<
   required = false,
   isSearchable = false,
   overlayMenu = true,
+  menuListHeading,
   ...props
 }: SelectProps<IsMulti, Group>) => {
   const theme = useTheme();
@@ -119,6 +122,7 @@ export const Select = <IsMulti extends boolean = false, Group extends GroupBase<
           Menu: Menu(dataUieName, menuCSS),
           ValueContainer,
           IndicatorsContainer,
+          ...(menuListHeading && {MenuList: MenuList(menuListHeading, dataUieName)}),
         }}
         tabIndex={TabIndex.UNFOCUSABLE}
         isDisabled={disabled}
diff --git a/packages/react-ui-kit/src/Form/SelectComponents.styles.ts b/packages/react-ui-kit/src/Form/SelectComponents.styles.ts
new file mode 100644
index 0000000000..8a1e08bf33
--- /dev/null
+++ b/packages/react-ui-kit/src/Form/SelectComponents.styles.ts
@@ -0,0 +1,40 @@
+/*
+ * Wire
+ * Copyright (C) 2024 Wire Swiss GmbH
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see http://www.gnu.org/licenses/.
+ *
+ */
+
+import {CSSObject} from '@emotion/react';
+
+import {Theme} from '../Theme/Theme';
+
+export const menuListHeadingContainerStyles = (theme: Theme): CSSObject => ({
+  display: 'flex',
+  justifyContent: 'space-between',
+  alignItems: 'center',
+  fontSize: theme.fontSizes.medium,
+  fontWeight: 600,
+  padding: '8px 16px',
+});
+
+export const menuListCloseButtonStyles: CSSObject = {
+  background: 'transparent',
+  border: 'none',
+  cursor: 'pointer',
+  padding: 0,
+  display: 'flex',
+  alignItems: 'center',
+};
diff --git a/packages/react-ui-kit/src/Form/SelectComponents.tsx b/packages/react-ui-kit/src/Form/SelectComponents.tsx
index 1060d28f8c..013bd26863 100644
--- a/packages/react-ui-kit/src/Form/SelectComponents.tsx
+++ b/packages/react-ui-kit/src/Form/SelectComponents.tsx
@@ -28,14 +28,17 @@ import {
   MenuProps,
   GroupBase,
   OptionsOrGroups,
+  MenuListProps,
 } from 'react-select';
 
 import {Option} from './Select';
+import {menuListCloseButtonStyles, menuListHeadingContainerStyles} from './SelectComponents.styles';
 
-import {CheckIcon} from '../Icon';
+import {CheckIcon, CloseIcon} from '../Icon';
 import {ArrowDown} from '../Icon/ArrowDown';
 import {Theme} from '../Layout';
 import {TabIndex} from '../types/enums';
+
 // SelectContainer
 export const SelectContainer = (props: ContainerProps) => {
   return (
@@ -134,6 +137,36 @@ export const Menu = (dataUieName: string, css?: CSSObject) => (props: MenuProps)
   );
 };
 
+// eslint-disable-next-line react/display-name
+export const MenuList = (menuListHeading: string, dataUieName: string) => (props: MenuListProps) => {
+  const {selectProps, children} = props;
+
+  const handleClose = () => {
+    if (selectProps && selectProps.onMenuClose) {
+      selectProps.onMenuClose();
+    }
+  };
+
+  return (
+    <components.MenuList {...props}>
+      <div
+        {...(dataUieName && {
+          'data-uie-name': `menu-list-${dataUieName}`,
+        })}
+      >
+        <div css={(theme: Theme) => menuListHeadingContainerStyles(theme)}>
+          {menuListHeading}
+          <button onClick={handleClose} css={menuListCloseButtonStyles} aria-label={`Close: ${menuListHeading}`}>
+            <CloseIcon width={16} height={16} />
+          </button>
+        </div>
+        {children}
+      </div>
+    </components.MenuList>
+  );
+};
+MenuList.displayName = 'MenuList';
+
 export const renderValue = value => {
   if (Array.isArray(value)) {
     const currentValue = (i: number) => value[i].props.children;