import { expect, within } from "storybook/test";
import type { Meta, StoryObj } from "@storybook/react-vite";
import { SwatchesRegularIcon } from "alouette-icons/phosphor-icons/Swatches";
import { Text } from "../primitives/Text";
import { View } from "../primitives/View";
import { Story, accentsWithoutNeutral } from "../story-components/Story";
import {
  ConfirmationMessage,
  ErrorMessage,
  InfoMessage,
  Message,
  WarningMessage,
} from "./Message";

type ThisStory = StoryObj<typeof Message>;

const SIZES = ["sm", "md", "lg"] as const;

export default {
  title: "alouette/Feedback/Message",
  component: Message,
  parameters: {
    componentSubtitle: "A versatile component for displaying feedback to users",
  },
  argTypes: {
    accent: {
      description: "The accent of the message",
      control: "select",
      options: accentsWithoutNeutral,
      table: { defaultValue: { summary: "info" } },
    },
    onDismiss: {
      description: "Callback function when dismiss button is clicked",
      control: "boolean",
    },
    children: {
      description: "The content of the message",
      control: "text",
    },
  },
} satisfies Meta<typeof Message>;

export const PreviewMessageStory: ThisStory = {
  parameters: {
    layout: "padded",
    chromatic: { disableSnapshot: true },
  },
  args: { accent: "info" },
  render: (args) => (
    <Message {...args} icon={<SwatchesRegularIcon />}>
      <Text>Example Message</Text>
    </Message>
  ),
};

export const Variants: ThisStory = {
  render: () => (
    <Story>
      <Story.Section withSurface title="Defaults">
        <Message icon={<SwatchesRegularIcon />} accent="info">
          Example Message
        </Message>
        <Message
          icon={<SwatchesRegularIcon />}
          accent="info"
          dismissIconAriaLabel="Close message"
          onDismiss={() => {}}
        >
          Example Message with dismiss
        </Message>
      </Story.Section>

      <Story.Section withSurface title="Accents">
        {accentsWithoutNeutral.map((accent) => (
          <Message key={accent} icon={<SwatchesRegularIcon />} accent={accent}>
            {`${accent} message`}
          </Message>
        ))}
      </Story.Section>

      <Story.Section withSurface title="Sizes">
        {SIZES.map((size) => (
          <View key={size} className="gap-xs">
            <Message icon={<SwatchesRegularIcon />} accent="info" size={size}>
              {`Example ${size} Message`}
            </Message>
            <Message
              icon={<SwatchesRegularIcon />}
              accent="info"
              size={size}
              dismissIconAriaLabel="Close message"
              onDismiss={() => {}}
            >
              {`Example ${size} Message with dismiss`}
            </Message>
          </View>
        ))}
      </Story.Section>

      <Story.Section title="Elevation">
        <Story.SubSection title="surface (default), on the screen background">
          <InfoMessage>Raised above the page.</InfoMessage>
        </Story.SubSection>
        <Story.SubSection withSurface title="flat, inside a surface">
          <InfoMessage variant="flat">Info Message</InfoMessage>
          <ConfirmationMessage variant="flat">
            Success Message
          </ConfirmationMessage>
          <WarningMessage variant="flat">Warning Message</WarningMessage>
          <ErrorMessage variant="flat">Error Message</ErrorMessage>
        </Story.SubSection>
      </Story.Section>

      <Story.Section withSurface title="Accent Messages">
        <InfoMessage>Info Message</InfoMessage>
        <ConfirmationMessage>Success Message</ConfirmationMessage>
        <WarningMessage>Warning Message</WarningMessage>
        <ErrorMessage>Error Message</ErrorMessage>
      </Story.Section>

      <Story.Section withSurface title="Narrow container">
        <View className="w-55 gap-xs" aria-label="Narrow messages">
          {SIZES.map((size) => (
            <Message
              key={size}
              icon={<SwatchesRegularIcon />}
              accent="info"
              size={size}
              dismissIconAriaLabel="Close message"
              onDismiss={() => {}}
            >
              {`Narrow ${size} message with a long enough text to wrap`}
            </Message>
          ))}
        </View>
      </Story.Section>

      <Story.Section withSurface title="Edge Cases">
        <Message icon={<SwatchesRegularIcon />} accent="info">
          Example Message with very very very very very very very very very very
          very very very very very very very very very very very very very long
          text
        </Message>

        <Message
          icon={<SwatchesRegularIcon />}
          accent="info"
          dismissIconAriaLabel="Close message"
          onDismiss={() => {}}
        >
          Dismissable Example Message with very very very very very very very
          very very very very very very very very very very very very very very
          very long text
        </Message>
      </Story.Section>
    </Story>
  ),
  play: async ({ canvasElement }) => {
    // A wrapping text must not squeeze the leading icon below its size.
    // Story renders its children once per mode, hence getAll.
    for (const narrow of within(canvasElement).getAllByLabelText(
      "Narrow messages",
    )) {
      // Two svgs per message: the leading icon, then the dismiss icon.
      const [smIcon, , mdIcon, , lgIcon] = narrow.querySelectorAll("svg");
      if (!smIcon || !mdIcon || !lgIcon) {
        throw new Error("Expected a leading icon in each narrow message");
      }
      await expect(smIcon.getBoundingClientRect().width).toBe(20);
      await expect(mdIcon.getBoundingClientRect().width).toBe(24);
      await expect(lgIcon.getBoundingClientRect().width).toBe(28);

      // ...and the text must wrap rather than push the dismiss button out.
      for (const dismiss of narrow.querySelectorAll("button")) {
        await expect(dismiss.getBoundingClientRect().right).toBeLessThanOrEqual(
          narrow.getBoundingClientRect().right,
        );
      }
    }
  },
};