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

import USelect from "../../ui.form-select/USelect.vue";
import URow from "../../ui.container-row/URow.vue";
import UCol from "../../ui.container-col/UCol.vue";
import UBadge from "../../ui.text-badge/UBadge.vue";
import UIcon from "../../ui.image-icon/UIcon.vue";
import ULink from "../../ui.button-link/ULink.vue";
import UAvatar from "../../ui.image-avatar/UAvatar.vue";
import UText from "../../ui.text-block/UText.vue";
import ULoader from "../../ui.loader/ULoader.vue";

import johnDoe from "./assets/images/john-doe.png";
import emilyDavis from "./assets/images/emily-davis.png";
import alexJohnson from "./assets/images/alex-johnson.png";
import patMorgan from "./assets/images/pat-morgan.png";

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

interface USelectArgs extends Props {
  slotTemplate?: string;
  enum: "size" | "openDirection" | "labelAlign";
  wrapperClass?: string;
}

export default {
  id: "3080",
  title: "Form Inputs & Controls / Select",
  component: USelect,
  args: {
    label: "Choose a city",
    modelValue: null,
    options: [
      { label: "New York", value: 1 },
      { label: "Los Angeles", value: 2 },
      { label: "Chicago", value: 3 },
      { label: "Houston", value: 4 },
      { label: "San Francisco", value: 5 },
    ],
  },
  argTypes: {
    ...getArgTypes(USelect.__name),
  },
  parameters: {
    docs: {
      ...getDocsDescription(USelect.__name),
      story: {
        height: "300px",
      },
    },
  },
} as Meta;

const DefaultTemplate: StoryFn<USelectArgs> = (args: USelectArgs) => ({
  components: { USelect, UIcon, ULink, UText, ULoader, URow },
  setup: () => ({ args, slots: getSlotNames(USelect.__name) }),
  template: `
    <USelect
      v-bind="args"
      v-model="args.modelValue"
      class="max-w-96"
    >
      ${args.slotTemplate || getSlotsFragment("")}
    </USelect>
  `,
});

const EnumTemplate: StoryFn<USelectArgs> = (args: USelectArgs, { argTypes }) => ({
  components: { USelect, UCol },
  setup: () => ({ args, argTypes, getArgs }),
  template: `
    <UCol :class="args.wrapperClass">
      <USelect
        v-for="option in argTypes?.[args.enum]?.options"
        v-bind="getArgs(args, option)"
        :key="option"
        v-model="args.modelValue"
        class="max-w-96"
      />
    </UCol>
  `,
});

const GroupValuesTemplate: StoryFn<USelectArgs> = (args: USelectArgs) => ({
  components: { USelect },
  setup() {
    return {
      args,
    };
  },
  template: `
    <USelect
      v-bind="args"
      v-model="args.modelValue"
      label="Single"
      class="max-w-96"
    />

    <USelect
      v-bind="args"
      v-model="args.modelValueMultiple"
      label="Multiple"
      multiple
      class="mt-5 max-w-96"
    />
  `,
});

export const Default = DefaultTemplate.bind({});
Default.args = {};
Default.parameters = {
  docs: {
    story: {
      height: "350px",
    },
  },
};

export const Placeholder = DefaultTemplate.bind({});
Placeholder.args = { placeholder: "Start typing to search for a city..." };

export const Description = DefaultTemplate.bind({});
Description.args = { description: "You can only select a city from the list." };

export const Error: StoryFn<USelectArgs> = (args: USelectArgs) => ({
  components: { USelect },
  setup: () => ({ args }),
  template: `
    <USelect
      v-bind="args"
      v-model="args.modelValue"
      class="max-w-96"
      :error="args.modelValue ? '' : 'Please select a city from the list.'"
    />
  `,
});

export const NotClearable = DefaultTemplate.bind({});
NotClearable.args = { clearable: false };

export const Searchable = DefaultTemplate.bind({});
Searchable.args = { searchable: true };
Searchable.parameters = {
  docs: {
    story: {
      height: "350px",
    },
  },
};

export const SearchModelValue = DefaultTemplate.bind({});
SearchModelValue.args = { search: "New York", searchable: true };
SearchModelValue.parameters = {
  docs: {
    story: {
      height: "350px",
    },
  },
};

export const NoCloseOnSelect = DefaultTemplate.bind({});
NoCloseOnSelect.args = { modelValue: 3, closeOnSelect: false };

export const Readonly = DefaultTemplate.bind({});
Readonly.args = { readonly: true, modelValue: 1, clearable: false };

export const Disabled = DefaultTemplate.bind({});
Disabled.args = { disabled: true, modelValue: 2, clearable: false };

export const LabelAlign = EnumTemplate.bind({});
LabelAlign.args = {
  enum: "labelAlign",
  description: "Select a city from the list.",
  wrapperClass: "gap-16",
};

export const Sizes = EnumTemplate.bind({});
Sizes.args = { enum: "size", multiple: true, modelValue: [], label: "{enumValue}" };

export const LargeItemList = DefaultTemplate.bind({});
LargeItemList.args = {
  options: [...new Array(1000)].map((_, index) => {
    return { value: index + 1, label: `value ${index + 1}`, badge: "badge" };
  }),
};
LargeItemList.parameters = {
  docs: {
    story: {
      height: "400px",
    },
  },
};

export const Multiple = DefaultTemplate.bind({});
Multiple.args = { multiple: true, modelValue: [] };

export const OpenDirection = EnumTemplate.bind({});
OpenDirection.args = { enum: "openDirection", label: "{enumValue}" };

export const GroupValue = GroupValuesTemplate.bind({});
GroupValue.args = {
  modelValue: "",
  groupValueKey: "libs",
  groupLabelKey: "language",
  labelKey: "name",
  valueKey: "name",
  // TODO: Implement recursion nesting and add ability to use group (nested) options with default options.
  options: [
    {
      language: "Javascript",
      libs: [{ name: "Vue.js" }, { name: "Adonis" }],
    },
    {
      language: "Ruby",
      libs: [
        { name: "Frameworks", isSubGroup: true, level: 2 },
        { name: "Rails", level: 3 },
        { name: "Sinatra", level: 3 },
      ],
    },
    {
      language: "Other",
      libs: [{ name: "Laravel" }, { name: "Phoenix" }],
    },
  ],
};
GroupValue.parameters = {
  docs: {
    story: {
      height: "500px",
    },
  },
};

export const OptionsLimit = DefaultTemplate.bind({});
OptionsLimit.args = { optionsLimit: 2 };
OptionsLimit.parameters = {
  docs: {
    description: {
      story: "`optionsLimit` prop controls the number of options displayed in the dropdown.",
    },
  },
};

export const VisibleOptions = DefaultTemplate.bind({});
VisibleOptions.args = { visibleOptions: 3 };
VisibleOptions.parameters = {
  docs: {
    description: {
      story: "`visibleOptions` prop controls the number of options you can see without a scroll.",
    },
  },
};

export const AddOption: StoryFn<USelectArgs> = (args: USelectArgs) => ({
  components: { USelect },
  setup: () => ({ args, showAlert: (message: string) => alert(message) }),
  template: `
    <USelect
      v-bind="args"
      v-model="args.modelValue"
      class="max-w-96"
      add-option
      @add="showAlert('You triggered the add action!')"
    />
  `,
});
AddOption.parameters = {
  docs: {
    description: {
      story:
        // eslint-disable-next-line vue/max-len
        "The `addOption` prop displays an 'Add option' button, while the `add` event allows handling custom functionality when the button is clicked.",
    },
  },
};

export const OptionSettings = DefaultTemplate.bind({});
OptionSettings.args = {
  options: [
    { label: "1. New York", value: "1" },
    { label: "2. Los Angeles", value: "2", isHvalueden: true },
    {
      label: "3. Chicago",
      value: "3",
      onClick: (option) =>
        alert("onClick function for the third option: " + JSON.stringify(option)),
    },
    { label: "4. Houston", value: "4" },
    { label: "5. San Francisco", value: "5" },
  ],
};
OptionSettings.parameters = {
  docs: {
    description: {
      story:
        // eslint-disable-next-line vue/max-len
        "The second option of the array is hidden (`isHidden` object property is set to `true`). <br/> The third option has `onClick` event handler: <br/> `onClick: (option: Option) => alert('onClick function for option 3: ' + JSON.stringify(option))`",
    },
  },
};

export const IconProps: StoryFn<USelectArgs> = (args) => ({
  components: { USelect, URow },
  setup() {
    const levelOptions = [
      { label: "Awesome", value: "1" },
      { label: "Good", value: "2" },
      { label: "Could be better", value: "3" },
      { label: "Terrible", value: "4" },
    ];

    const roleOptions = [
      { label: "Admin", value: "1" },
      { label: "CEO", value: "2" },
      { label: "Manager", value: "3" },
      { label: "Guest", value: "4" },
    ];

    return { args, levelOptions, roleOptions };
  },
  template: `
    <URow>
      <USelect
        v-model="args.modelValueLeft"
        left-icon="feedback"
        label="Choose the level of our services"
        placeholder="Share your feedback with us"
        :options="levelOptions"
      />
      <USelect
        v-model="args.modelValueRight"
        right-icon="person"
        label="Select your role"
        placeholder="Choose a role from the list"
        :options="roleOptions"
      />
    </URow>
  `,
});

export const Slots: StoryFn<USelectArgs> = (args) => ({
  components: { USelect, URow, UCol, UIcon, UText, ULink },
  setup: () => ({
    args,
    leftSlotModel: ref("paypal"),
    rightSlotModel: ref("bank"),
    descriptionSlotModel: ref(null),
    errorSlotModel: ref(null),
  }),
  template: `
    <UCol gap="3xl">
      <URow block>
        <USelect
          v-model="leftSlotModel"
          label="Select Payment Method"
          :options="[
            { label: 'Visa', value: 'visa', icon: 'credit_card', details: '•••• 4242' },
            { label: 'PayPal', value: 'paypal', icon: 'payments', details: 'user@example.com' },
            { label: 'Bank Transfer', value: 'bank', icon: 'account_balance', details: 'Acct **** 1234' },
            { label: 'Apple Pay', value: 'apple', icon: 'phone_iphone', details: 'iPhone 15' },
          ]"
        >
          <template #left="{ options }">
            <UIcon
              v-if="leftSlotModel"
              :name="options?.icon"
              color="primary"
              size="sm"
            />
          </template>
        </USelect>

        <USelect
          v-model="rightSlotModel"
          label="Select Payment Method"
          :options="[
            { label: 'Visa', value: 'visa', icon: 'credit_card', details: '•••• 4242' },
            { label: 'PayPal', value: 'paypal', icon: 'payments', details: 'user@example.com' },
            { label: 'Bank Transfer', value: 'bank', icon: 'account_balance', details: 'Acct **** 1234' },
            { label: 'Apple Pay', value: 'apple', icon: 'phone_iphone', details: 'iPhone 15' },
          ]"
        >
          <template #right="{ options }">
            <UText
              v-if="rightSlotModel"
              size="sm"
              variant="lifted"
              class="text-nowrap"
            >
              {{ options?.details }}
            </UText>
          </template>
        </USelect>
      </URow>

      <URow block>
        <USelect
          v-bind="args"
          v-model="descriptionSlotModel"
          label="City"
          :options="args.options"
        >
          <template #description>
            <UText size="sm" variant="lifted">
              Only cities we ship to, see
              <ULink label="shipping zones" underlined size="sm" />.
            </UText>
          </template>
        </USelect>

        <USelect
          v-bind="args"
          v-model="errorSlotModel"
          label="City"
          :options="args.options"
          :error="true"
        >
          <template #error>
            <UText size="sm" color="error">
              <ul>
                <li>Please select an option from the list</li>
                <li>The current value is not available anymore</li>
                <li>Pick a valid choice to continue</li>
              </ul>
            </UText>
          </template>
        </USelect>
      </URow>
    </UCol>
  `,
});
Slots.parameters = {
  docs: {
    story: {
      height: "400px",
    },
  },
};

export const ToggleSlots: StoryFn<USelectArgs> = (args) => ({
  components: { USelect, URow, UIcon },
  setup() {
    const beforeToggleModel = ref(null);
    const toggleModel = ref(null);
    const afterToggleModel = ref(null);

    return { args, beforeToggleModel, toggleModel, afterToggleModel };
  },
  template: `
    <URow>
      <USelect
        v-model="beforeToggleModel"
        label="Before Toggle Slot"
        :options="[
          { label: 'John Doe', value: '1' },
          { label: 'Jane Smith', value: '2' },
          { label: 'Mike Johnson', value: '3' },
        ]"
      >
        <template #before-toggle>
          <UIcon
            name="person"
            color="primary"
            size="sm"
          />
        </template>
      </USelect>

      <USelect
        v-model="toggleModel"
        label="Toggle Slot"
        :options="[
          { label: 'High', value: 'high' },
          { label: 'Medium', value: 'medium' },
          { label: 'Low', value: 'low' }
        ]"
      >
        <template #toggle="{ opened }">
          <UIcon
            name="expand_circle_down"
            :class="{ 'rotate-180': opened }"
            color="primary"
            size="sm"
          />
        </template>
      </USelect>

      <USelect
        v-model="afterToggleModel"
        label="After Toggle Slot"
        :options="[
          { label: 'In Progress', value: 'in_progress' },
          { label: 'Done', value: 'done' },
          { label: 'Blocked', value: 'blocked' }
        ]"
      >
        <template #after-toggle>
          <UIcon
            name="info"
            color="primary"
            size="sm"
          />
        </template>
      </USelect>
    </URow>
  `,
});

export const ClearSlot = DefaultTemplate.bind({});
ClearSlot.args = {
  slotTemplate: `
    <template #clear="{ clear }">
      <ULink label="clear" @click="clear" class="mr-1" />
    </template>
  `,
};

export const SelectedOptionsSlots: StoryFn<USelectArgs> = (args) => ({
  components: { USelect, URow, UIcon, UText },
  setup: () => ({ args }),
  template: `
    <URow>
      <USelect
        v-model="args.selectedOption"
        label="Selected option slot"
        :options="[
          { label: 'Paris', value: '1', icon: 'flight' },
          { label: 'Venice', value: '2', icon: 'sailing' },
          { label: 'Rome', value: '3', icon: 'directions_car' },
          { label: 'Milan', value: '4', icon: 'directions_bike' },
        ]"
      >
        <template #selected-option="{ option }">
          <URow align="center" gap="2xs">
            <UText>{{ option.label }}</UText>
            <UIcon
              :name="option.icon"
              size="2xs"
              color="success"
            />
          </URow>
        </template>
      </USelect>

      <USelect
        v-model="args.selectedOptions"
        label="Selected options slot"
        :options="[
          { label: 'Paris', value: '1', icon: 'flight' },
          { label: 'Venice', value: '2', icon: 'sailing' },
          { label: 'Rome', value: '3', icon: 'directions_car' },
          { label: 'Milan', value: '4', icon: 'directions_bike' },
        ]"
        multiple
      >
        <template #selected-options="{ options }">
          <URow
            v-for="(option, index) in options"
            :key="index"
            align="center"
            gap="2xs"
          >
            <UText line>{{ option.label }}</UText>
            <UIcon
              :name="option.icon"
              size="2xs"
              color="success"
            />
            <UIcon
              v-if="index !== options.length - 1"
              name="east"
              size="2xs"
              class="mr-1"
            />
          </URow>
        </template>
      </USelect>
    </URow>
  `,
});
SelectedOptionsSlots.parameters = {
  docs: {
    description: {
      story:
        // eslint-disable-next-line vue/max-len
        "The `selected-option` and `selected-options` slots allow you to customize the selected option or options display. <br/> In this example, we've added an icon to the selected option and a separator between options in the `selected-options` slot.",
    },
  },
};

export const SelectedCounterSlot = DefaultTemplate.bind({});
SelectedCounterSlot.args = {
  multiple: true,
  slotTemplate: `
    <template #selected-counter="{ count }">
      <UText v-if="count">, and {{ count }} more variant(s)</UText>
    </template>
  `,
};

export const EmptySlot = DefaultTemplate.bind({});
EmptySlot.args = {
  options: [],
  slotTemplate: `
    <template #empty>
      <URow align="center">
        <ULoader loading size="sm" />
        <UText label="Loading, this may take a while..." />
      </URow>
    </template>
  `,
};

export const OptionSlots: StoryFn<USelectArgs> = (args) => ({
  components: { USelect, URow, UCol, UAvatar, UIcon, UBadge, UText },
  setup: () => ({ args, johnDoe, emilyDavis, alexJohnson, patMorgan }),
  template: `
    <URow>
      <USelect
        v-model="args.beforeOptionModel"
        label="Before option slot"
        :options="[
          {
            label: 'John Doe',
            value: '1',
            role: 'Developer',
            avatar: johnDoe,
            status: 'online',
            statusColor: 'success',
          },
          {
            label: 'Jane Smith',
            value: '2',
            role: 'Designer',
            avatar: emilyDavis,
            status: 'away',
            statusColor: 'warning',
          },
          {
            label: 'Mike Johnson',
            value: '3',
            role: 'Product Manager',
            avatar: alexJohnson,
            status: 'offline',
            statusColor: 'grayscale',
          },
          {
            label: 'Sarah Wilson',
            value: '4',
            role: 'QA Engineer',
            avatar: patMorgan,
            status: 'online',
            statusColor: 'success',
          },
        ]"
      >
        <template #before-option="{ option }">
          <UAvatar :src="option.avatar" size="sm" />
        </template>
      </USelect>

      <USelect
        v-model="args.optionModel"
        label="Option slot"
        :options="[
          {
            label: 'John Doe',
            value: '1',
            role: 'Developer',
            avatar: johnDoe,
            status: 'online',
            statusColor: 'success',
          },
          {
            label: 'Jane Smith',
            value: '2',
            role: 'Designer',
            avatar: emilyDavis,
            status: 'away',
            statusColor: 'warning',
          },
          {
            label: 'Mike Johnson',
            value: '3',
            role: 'Product Manager',
            avatar: alexJohnson,
            status: 'offline',
            statusColor: 'grayscale',
          },
          {
            label: 'Sarah Wilson',
            value: '4',
            role: 'QA Engineer',
            avatar: patMorgan,
            status: 'online',
            statusColor: 'success',
          },
        ]"
      >
        <template #option="{ option }">
          <URow align="center" gap="xs">
            <UCol gap="none">
              <UText size="sm">{{ option.label }}</UText>
              <UText variant="lifted" size="xs">{{ option.role }}</UText>
            </UCol>
            <UBadge
              :label="option.status"
              :color="option.statusColor"
              size="sm"
              variant="subtle"
            />
          </URow>
        </template>
      </USelect>

      <USelect
        v-model="args.afterOptionModel"
        label="After option slot"
        :options="[
          {
            label: 'John Doe',
            value: '1',
            role: 'Developer',
            avatar: johnDoe,
            status: 'online',
            statusColor: 'success',
          },
          {
            label: 'Jane Smith',
            value: '2',
            role: 'Designer',
            avatar: emilyDavis,
            status: 'away',
            statusColor: 'warning',
          },
          {
            label: 'Mike Johnson',
            value: '3',
            role: 'Product Manager',
            avatar: alexJohnson,
            status: 'offline',
            statusColor: 'grayscale',
          },
          {
            label: 'Sarah Wilson',
            value: '4',
            role: 'QA Engineer',
            avatar: patMorgan,
            status: 'online',
            statusColor: 'success',
          },
        ]"
      >
        <template #after-option="{ option }">
          <UBadge
            :label="option.status"
            :color="option.statusColor"
            size="sm"
            variant="subtle"
          />
        </template>
      </USelect>
    </URow>
  `,
});