From 7b0e16f0aee714f374f8f0cf0d4d7a2c6a19fa77 Mon Sep 17 00:00:00 2001
From: lubej <9722540+lubej@users.noreply.github.com>
Date: Wed, 5 Jul 2023 14:34:28 +0200
Subject: [PATCH] Replace network select

---
 src/app/components/Select/index.tsx           |  85 +++++++--
 .../HomePage/Graph/NetworkSelector/index.tsx  | 177 +++++++++++-------
 .../HomePage/Graph/ParaTimeSelector/index.tsx |   2 +-
 src/stories/ButtonsShowroom.stories.tsx       |   8 -
 4 files changed, 172 insertions(+), 100 deletions(-)

diff --git a/src/app/components/Select/index.tsx b/src/app/components/Select/index.tsx
index 083baf43f4..e292f0c5ad 100644
--- a/src/app/components/Select/index.tsx
+++ b/src/app/components/Select/index.tsx
@@ -1,6 +1,6 @@
 import SelectUnstyled, { SelectProps, selectClasses, SelectRootSlotProps } from '@mui/base/Select'
 import Option, { optionClasses } from '@mui/base/Option'
-import Popper from '@mui/base/Popper'
+import Popper, { PopperPlacementType } from '@mui/base/Popper'
 import { styled } from '@mui/material/styles'
 import Box from '@mui/material/Box'
 import {
@@ -21,8 +21,11 @@ import ExpandLessIcon from '@mui/icons-material/ExpandLess'
 import ExpandMoreIcon from '@mui/icons-material/ExpandMore'
 import { COLORS } from '../../../styles/theme/colors'
 import { useTranslation } from 'react-i18next'
+import { WithOptionalOwnerState } from '@mui/base/utils'
+import { SelectPopperSlotProps, SelectSlots } from '@mui/base/Select/Select.types'
+import * as React from 'react'
 
-const StyledButton = styled(Button)(({ theme }) => ({
+export const StyledSelectButton = styled(Button)(({ theme }) => ({
   height: '36px',
   minWidth: '135px',
   padding: `0 ${theme.spacing(4)}`,
@@ -35,7 +38,7 @@ const StyledButton = styled(Button)(({ theme }) => ({
   },
 }))
 
-const StyledListbox = styled('ul')(({ theme }) => ({
+export const StyledSelectListbox = styled('ul')(({ theme }) => ({
   boxSizing: 'border-box',
   padding: theme.spacing(0),
   margin: `${theme.spacing(3)} ${theme.spacing(0)}`,
@@ -47,7 +50,7 @@ const StyledListbox = styled('ul')(({ theme }) => ({
   filter: 'drop-shadow(0px 4px 4px rgba(0, 0, 0, 0.25))',
 }))
 
-const StyledOption = styled(Option)(({ theme }) => ({
+export const StyledSelectOption = styled(Option)(({ theme }) => ({
   display: 'flex',
   alignItems: 'center',
   boxSizing: 'border-box',
@@ -85,28 +88,48 @@ const TertiaryButton = forwardRef(
     const { t } = useTranslation()
 
     return (
-      <StyledButton {...restProps} ref={ref} color="tertiary">
+      <StyledSelectButton {...restProps} ref={ref} color="tertiary">
         <Typography variant="select">{children ? children : t('select.placeholder')}</Typography>
         {ownerState.open ? <ExpandLessIcon /> : <ExpandMoreIcon />}
-      </StyledButton>
+      </StyledSelectButton>
     )
   },
 )
 
 const CustomSelect = forwardRef(function CustomSelect<TValue extends string | number>(
-  props: SelectProps<TValue, false>,
+  {
+    root,
+    listbox,
+    popper,
+    placement,
+    ...restProps
+  }: SelectProps<TValue, false> & SelectSlots<TValue, false> & Partial<SelectPopperSlotProps<TValue, false>>,
   ref: ForwardedRef<HTMLButtonElement>,
 ) {
   const slots: SelectProps<TValue, false>['slots'] = {
-    root: TertiaryButton,
-    listbox: StyledListbox,
-    popper: StyledPopper,
-    ...props.slots,
+    root,
+    listbox,
+    popper,
+    ...restProps.slots,
   }
 
-  return <SelectUnstyled {...props} ref={ref} slots={slots} />
+  return (
+    <SelectUnstyled
+      {...restProps}
+      ref={ref}
+      slots={slots}
+      slotProps={{
+        popper: {
+          placement,
+        },
+      }}
+    />
+  )
 }) as <TValue extends string | number>(
-  props: SelectProps<TValue, false> & RefAttributes<HTMLButtonElement>,
+  props: SelectProps<TValue, false> &
+    SelectSlots<TValue, false> &
+    Partial<SelectPopperSlotProps<TValue, false>> &
+    RefAttributes<HTMLButtonElement>,
 ) => JSX.Element
 
 export interface SelectOptionBase {
@@ -114,11 +137,23 @@ export interface SelectOptionBase {
   value: string | number
 }
 
+const SelectOption = <T extends SelectOptionBase>({ value, label }: T): ReactElement => (
+  <StyledSelectOption key={value} value={value}>
+    <Typography variant="select">{label}</Typography>
+  </StyledSelectOption>
+)
+
 interface SelectCmpProps<T extends SelectOptionBase> {
   label?: string
   options: T[]
   defaultValue?: T['value']
   handleChange?: (selectedOption: T['value'] | null) => void
+  placement?: PopperPlacementType
+  className?: string
+  root?: React.ElementType
+  Option?: typeof SelectOption<T> | undefined
+  listbox?: React.ElementType
+  popper?: React.ComponentType<WithOptionalOwnerState<SelectPopperSlotProps<T['value'], false>>>
 }
 
 const SelectCmp = <T extends SelectOptionBase>({
@@ -126,6 +161,12 @@ const SelectCmp = <T extends SelectOptionBase>({
   options,
   defaultValue,
   handleChange,
+  placement = 'bottom-start',
+  className,
+  root = TertiaryButton,
+  listbox = StyledSelectListbox,
+  popper = StyledPopper,
+  Option = SelectOption,
 }: SelectCmpProps<T>): ReactElement => {
   const selectId = useId()
 
@@ -137,17 +178,23 @@ const SelectCmp = <T extends SelectOptionBase>({
   )
 
   return (
-    <Box>
+    <Box className={className}>
       {label && (
         <label htmlFor={selectId}>
           <Typography variant="body2">{label}</Typography>
         </label>
       )}
-      <CustomSelect<T['value']> id={selectId} defaultValue={defaultValue} onChange={onChange}>
-        {options.map(({ label, value }) => (
-          <StyledOption key={value} value={value}>
-            <Typography variant="select">{label}</Typography>
-          </StyledOption>
+      <CustomSelect<T['value']>
+        id={selectId}
+        defaultValue={defaultValue}
+        onChange={onChange}
+        root={root}
+        listbox={listbox}
+        popper={popper}
+        placement={placement}
+      >
+        {options.map((props: T) => (
+          <Option key={props.value.toString()} {...props} />
         ))}
       </CustomSelect>
     </Box>
diff --git a/src/app/pages/HomePage/Graph/NetworkSelector/index.tsx b/src/app/pages/HomePage/Graph/NetworkSelector/index.tsx
index 48b0b3a130..dbf5e23be3 100644
--- a/src/app/pages/HomePage/Graph/NetworkSelector/index.tsx
+++ b/src/app/pages/HomePage/Graph/NetworkSelector/index.tsx
@@ -1,100 +1,133 @@
-import { FC, useState } from 'react'
+import { FC, ForwardedRef, forwardRef, ReactElement } from 'react'
 import { useTranslation } from 'react-i18next'
-import { useTheme } from '@mui/material/styles'
-import { useScreenSize } from '../../../../hooks/useScreensize'
-import AddIcon from '@mui/icons-material/Add'
 import Box from '@mui/material/Box'
-import Button from '@mui/material/Button'
-import IconButton from '@mui/material/IconButton'
-import RemoveIcon from '@mui/icons-material/Remove'
 import Typography from '@mui/material/Typography'
 import { styled } from '@mui/material/styles'
 import { COLORS } from '../../../../../styles/theme/colors'
 import { getNetworkNames, Network } from '../../../../../types/network'
-import Collapse from '@mui/material/Collapse'
 import { RouteUtils } from '../../../../utils/route-utils'
+import {
+  Select,
+  SelectOptionBase,
+  StyledSelectButton,
+  StyledSelectListbox,
+  StyledSelectOption,
+} from '../../../../components/Select'
+import { SelectRootSlotProps } from '@mui/base/Select'
+import ExpandLessIcon from '@mui/icons-material/ExpandLess'
+import ExpandMoreIcon from '@mui/icons-material/ExpandMore'
+import { getNetworkIcons } from '../../../../utils/content'
+import { optionClasses } from '@mui/base/Option'
 
-const StyledNetworkSelector = styled(Box)(() => ({
+interface NetworkOption extends SelectOptionBase {
+  label: Network
+  value: Network
+}
+
+const StyledNetworkSelector = styled(Select<NetworkOption>)(({ theme }) => ({
   position: 'absolute',
-  bottom: 0,
+  bottom: theme.spacing(5),
   display: 'flex',
   width: '100%',
   justifyContent: 'center',
 }))
 
-const StyledBox = styled(Box)(({ theme }) => ({
+const StyledButton = styled(StyledSelectButton)(({ theme }) => ({
+  height: '44px',
+  minWidth: '200px',
   backgroundColor: theme.palette.layout.primaryBackground,
-  border: `solid 3px ${theme.palette.layout.lightBorder}`,
-  borderRadius: '45px',
-  padding: `${theme.spacing(3)} ${theme.spacing(4)}`,
-  display: 'inline-flex',
-  alignItems: 'center',
+  border: `2px solid ${theme.palette.layout.lightBorder}`,
+  color: theme.palette.layout.main,
+  '&:hover': {
+    backgroundColor: theme.palette.layout.primaryBackground,
+  },
 }))
 
-export const SelectNetworkButton = styled(Button, {
-  shouldForwardProp: prop => prop !== 'isSelectedNetwork',
-})<{ isSelectedNetwork: boolean }>(({ isSelectedNetwork, theme }) => ({
-  height: '30px',
-  padding: `${theme.spacing(2)} ${theme.spacing(3)}`,
-  textTransform: 'capitalize',
-  fontSize: '16px',
-  borderRadius: '9px',
-  backgroundColor: isSelectedNetwork
-    ? theme.palette.layout.primaryBackground
-    : theme.palette.layout.secondary,
-  borderColor: isSelectedNetwork ? theme.palette.layout.hoverBorder : theme.palette.layout.secondary,
-  borderWidth: theme.spacing(1),
-  color: theme.palette.layout.main,
-  '&:hover, &:focus-visible': {
-    backgroundColor: theme.palette.layout.secondary,
-    borderWidth: theme.spacing(1),
-    borderColor: COLORS.white,
+const StyledListbox = styled(StyledSelectListbox)(() => ({
+  minWidth: '200px',
+  background: COLORS.white,
+  color: COLORS.grayDark,
+}))
+
+const StyledOption = styled(StyledSelectOption)(() => ({
+  height: '44px',
+  color: COLORS.grayDark,
+  svg: {
+    color: COLORS.grayExtraDark,
+  },
+  [`&:hover:not(.${optionClasses.disabled}),
+  &.${optionClasses.highlighted}`]: {
+    backgroundColor: 'transparent',
   },
 }))
-SelectNetworkButton.defaultProps = {
-  size: 'small',
-  variant: 'outlined',
-  sx: { ml: 4 },
+
+const SelectOption = ({ value }: NetworkOption): ReactElement => {
+  const { t } = useTranslation()
+
+  const labels = getNetworkNames(t)
+  const icons = getNetworkIcons()
+
+  return (
+    <StyledOption key={value} value={value}>
+      <Box
+        sx={theme => ({ display: 'flex', gap: theme.spacing(3), pl: theme.spacing(3), alignItems: 'center' })}
+      >
+        {icons[value]}
+        <Typography variant="inherit">{labels[value]}</Typography>
+      </Box>
+    </StyledOption>
+  )
 }
 
-type NetworkSelectorProps = {
+const NetworkSelectorButton = forwardRef(
+  (props: SelectRootSlotProps<Network, false>, ref: ForwardedRef<HTMLButtonElement>) => {
+    const { ownerState, ...restProps } = props
+    const { open, value } = ownerState
+    // Expecting value as Network
+    const networkValue = value as Network
+    const { t } = useTranslation()
+    const label = getNetworkNames(t)
+    const icons = getNetworkIcons()
+
+    return (
+      <StyledButton {...restProps} ref={ref} color="inherit">
+        <Box
+          sx={theme => ({
+            display: 'flex',
+            gap: theme.spacing(3),
+            pl: theme.spacing(3),
+            alignItems: 'center',
+          })}
+        >
+          {icons[networkValue]}
+          <Typography variant="inherit">{label[networkValue]}</Typography>
+        </Box>
+        {open ? <ExpandLessIcon /> : <ExpandMoreIcon />}
+      </StyledButton>
+    )
+  },
+)
+
+interface NetworkSelectProps {
   network: Network
-  setNetwork: (network: Network) => void
+  setNetwork: (network: Network | null) => void
 }
 
-export const NetworkSelector: FC<NetworkSelectorProps> = ({ network, setNetwork }) => {
-  const { t } = useTranslation()
-  const theme = useTheme()
-  const { isMobile } = useScreenSize()
-  const [open, setOpen] = useState(false)
-  const options: Network[] = RouteUtils.getEnabledNetworks()
-  const labels = getNetworkNames(t)
+export const NetworkSelector: FC<NetworkSelectProps> = ({ network, setNetwork }) => {
+  const options = RouteUtils.getEnabledNetworks().map(network => ({
+    label: network,
+    value: network,
+  }))
 
   return (
-    <StyledNetworkSelector>
-      <StyledBox>
-        {!isMobile && (
-          <Typography component="span" sx={{ fontSize: '12px', color: theme.palette.layout.main }}>
-            {t('home.selectNetwork')}
-          </Typography>
-        )}
-        <Box sx={{ height: 30, display: 'flex' }}>
-          {options.map(option => (
-            <Collapse orientation="horizontal" in={open || network === option} key={option}>
-              <SelectNetworkButton onClick={() => setNetwork(option)} isSelectedNetwork={option === network}>
-                {labels[option]}
-              </SelectNetworkButton>
-            </Collapse>
-          ))}
-        </Box>
-        <IconButton aria-label={t('home.selectNetworkAria')} onClick={() => setOpen(!open)}>
-          {open ? (
-            <RemoveIcon fontSize="medium" sx={{ color: theme.palette.layout.main, fontSize: '18px' }} />
-          ) : (
-            <AddIcon fontSize="medium" sx={{ color: theme.palette.layout.main, fontSize: '18px' }} />
-          )}
-        </IconButton>
-      </StyledBox>
-    </StyledNetworkSelector>
+    <StyledNetworkSelector
+      defaultValue={network}
+      handleChange={setNetwork}
+      options={options}
+      placement="top-start"
+      root={NetworkSelectorButton}
+      Option={SelectOption}
+      listbox={StyledListbox}
+    />
   )
 }
diff --git a/src/app/pages/HomePage/Graph/ParaTimeSelector/index.tsx b/src/app/pages/HomePage/Graph/ParaTimeSelector/index.tsx
index bc4768922f..29db46d6f4 100644
--- a/src/app/pages/HomePage/Graph/ParaTimeSelector/index.tsx
+++ b/src/app/pages/HomePage/Graph/ParaTimeSelector/index.tsx
@@ -260,7 +260,7 @@ const ParaTimeSelectorCmp: FC<ParaTimeSelectorProps> = ({
           )}
         </ParaTimeSelectorGlobe>
         {step === ParaTimeSelectorStep.Explore && (
-          <NetworkSelector network={network} setNetwork={setNetwork} />
+          <NetworkSelector network={network} setNetwork={network => setNetwork(network ?? Network.mainnet)} />
         )}
       </ParaTimeSelectorGlow>
       {activeMobileGraphTooltip.current && (
diff --git a/src/stories/ButtonsShowroom.stories.tsx b/src/stories/ButtonsShowroom.stories.tsx
index 2beb629e1b..72e86cad86 100644
--- a/src/stories/ButtonsShowroom.stories.tsx
+++ b/src/stories/ButtonsShowroom.stories.tsx
@@ -7,8 +7,6 @@ import { GetStartedBtn } from '../app/pages/HomePage/Graph/HelpScreen'
 import { ViewResultButton } from '../app/pages/SearchResultsPage/ResultsGroupByType'
 import { LoadMoreButton } from '../app/components/LoadMoreButton'
 import { withRouter } from 'storybook-addon-react-router-v6'
-import { SelectNetworkButton } from '../app/pages/HomePage/Graph/NetworkSelector'
-import { NetworkButton } from '../app/components/PageLayout/NetworkButton'
 import WidgetsIcon from '@mui/icons-material/Widgets'
 
 export default {
@@ -37,12 +35,6 @@ const Template: StoryFn = () => {
       <br />
       <br />
 
-      <SelectNetworkButton isSelectedNetwork={false}>SelectNetworkButton</SelectNetworkButton>
-      <NetworkButton isOutOfDate={false} layer="emerald" network="mainnet" onClick={() => {}} />
-
-      <br />
-      <br />
-
       <LoadMoreButton pagination={{ linkToPage: () => '', selectedPage: 1 }} />
     </div>
   )