import * as React from "react";
import {act, render} from "@testing-library/react";
import {Button} from "@nextui-org/button";
import userEvent from "@testing-library/user-event";
import {User} from "@nextui-org/user";

import {Dropdown, DropdownTrigger, DropdownMenu, DropdownItem, DropdownSection} from "../src";

// e.g. console.error Warning: Function components cannot be given refs.
// Attempts to access this ref will fail. Did you mean to use React.forwardRef()?
const spy = jest.spyOn(console, "error").mockImplementation(() => {});

describe("Dropdown", () => {
  afterEach(() => {
    jest.clearAllMocks();
  });

  it("should render correctly (static)", () => {
    const wrapper = render(
      <Dropdown>
        <DropdownTrigger>
          <Button>Trigger</Button>
        </DropdownTrigger>
        <DropdownMenu aria-label="Actions" onAction={alert}>
          <DropdownItem key="new">New file</DropdownItem>
          <DropdownItem key="copy">Copy link</DropdownItem>
          <DropdownItem key="edit">Edit file</DropdownItem>
          <DropdownItem key="delete" color="danger">
            Delete file
          </DropdownItem>
        </DropdownMenu>
      </Dropdown>,
    );

    expect(() => wrapper.unmount()).not.toThrow();
  });

  it("should render correctly (dynamic)", () => {
    const menuItems = [
      {key: "new", name: "New File"},
      {key: "copy", name: "Copy Link"},
      {key: "edit", name: "Edit File"},
      {key: "delete", name: "Delete File"},
    ];

    const wrapper = render(
      <Dropdown>
        <DropdownTrigger>
          <Button>Trigger</Button>
        </DropdownTrigger>
        <DropdownMenu aria-label="Actions" items={menuItems}>
          {(item: any) => <DropdownItem key={item.key}>{item.name}</DropdownItem>}
        </DropdownMenu>
      </Dropdown>,
    );

    expect(() => wrapper.unmount()).not.toThrow();
  });

  it("should render correctly with section (static)", () => {
    const wrapper = render(
      <Dropdown>
        <DropdownTrigger>
          <Button>Trigger</Button>
        </DropdownTrigger>
        <DropdownMenu aria-label="Actions" onAction={alert}>
          <DropdownSection title="Actions">
            <DropdownItem key="new">New file</DropdownItem>
            <DropdownItem key="copy">Copy link</DropdownItem>
          </DropdownSection>
          <DropdownSection title="Danger Zone">
            <DropdownItem key="edit">Edit file</DropdownItem>
            <DropdownItem key="delete" color="danger">
              Delete file
            </DropdownItem>
          </DropdownSection>
        </DropdownMenu>
      </Dropdown>,
    );

    expect(() => wrapper.unmount()).not.toThrow();
  });

  it("should render correctly with section (dynamic)", () => {
    const menuItems = [
      {
        title: "Actions",
        children: [
          {key: "new", name: "New File"},
          {key: "copy", name: "Copy Link"},
          {key: "edit", name: "Edit File"},
        ],
      },
      {
        title: "Danger Zone",
        children: [{key: "delete", name: "Delete File"}],
      },
    ];

    const wrapper = render(
      <Dropdown>
        <DropdownTrigger>
          <Button>Trigger</Button>
        </DropdownTrigger>
        <DropdownMenu aria-label="Actions" items={menuItems}>
          {(section: any) => (
            <DropdownSection
              aria-label={section.title}
              items={section.children}
              title={section.title}
            >
              {/* @ts-ignore */}
              {(item: any) => <DropdownItem key={item.key}>{item.name}</DropdownItem>}
            </DropdownSection>
          )}
        </DropdownMenu>
      </Dropdown>,
    );

    expect(() => wrapper.unmount()).not.toThrow();
  });

  it("should not throw any error when clicking button", async () => {
    const wrapper = render(
      <Dropdown>
        <DropdownTrigger>
          <Button data-testid="trigger-test">Trigger</Button>
        </DropdownTrigger>
        <DropdownMenu aria-label="Actions" onAction={alert}>
          <DropdownItem key="new">New file</DropdownItem>
          <DropdownItem key="copy">Copy link</DropdownItem>
          <DropdownItem key="edit">Edit file</DropdownItem>
          <DropdownItem key="delete" color="danger">
            Delete file
          </DropdownItem>
        </DropdownMenu>
      </Dropdown>,
    );

    let triggerButton = wrapper.getByTestId("trigger-test");

    expect(triggerButton).toBeTruthy();

    await act(async () => {
      await userEvent.click(triggerButton);
    });

    expect(spy).toBeCalledTimes(0);
  });

  it("should work with single selection (controlled)", async () => {
    let onOpenChange = jest.fn();
    let onSelectionChange = jest.fn();

    const wrapper = render(
      <Dropdown onOpenChange={onOpenChange}>
        <DropdownTrigger>
          <Button data-testid="trigger-test">Trigger</Button>
        </DropdownTrigger>
        <DropdownMenu
          disallowEmptySelection
          aria-label="Actions"
          selectionMode="single"
          onSelectionChange={onSelectionChange}
        >
          <DropdownItem key="new">New file</DropdownItem>
          <DropdownItem key="copy">Copy link</DropdownItem>
          <DropdownItem key="edit">Edit file</DropdownItem>
          <DropdownItem key="delete" color="danger">
            Delete file
          </DropdownItem>
        </DropdownMenu>
      </Dropdown>,
    );

    let triggerButton = wrapper.getByTestId("trigger-test");

    expect(onOpenChange).toBeCalledTimes(0);

    act(() => {
      triggerButton.click();
    });

    expect(onOpenChange).toBeCalledTimes(1);

    let menu = wrapper.getByRole("menu");

    expect(menu).toBeTruthy();

    // validates if the menu has the triggerButton id as aria-labelledby
    expect(menu.getAttribute("aria-labelledby")).toBe(triggerButton.id);

    let menuItems = wrapper.getAllByRole("menuitemradio");

    expect(menuItems.length).toBe(4);

    await act(async () => {
      await userEvent.click(menuItems[1]);

      expect(onSelectionChange).toBeCalledTimes(1);
      expect(onOpenChange).toBeCalled();
    });
  });

  it("should work with multiple selection (controlled)", async () => {
    let onOpenChange = jest.fn();
    let onSelectionChange = jest.fn();

    const wrapper = render(
      <Dropdown onOpenChange={onOpenChange}>
        <DropdownTrigger>
          <Button data-testid="trigger-test">Trigger</Button>
        </DropdownTrigger>
        <DropdownMenu
          disallowEmptySelection
          aria-label="Actions"
          selectionMode="multiple"
          onSelectionChange={onSelectionChange}
        >
          <DropdownItem key="new">New file</DropdownItem>
          <DropdownItem key="copy">Copy link</DropdownItem>
          <DropdownItem key="edit">Edit file</DropdownItem>
          <DropdownItem key="delete" color="danger">
            Delete file
          </DropdownItem>
        </DropdownMenu>
      </Dropdown>,
    );

    let triggerButton = wrapper.getByTestId("trigger-test");

    expect(onOpenChange).toBeCalledTimes(0);

    act(() => {
      triggerButton.click();
    });

    expect(onOpenChange).toBeCalledTimes(1);

    let menu = wrapper.getByRole("menu");

    expect(menu).toBeTruthy();

    // validates if the menu has the triggerButton id as aria-labelledby
    expect(menu.getAttribute("aria-labelledby")).toBe(triggerButton.id);

    let menuItems = wrapper.getAllByRole("menuitemcheckbox");

    expect(menuItems.length).toBe(4);

    await act(async () => {
      await userEvent.click(menuItems[0]);

      expect(onSelectionChange).toBeCalledTimes(1);
      expect(onOpenChange).toBeCalled();
    });
  });

  it("should show checkmarks if selectionMode is single and has a selected item", () => {
    const wrapper = render(
      <Dropdown isOpen>
        <DropdownTrigger>
          <Button data-testid="trigger-test">Trigger</Button>
        </DropdownTrigger>
        <DropdownMenu aria-label="Actions" selectedKeys={["new"]} selectionMode="single">
          <DropdownItem key="new">New file</DropdownItem>
          <DropdownItem key="copy">Copy link</DropdownItem>
          <DropdownItem key="edit">Edit file</DropdownItem>
          <DropdownItem key="delete" color="danger">
            Delete file
          </DropdownItem>
        </DropdownMenu>
      </Dropdown>,
    );

    let menuItems = wrapper.getAllByRole("menuitemradio");

    expect(menuItems.length).toBe(4);

    expect(menuItems[0].getAttribute("aria-checked")).toBe("true");
    expect(menuItems[1].getAttribute("aria-checked")).toBe("false");
    expect(menuItems[2].getAttribute("aria-checked")).toBe("false");
    expect(menuItems[3].getAttribute("aria-checked")).toBe("false");

    let svg = menuItems[0].querySelector("svg");

    expect(svg).toBeTruthy();

    expect(svg?.getAttribute("data-selected")).toBe("true");
  });

  it("should show multiple checkmarks if selectionMode is multiple and has selected items", () => {
    const wrapper = render(
      <Dropdown isOpen>
        <DropdownTrigger>
          <Button data-testid="trigger-test">Trigger</Button>
        </DropdownTrigger>
        <DropdownMenu aria-label="Actions" selectedKeys={["new", "copy"]} selectionMode="multiple">
          <DropdownItem key="new">New file</DropdownItem>
          <DropdownItem key="copy">Copy link</DropdownItem>
          <DropdownItem key="edit">Edit file</DropdownItem>
          <DropdownItem key="delete" color="danger">
            Delete file
          </DropdownItem>
        </DropdownMenu>
      </Dropdown>,
    );

    let menuItems = wrapper.getAllByRole("menuitemcheckbox");

    expect(menuItems.length).toBe(4);

    expect(menuItems[0].getAttribute("aria-checked")).toBe("true");
    expect(menuItems[1].getAttribute("aria-checked")).toBe("true");
    expect(menuItems[2].getAttribute("aria-checked")).toBe("false");
    expect(menuItems[3].getAttribute("aria-checked")).toBe("false");

    let checkmark1 = menuItems[0].querySelector("svg");

    expect(checkmark1).toBeTruthy();

    expect(checkmark1?.getAttribute("data-selected")).toBe("true");

    let checkmark2 = menuItems[1].querySelector("svg");

    expect(checkmark2).toBeTruthy();

    expect(checkmark2?.getAttribute("data-selected")).toBe("true");
  });

  it("should not show checkmarks if selectionMode not defined", () => {
    const wrapper = render(
      <Dropdown isOpen>
        <DropdownTrigger>
          <Button data-testid="trigger-test">Trigger</Button>
        </DropdownTrigger>
        <DropdownMenu aria-label="Actions" selectedKeys={["new", "copy"]}>
          <DropdownItem key="new">New file</DropdownItem>
          <DropdownItem key="copy">Copy link</DropdownItem>
          <DropdownItem key="edit">Edit file</DropdownItem>
          <DropdownItem key="delete" color="danger">
            Delete file
          </DropdownItem>
        </DropdownMenu>
      </Dropdown>,
    );

    let menuItems = wrapper.getAllByRole("menuitem");

    expect(menuItems.length).toBe(4);

    expect(menuItems[0].getAttribute("aria-checked")).toBeFalsy();
    expect(menuItems[1].getAttribute("aria-checked")).toBeFalsy();
    expect(menuItems[2].getAttribute("aria-checked")).toBeFalsy();
    expect(menuItems[3].getAttribute("aria-checked")).toBeFalsy();

    let checkmark1 = menuItems[0].querySelector("svg");

    expect(checkmark1).toBeFalsy();
  });

  it("should not open on disabled button", () => {
    const wrapper = render(
      <Dropdown>
        <DropdownTrigger>
          <Button isDisabled data-testid="trigger-test">
            Trigger
          </Button>
        </DropdownTrigger>
        <DropdownMenu aria-label="Actions">
          <DropdownItem key="new">New file</DropdownItem>
          <DropdownItem key="copy">Copy link</DropdownItem>
          <DropdownItem key="edit">Edit file</DropdownItem>
          <DropdownItem key="delete" color="danger">
            Delete file
          </DropdownItem>
        </DropdownMenu>
      </Dropdown>,
    );

    let triggerButton = wrapper.getByTestId("trigger-test");

    expect(triggerButton).toBeTruthy();

    act(() => {
      triggerButton.click();
    });

    let menu = wrapper.queryByRole("menu");

    expect(menu).toBeFalsy();
  });

  it("should not open on disabled dropdown", () => {
    const wrapper = render(
      <Dropdown isDisabled>
        <DropdownTrigger>
          <Button data-testid="trigger-test">Trigger</Button>
        </DropdownTrigger>
        <DropdownMenu aria-label="Actions">
          <DropdownItem key="new">New file</DropdownItem>
          <DropdownItem key="copy">Copy link</DropdownItem>
          <DropdownItem key="edit">Edit file</DropdownItem>
          <DropdownItem key="delete" color="danger">
            Delete file
          </DropdownItem>
        </DropdownMenu>
      </Dropdown>,
    );

    let triggerButton = wrapper.getByTestId("trigger-test");

    expect(triggerButton).toBeTruthy();

    act(() => {
      triggerButton.click();
    });

    let menu = wrapper.queryByRole("menu");

    expect(menu).toBeFalsy();
  });

  it("should not select on disabled item", () => {
    const onSelectionChange = jest.fn();
    const wrapper = render(
      <Dropdown isOpen>
        <DropdownTrigger>
          <Button data-testid="trigger-test">Trigger</Button>
        </DropdownTrigger>
        <DropdownMenu
          aria-label="Actions"
          disabledKeys={["copy"]}
          selectionMode="single"
          onSelectionChange={onSelectionChange}
        >
          <DropdownItem key="new">New file</DropdownItem>
          <DropdownItem key="copy">Copy link</DropdownItem>
          <DropdownItem key="edit">Edit file</DropdownItem>
          <DropdownItem key="delete" color="danger">
            Delete file
          </DropdownItem>
        </DropdownMenu>
      </Dropdown>,
    );

    let menuItems = wrapper.getAllByRole("menuitemradio");

    expect(menuItems.length).toBe(4);

    act(() => {
      menuItems[1].click();
    });

    expect(onSelectionChange).toBeCalledTimes(0);
  });

  it("should render without error (custom trigger + isDisabled)", async () => {
    const spy = jest.spyOn(console, "error").mockImplementation(() => {});

    render(
      <Dropdown isDisabled>
        <DropdownTrigger>
          <div>Trigger</div>
        </DropdownTrigger>
        <DropdownMenu aria-label="Actions" onAction={alert}>
          <DropdownItem key="new">New file</DropdownItem>
          <DropdownItem key="copy">Copy link</DropdownItem>
          <DropdownItem key="edit">Edit file</DropdownItem>
          <DropdownItem key="delete" color="danger">
            Delete file
          </DropdownItem>
        </DropdownMenu>
      </Dropdown>,
    );

    expect(spy).toBeCalledTimes(0);

    spy.mockRestore();

    render(
      <Dropdown isDisabled>
        <DropdownTrigger>
          <User as="button" description="@tonyreichert" name="Tony Reichert" />
        </DropdownTrigger>
        <DropdownMenu aria-label="Actions" onAction={alert}>
          <DropdownItem key="new">New file</DropdownItem>
          <DropdownItem key="copy">Copy link</DropdownItem>
          <DropdownItem key="edit">Edit file</DropdownItem>
          <DropdownItem key="delete" color="danger">
            Delete file
          </DropdownItem>
        </DropdownMenu>
      </Dropdown>,
    );

    expect(spy).toBeCalledTimes(0);

    spy.mockRestore();
  });
});