import {
  getArgs,
  getArgTypes,
  getSlotNames,
  getSlotsFragment,
  getDocsDescription,
} from "../../utils/storybook";

import UDot from "../../ui.other-dot/UDot.vue";
import URow from "../../ui.container-row/URow.vue";
import UBadge from "../../ui.text-badge/UBadge.vue";

import type { Meta, StoryFn } from "@storybook/vue3-vite";
import type { Props } from "../types";

interface UDotArgs extends Props {
  slotTemplate?: string;
  enum: "size" | "color";
}

export default {
  id: "100010",
  title: "Other / Dot",
  component: UDot,
  argTypes: {
    ...getArgTypes(UDot.__name),
  },
  parameters: {
    docs: {
      ...getDocsDescription(UDot.__name),
    },
  },
} as Meta;

const DefaultTemplate: StoryFn<UDotArgs> = (args: UDotArgs) => ({
  components: { UDot },
  setup: () => ({ args, slots: getSlotNames(UDot.__name) }),
  template: `
    <UDot v-bind="args">
      ${args.slotTemplate || getSlotsFragment("")}
    </UDot>
  `,
});

const EnumTemplate: StoryFn<UDotArgs> = (args: UDotArgs, { argTypes }) => ({
  components: { URow, UDot, UBadge },
  setup: () => ({ args, argTypes, getArgs }),
  template: `
    <URow>
      <UDot
        v-for="option in argTypes?.[args.enum]?.options"
        v-bind="getArgs(args, option)"
        :key="option"
      />
    </URow>
  `,
});

export const Default = DefaultTemplate.bind({});
Default.args = {};

export const Colors = EnumTemplate.bind({});
Colors.args = { enum: "color" };

export const Sizes = EnumTemplate.bind({});
Sizes.args = { enum: "size" };