yaak/src-web/components/EnvironmentActionsDropdown.tsx

import classNames from 'classnames';
import { memo, useMemo } from 'react';
import { Button } from './core/Button';
import type { DropdownItem } from './core/Dropdown';
import { Dropdown } from './core/Dropdown';
import { Icon } from './core/Icon';
import { InlineCode } from './core/InlineCode';
import { useEnvironments } from '../hooks/useEnvironments';
import { useActiveEnvironment } from '../hooks/useActiveEnvironment';
import { useUpdateEnvironment } from '../hooks/useUpdateEnvironment';
import { useCreateEnvironment } from '../hooks/useCreateEnvironment';
import { usePrompt } from '../hooks/usePrompt';
import { useDialog } from './DialogContext';
import { EnvironmentEditDialog } from './EnvironmentEditDialog';
import { useAppRoutes } from '../hooks/useAppRoutes';

type Props = {
  className?: string;
};

export const EnvironmentActionsDropdown = memo(function EnvironmentActionsDropdown({
  className,
}: Props) {
  const environments = useEnvironments();
  const activeEnvironment = useActiveEnvironment();
  const updateEnvironment = useUpdateEnvironment(activeEnvironment?.id ?? null);
  const createEnvironment = useCreateEnvironment();
  const prompt = usePrompt();
  const dialog = useDialog();
  const routes = useAppRoutes();

  const items: DropdownItem[] = useMemo(() => {
    const environmentItems = environments.map(
      (e) => ({
        key: e.id,
        label: e.name,
        rightSlot: e.id === activeEnvironment?.id ? <Icon icon="check" /> : undefined,
        onSelect: async () => {
          routes.setEnvironment(e);
        },
      }),
      [activeEnvironment?.id],
    );
    const activeEnvironmentItems: DropdownItem[] =
      environments.length <= 1
        ? []
        : [
            ...environmentItems,
            {
              type: 'separator',
              label: activeEnvironment?.name,
            },
          ];

    return [
      ...activeEnvironmentItems,
      {
        key: 'edit',
        label: 'Edit',
        leftSlot: <Icon icon="sun" />,
        onSelect: async () => {
          dialog.show({
            title: 'Environments',
            render: () => <EnvironmentEditDialog />,
          });
        },
      },
      {
        key: 'rename',
        label: 'Rename',
        leftSlot: <Icon icon="pencil" />,
        onSelect: async () => {
          const name = await prompt({
            title: 'Rename Environment',
            description: (
              <>
                Enter a new name for <InlineCode>{activeEnvironment?.name}</InlineCode>
              </>
            ),
            name: 'name',
            label: 'Name',
            defaultValue: activeEnvironment?.name,
          });
          updateEnvironment.mutate({ name });
        },
      },
      // {
      //   key: 'delete',
      //   label: 'Delete',
      //   leftSlot: <Icon icon="trash" />,
      //   onSelect: deleteEnv.mutate,
      //   variant: 'danger',
      // },
      { type: 'separator' },
      {
        key: 'create-environment',
        label: 'Create Environment',
        leftSlot: <Icon icon="plus" />,
        onSelect: async () => {
          const name = await prompt({
            name: 'name',
            label: 'Name',
            defaultValue: 'My Environment',
            description: 'Enter a name for the new environment',
            title: 'Create Environment',
          });
          createEnvironment.mutate({ name });
        },
      },
    ];
  }, [
    // deleteEnvironment.mutate,
    activeEnvironment?.name,
    createEnvironment,
    dialog,
    environments,
    prompt,
    updateEnvironment,
    routes,
  ]);

  return (
    <Dropdown items={items}>
      <Button
        size="sm"
        className={classNames(className, 'text-gray-800 !px-2 truncate')}
        forDropdown
      >
        {activeEnvironment?.name ?? <span className="italic text-gray-500">No Environment</span>}
      </Button>
    </Dropdown>
  );
});