import React from 'react';
import { MenuContext, Provider, RootMenuTriggerStateContext } from 'react-aria-components';

export const DropdownButtonListContext = React.createContext<{
  closeOnAction: boolean;
} | null>(null);

interface DropdownButtonListContextProviderProps extends React.PropsWithChildren {
  closeOnAction: boolean;
}

export const DropdownButtonListContextProvider: React.FC<DropdownButtonListContextProviderProps> = ({
  children,
  closeOnAction,
  ...rest
}) => {
  rest satisfies Record<PropertyKey, never>;

  const menuContext = React.useContext(MenuContext)!;
  const rootMenuTriggerState = React.useContext(RootMenuTriggerStateContext)!;

  return (
    <DropdownButtonListContext.Provider value={{ closeOnAction }}>
      <Provider
        values={[
          [
            MenuContext,
            {
              ...menuContext,
              onClose: () => {
                if (closeOnAction) {
                  rootMenuTriggerState.setOpen(false);
                }
              },
            },
          ],
        ]}
      >
        {children}
      </Provider>
    </DropdownButtonListContext.Provider>
  );
};