import { expect, screen, userEvent, waitFor, within } from "storybook/test";
import type { Meta, StoryObj } from "@storybook/react-vite";
import {
  ArrowLeftDuotoneIcon,
  ArrowLeftRegularIcon,
} from "alouette-icons/phosphor-icons/ArrowLeft";
import type { ReactNode } from "react";
import { Text } from "../primitives/Text";
import { View } from "../primitives/View";
import { Story, neutralAndAccents } from "../story-components/Story";
import { StoryGrid, stateTitle } from "../story-components/StoryGrid";
import { IconButton, type IconButtonProps } from "./IconButton";

type ThisStory = StoryObj<typeof IconButton>;

export default {
  title: "alouette/Actions/IconButton",
  component: IconButton,
  parameters: {
    componentSubtitle:
      "A standardized circular icon button with consistent touch targets, visual feedback, and accessibility",
    docs: {
      description: {
        component: `### Variants
- \`size\`: \`sm\` | \`md\` | any number (custom diameter px)
- \`iconSize\`: \`"fill"\` makes the icon fill 80% of the button (default 50%)
- \`variant\`: tonal | filled | outlined | soft (the glyph takes the accent on hover/focus)
- \`activeIcon\`: replaces \`icon\` while the button is hovered, focused or pressed — usually the duotone twin
- Wrap in \`<AccentTheme accent="brand"/>\` (or any accent: brand|info|success|warning|danger) to switch the interactive token set; it composes with current light/dark mode`,
      },
    },
  },
} satisfies Meta<typeof IconButton>;

export const PreviewIconButtonStory: ThisStory = {
  parameters: {
    layout: "padded",
    chromatic: { disableSnapshot: true },
  },
  args: {
    size: "md",
    "aria-label": "Go back",
  },
  render: (args) => (
    <IconButton
      {...args}
      icon={<ArrowLeftRegularIcon />}
      activeIcon={<ArrowLeftDuotoneIcon />}
    />
  ),
};

interface BackIconButtonProps {
  forceStyle?: IconButtonProps["forceStyle"];
  /** Renders `icon` alone, so the glyph keeps one weight in every state. */
  withoutActiveIcon?: boolean;
}

function BackIconButton({
  forceStyle,
  withoutActiveIcon,
}: BackIconButtonProps): ReactNode {
  return (
    <IconButton
      forceStyle={forceStyle}
      icon={<ArrowLeftRegularIcon />}
      activeIcon={withoutActiveIcon ? undefined : <ArrowLeftDuotoneIcon />}
      aria-label="Go back"
    />
  );
}

function IconWeightRow({
  withoutActiveIcon,
}: Pick<BackIconButtonProps, "withoutActiveIcon">): ReactNode {
  return (
    <StoryGrid.Row>
      <StoryGrid.Col title="default">
        <BackIconButton withoutActiveIcon={withoutActiveIcon} />
      </StoryGrid.Col>
      <StoryGrid.Col title="hover">
        <BackIconButton
          forceStyle="hover"
          withoutActiveIcon={withoutActiveIcon}
        />
      </StoryGrid.Col>
      <StoryGrid.Col title="focus-visible">
        <BackIconButton
          forceStyle="focus"
          withoutActiveIcon={withoutActiveIcon}
        />
      </StoryGrid.Col>
      <StoryGrid.Col title="press">
        <BackIconButton
          forceStyle="press"
          withoutActiveIcon={withoutActiveIcon}
        />
      </StoryGrid.Col>
    </StoryGrid.Row>
  );
}

export const Variants: ThisStory = {
  render: () => (
    <Story>
      <Story.Section title="Sizes">
        <StoryGrid.Row>
          {([24, 40] as const).map((size) => (
            <StoryGrid.Col key={size} title={String(size)}>
              <IconButton
                size={size}
                icon={<ArrowLeftRegularIcon />}
                activeIcon={<ArrowLeftDuotoneIcon />}
                aria-label="Go back"
              />
            </StoryGrid.Col>
          ))}
        </StoryGrid.Row>
      </Story.Section>

      <Story.Section title="Icon weight">
        <Story.SubSection title="activeIcon">
          <IconWeightRow />
        </Story.SubSection>
        <Story.SubSection title="Without activeIcon">
          <IconWeightRow withoutActiveIcon />
        </Story.SubSection>
      </Story.Section>

      <Story.Section title="Disabled with a reason">
        <IconButton
          icon={<ArrowLeftRegularIcon />}
          aria-label="Go back"
          disabledReason="Nothing to go back to"
        />
      </Story.Section>

      <Story.Section withSurface title="Variants">
        {neutralAndAccents.map((accent) => (
          <Story.SubSection key={accent} title={accent}>
            <StoryGrid.Row>
              {(
                [undefined, "hover", "focus", "press", "disabled"] as const
              ).map((state) => (
                <StoryGrid.Col key={state} title={stateTitle(state)}>
                  <View className="gap-xs">
                    {(["tonal", "filled", "soft"] as const).map((variant) => (
                      <View
                        key={variant}
                        className="flex-row gap-xs items-center"
                      >
                        <IconButton
                          accent={accent}
                          variant={variant}
                          disabled={state === "disabled"}
                          forceStyle={state === "disabled" ? undefined : state}
                          icon={<ArrowLeftRegularIcon />}
                          activeIcon={<ArrowLeftDuotoneIcon />}
                          aria-label="Go back"
                        />
                        <Text className="text-xs">{variant}</Text>
                      </View>
                    ))}
                  </View>
                </StoryGrid.Col>
              ))}
            </StoryGrid.Row>
          </Story.SubSection>
        ))}
      </Story.Section>
    </Story>
  ),
};

/**
 * A soft icon button has no ground to carry its accent at rest, so the glyph
 * takes `text-accent` on hover and focus.
 */
export const SoftAccentOnInteraction: ThisStory = {
  render: () => (
    <Story noDarkMode>
      <Story.Section withSurface title="Soft, danger">
        <StoryGrid.Row>
          <StoryGrid.Col title="Interactive">
            <IconButton
              accent="danger"
              variant="soft"
              icon={<ArrowLeftRegularIcon />}
              aria-label="Go back"
            />
          </StoryGrid.Col>
          <StoryGrid.Col title="hover">
            <IconButton
              accent="danger"
              variant="soft"
              forceStyle="hover"
              icon={<ArrowLeftRegularIcon />}
              aria-label="Go back (hover)"
            />
          </StoryGrid.Col>
        </StoryGrid.Row>
      </Story.Section>
    </Story>
  ),
  play: async ({ canvasElement }) => {
    const canvas = within(canvasElement);
    const button = canvas.getByRole("button", { name: "Go back" });
    const glyph = button.querySelector("svg")!;
    const accentColor = getComputedStyle(
      canvas
        .getByRole("button", { name: "Go back (hover)" })
        .querySelector("svg")!,
    ).color;
    const restColor = getComputedStyle(glyph).color;

    await expect(restColor).not.toBe(accentColor);

    // user-event's hover never sets CSS `:hover`, so only focus is asserted;
    // the forced `hover` button covers the hover appearance.
    await userEvent.tab();
    await expect(button).toHaveFocus();
    await waitFor(() =>
      expect(getComputedStyle(glyph).color).toBe(accentColor),
    );
  },
};

export const DisabledReason: ThisStory = {
  render: () => (
    <Story noDarkMode>
      <Story.Section title="Disabled with a reason">
        <IconButton
          icon={<ArrowLeftRegularIcon />}
          aria-label="Go back"
          disabledReason="Nothing to go back to"
        />
      </Story.Section>
    </Story>
  ),
  play: async ({ canvasElement }) => {
    const button = within(canvasElement).getByRole("button", {
      name: "Go back",
    });
    await expect(button).toHaveAttribute("aria-disabled", "true");
    await expect(button).toHaveAccessibleDescription("Nothing to go back to");

    await userEvent.hover(button);
    await expect(
      await screen.findByRole("tooltip", { hidden: true }),
    ).toHaveTextContent("Nothing to go back to");
    await userEvent.unhover(button);
    await waitFor(() =>
      expect(
        screen.queryByRole("tooltip", { hidden: true }),
      ).not.toBeInTheDocument(),
    );
  },
};