RichSelectList.tsx

import React, {
  type ReactElement,
  useMemo,
  type SyntheticEvent,
  useRef,
  useId,
} from "react";
import classNames from "classnames";
import {
  ColorBaseDestructive700,
  ColorBaseGray800,
} from "@cambly/syntax-design-tokens";
import Typography from "../Typography/Typography";
import useIsHydrated from "../useIsHydrated";
import Popover from "../Popover/Popover";
import {
  Label as ReactAriaLabel,
  Provider as ReactAriaProvider,
} from "react-aria-components";
import { useControlledState } from "@react-stately/utils";
import { setInteractionModality } from "@react-aria/interactions";
import { DialogContext } from "../Dialog/Dialog";
import styles from "../SelectList/SelectList.module.css";
import RichSelectBox, {
  RichSelectBoxContext,
  convertSelection,
  type RichSelectBoxProps,
} from "./RichSelectBox";
import TapArea from "../TapArea/TapArea";
import { type OverlayHandlerRef } from "../react-aria-utils/Triggerable";
import Box from "../Box/Box";
import RichSelectSection from "./RichSelectSection";
import RichSelectChip from "./RichSelectChip";
import RichSelectRadioButton from "./RichSelectRadioButton";
import { useField } from "react-aria";
import { useTheme } from "../ThemeProvider/ThemeProvider";

const NOOP = () => undefined;

const iconSize = {
  sm: 20,
  md: 24,
  lg: 24,
} as const;

export type RichSelectListProps = RichSelectBoxProps & {
  /** Test id for the select element */
  "data-testid"?: string;
  /**
   * Disables the select dropdown entirely
   * @defaultValue false
   */
  disabled?: boolean;
  /** Callback to be called when select is clicked */
  onClick?: (event: SyntheticEvent<HTMLDivElement>) => void;
  /** Text shown below select box if there is an input error. */
  errorText?: string;
  /** Text shown below select box */
  helperText?: string;
  /**
   * RichSelectList id, if not provided, a unique id will be generated
   */
  id?: string;
  /** Text shown above select box */
  label: string;
  /**
   * Text showing in select box if no option has been chosen.
   * There should always have a placeholder unless there is a default option selected
   */
  placeholderText?: string;
  /** Use to render (override) text shown in the select box */
  selectTextValue?: (selectedValues?: "all" | string[]) => string | undefined;
  /**
   * Size of the select box
   *
   * @defaultValue "md"
   */
  size?: "sm" | "md" | "lg";
};

/**
 * [RichSelectList](https://cambly-syntax.vercel.app/?path=/docs/components-richselectlist--docs) is a dropdown menu that allows users to select one or multiple options from a list.
 *
 * Example Usage:
 ```
  <RichSelectList
    label="My Label"
    multiple
    onChange={() => { ... }}
    primaryButtonText="Save"
    primaryButtonAccessibilityLabel="Save"
    secondaryButtonText="Clear"
    secondaryButtonAccessibilityLabel="Clear"
  >
    <RichSelectList.Section label="Cities">
      <RichSelectList.Chip label="San Francisco" value="sf" />
      <RichSelectList.Chip label="New York" value="ny" disabled />
      <RichSelectList.Chip label="Tulsa" value="tulsa" />
      <RichSelectList.Chip label="Chicago" value="chicago" disabled />
    </RichSelectList.Section>
  </RichSelectList>
 ```
 */
function RichSelectList(props: RichSelectListProps): ReactElement {
  const {
    autosave,
    children,
    "data-testid": dataTestId,
    disabled: disabledProp = false,
    errorText,
    helperText,
    label,
    id,
    onChange,
    onClick = NOOP,
    placeholderText,
    selectTextValue,
    selectedValues: selectedValuesProp,
    defaultSelectedValues: defaultSelectedValuesProp,
    size = "md",
    ...richSelectBoxProps
  } = props;

  const reactId = useId();
  const inputId = id ?? reactId;
  const isHydrated = useIsHydrated();
  const disabled = !isHydrated || disabledProp;
  const { themeName } = useTheme();

  // passed to popover, which attached open/close methods
  const overlayHandlerRef = useRef<OverlayHandlerRef>({});

  const selectedKeysProp = useMemo(
    () => convertSelection(selectedValuesProp),
    [selectedValuesProp],
  );
  const defaultSelectedKeys = useMemo(
    () => convertSelection(defaultSelectedValuesProp, new Set()),
    [defaultSelectedValuesProp],
  );
  const [selectedKeys, setSelectedKeys] = useControlledState(
    selectedKeysProp!, // eslint-disable-line @typescript-eslint/no-non-null-assertion -- there is a bug in the typedef for useControlledState from react-stately.  Internally they rely on value (first arg) able to be undefined
    defaultSelectedKeys,
    (value) => {
      const _value = value === "all" ? "all" : [...value].map(String);
      onChange(_value);
      if (!autosave) overlayHandlerRef.current.close?.();
    },
  );

  const selectedTextValue = useMemo(() => {
    if (selectTextValue)
      return (
        selectTextValue(
          selectedKeys === "all" ? "all" : [...selectedKeys].map(String),
        ) ?? placeholderText
      );
    if (selectedKeys === "all") return "All selected";
    if (selectedKeys.size) return `${selectedKeys.size} selected`;
    return placeholderText;
  }, [selectTextValue, selectedKeys, placeholderText]);

  const fieldRef = useRef<HTMLDivElement>(null);
  const { labelProps, fieldProps, descriptionProps, errorMessageProps } =
    useField({
      label, // this is the label for the select box
      description: helperText,
      errorMessage: errorText,
    });

  return (
    <ReactAriaProvider
      values={[
        [RichSelectBoxContext, { autoFocus: true }],
        [DialogContext, { padding: autosave ? undefined : 0 }], // this is altering Popover's internal dialog padding to show the sticky save/close buttons. Ideally this could be avoided
      ]}
    >
      <div
        className={classNames(styles.selectContainer, {
          [styles.opacityOverlay]: disabled,
          [styles.selectContainerCambio]: themeName === "cambio",
        })}
        onClick={onClick}
      >
        {label && (
          <>
            <ReactAriaLabel
              data-testid={[dataTestId, "label"].filter(Boolean).join("-")}
              className={classNames(
                themeName === "cambio" && styles.labelCambio,
              )}
              {...labelProps}
              onClick={() => {
                if (disabled) return;
                fieldRef.current?.focus();
                setInteractionModality("keyboard"); // Show the focus ring so the user knows where focus went
              }}
            >
              <Box paddingX={themeName === "classic" ? 1 : 3}>
                <Typography size={100} color="gray700">
                  {label}
                </Typography>
              </Box>
            </ReactAriaLabel>
          </>
        )}
        <Popover
          ref={overlayHandlerRef}
          disabled={disabled}
          content={
            // this Box wrapper is to reapply the padding that was stripped from popover's dialog to show the sticky save/close buttons. Ideally this could be avoided
            <Box
              padding={autosave ? undefined : 5}
              dangerouslySetInlineStyle={
                autosave ? undefined : { __style: { paddingBottom: 0 } }
              }
            >
              <RichSelectBox
                autosave={autosave}
                selectedValues={selectedKeys}
                defaultSelectedValues={defaultSelectedKeys}
                onChange={(selected) => setSelectedKeys(new Set(selected))}
                {...richSelectBoxProps}
                accessibilityLabel={inputId}
              >
                {children}
              </RichSelectBox>
            </Box>
          }
        >
          <TapArea
            data-testid={dataTestId}
            disabled={disabled}
            onClick={onClick}
            rounding={size === "lg" ? "lg" : "md"}
            {...fieldProps}
            ref={fieldRef}
          >
            <div className={styles.selectWrapper}>
              <div
                className={classNames(
                  styles.selectBox,
                  themeName === "classic"
                    ? styles.selectBoxClassic
                    : styles.selectBoxCambio,
                  themeName === "classic" && styles[size],
                  {
                    [styles.unselected]:
                      !errorText &&
                      selectedKeys !== "all" &&
                      !selectedKeys.size,
                    [styles.selected]:
                      !errorText &&
                      (selectedKeys === "all" || selectedKeys.size),
                    [themeName === "classic"
                      ? styles.selectError
                      : styles.selectErrorCambio]: errorText,
                  },
                )}
              >
                {selectedTextValue}
              </div>
              <div className={styles.arrowIcon}>
                <svg
                  focusable="false"
                  aria-hidden="true"
                  viewBox="0 0 24 24"
                  width={iconSize[size]}
                >
                  <path
                    fill={
                      errorText ? ColorBaseDestructive700 : ColorBaseGray800
                    }
                    d="M15.88 9.29 12 13.17 8.12 9.29a.9959.9959 0 0 0-1.41 0c-.39.39-.39 1.02 0 1.41l4.59 4.59c.39.39 1.02.39 1.41 0l4.59-4.59c.39-.39.39-1.02 0-1.41-.39-.38-1.03-.39-1.42 0z"
                  />
                </svg>
              </div>
            </div>
          </TapArea>
        </Popover>
        {(helperText || errorText) && (
          <Box paddingX={themeName === "classic" ? 1 : 0}>
            <Typography
              size={100}
              color={errorText ? "destructive-primary" : "gray700"}
              {...(errorText ? errorMessageProps : descriptionProps)}
            >
              {errorText ? errorText : helperText}
            </Typography>
          </Box>
        )}
      </div>
    </ReactAriaProvider>
  );
}

export default Object.assign(RichSelectList, {
  Section: RichSelectSection,
  Chip: RichSelectChip,
  RadioButton: RichSelectRadioButton,
});