/**
 * Unit tests for the ImportAgentDialog component.
 *
 * Tests:
 * - Initial render shows file picker and paste mode toggle
 * - Mode toggle switches between file picker and paste textarea
 * - Uploading valid JSON shows the agent preview
 * - Uploading invalid JSON shows an error alert
 * - Uploading a payload with unsupported version shows an error
 * - Uploading a non-agent type shows an error
 * - Import button is NOT shown until valid JSON is parsed
 * - "Back" button returns from preview to the picker
 * - Import button triggers the mutation and shows success state
 * - Warnings are displayed after import with warnings
 * - "Cancel" button calls onOpenChange(false)
 * - "Done" button calls onOpenChange(false) after success
 * - Paste mode: Parse JSON button parses pasted content
 * - Paste mode: Parse JSON button is disabled for empty content
 */
import { fireEvent, render, screen, waitFor } from "@testing-library/react";
import userEvent from "@testing-library/user-event";
import { beforeEach, describe, expect, it, vi } from "vitest";
import { ImportAgentDialog } from "./import-agent-dialog";

// ---------------------------------------------------------------------------
// Mocks
// ---------------------------------------------------------------------------

const mutate = vi.fn().mockImplementation((payload, options) => {
  options?.onSuccess?.({
    agent: {
      id: "agent-123",
      name: `${payload.agent.name} (imported)`,
    },
    warnings: [],
  });
});

vi.mock("@/lib/agent.query", () => ({
  useImportAgent: () => ({
    mutate,
    isPending: false,
  }),
}));

vi.mock("@/lib/hooks/use-app-name", () => ({
  useAppName: () => "Archestra",
}));

vi.mock("@/components/editor", () => ({
  Editor: (props: {
    value?: string;
    onChange?: (value: string | undefined) => void;
    options?: { ariaLabel?: string; placeholder?: string };
  }) => (
    <textarea
      aria-label={props.options?.ariaLabel}
      placeholder={props.options?.placeholder}
      value={props.value ?? ""}
      onChange={(event) => props.onChange?.(event.target.value)}
    />
  ),
}));

// ---------------------------------------------------------------------------
// Fixtures
// ---------------------------------------------------------------------------

const validPayload = {
  version: "1",
  exportedAt: new Date().toISOString(),
  sourceInstance: null,
  agent: {
    name: "Test Import Agent",
    agentType: "agent",
    description: "A test agent for import dialog",
    systemPrompt: "Be helpful",
    icon: "🤖",
    scope: "personal",
    considerContextUntrusted: false,
    toolAssignmentMode: "manual",
    toolExposureMode: "full",
    incomingEmailEnabled: false,
    incomingEmailSecurityMode: "private",
    incomingEmailAllowedDomain: null,
    passthroughHeaders: null,
  },
  labels: [{ key: "env", value: "test" }],
  suggestedPrompts: [],
  tools: [
    {
      toolName: "web_search",
      catalogName: "Web Catalog",
      credentialResolutionMode: "dynamic",
    },
    {
      toolName: "code_exec",
      catalogName: "Dev Tools",
      credentialResolutionMode: "static",
    },
  ],
  delegations: [{ targetAgentName: "Sub Agent" }],
  knowledgeBases: [{ name: "Company Wiki" }],
  connectors: [{ name: "Confluence", connectorType: "confluence" }],
};

const validPayloadJson = JSON.stringify(validPayload);

const invalidJson = "{ this is not valid JSON }";

const unknownVersionPayload = JSON.stringify({
  ...validPayload,
  version: "99",
});

const gatewayPayload = JSON.stringify({
  ...validPayload,
  agent: { ...validPayload.agent, agentType: "mcp_gateway" },
});

/** Simulate a file upload on a hidden <input type="file"> element */
async function simulateFileUpload(
  user: ReturnType<typeof userEvent.setup>,
  content: string,
  filename = "agent-export.json",
) {
  const file = new File([content], filename, { type: "application/json" });
  const input = document.querySelector(
    'input[type="file"]',
  ) as HTMLInputElement;
  await user.upload(input, file);
}

// ---------------------------------------------------------------------------
// Tests
// ---------------------------------------------------------------------------

describe("ImportAgentDialog", () => {
  beforeEach(() => {
    mutate.mockClear();
    mutate.mockImplementation((payload, options) => {
      options?.onSuccess?.({
        agent: {
          id: "agent-123",
          name: `${payload.agent.name} (imported)`,
        },
        warnings: [],
      });
    });
  });

  it("renders the dialog with title, file picker, and mode toggle when open", () => {
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    expect(screen.getByText("Import Agent")).toBeInTheDocument();
    expect(
      screen.getByRole("button", { name: /upload file/i }),
    ).toBeInTheDocument();
    expect(
      screen.getByRole("button", { name: /paste json/i }),
    ).toBeInTheDocument();
    expect(screen.getByText(/drag and drop/i)).toBeInTheDocument();
    expect(screen.getByRole("button", { name: /cancel/i })).toBeInTheDocument();
  });

  it("does not render when open is false", () => {
    render(<ImportAgentDialog open={false} onOpenChange={vi.fn()} />);

    expect(screen.queryByText("Import Agent")).not.toBeInTheDocument();
  });

  it("switches to paste mode when 'Paste JSON' button is clicked", async () => {
    const user = userEvent.setup();
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    await user.click(screen.getByRole("button", { name: /paste json/i }));

    expect(screen.getByLabelText(/paste agent json here/i)).toBeInTheDocument();
    expect(
      screen.getByRole("button", { name: /parse json/i }),
    ).toBeInTheDocument();
    // File picker should be gone
    expect(screen.queryByText(/drag and drop/i)).not.toBeInTheDocument();
  });

  it("Parse JSON button is disabled when paste area is empty", async () => {
    const user = userEvent.setup();
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    await user.click(screen.getByRole("button", { name: /paste json/i }));

    const parseBtn = screen.getByRole("button", { name: /parse json/i });
    expect(parseBtn).toBeDisabled();
  });

  it("shows agent preview after uploading valid JSON file", async () => {
    const user = userEvent.setup();
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    await simulateFileUpload(user, validPayloadJson);

    await waitFor(() => {
      expect(screen.getByText("Ready to Import")).toBeInTheDocument();
    });

    // Agent name shown in preview
    expect(screen.getByText("Test Import Agent")).toBeInTheDocument();

    // Association counts
    expect(screen.getByText(/Tools \(2\)/i)).toBeInTheDocument();
    expect(screen.getByText(/Delegations \(1\)/i)).toBeInTheDocument();
    expect(screen.getByText(/Knowledge \(1\)/i)).toBeInTheDocument();
    expect(screen.getByText(/Connectors \(1\)/i)).toBeInTheDocument();

    // Import button is now shown
    expect(
      screen.getByRole("button", { name: /import agent/i }),
    ).toBeInTheDocument();
  });

  it("shows error alert for invalid JSON", async () => {
    const user = userEvent.setup();
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    await simulateFileUpload(user, invalidJson);

    await waitFor(() => {
      expect(screen.getByText("Invalid Configuration")).toBeInTheDocument();
      expect(screen.getByText(/invalid json file/i)).toBeInTheDocument();
    });

    // Import button must NOT be shown
    expect(
      screen.queryByRole("button", { name: /import agent/i }),
    ).not.toBeInTheDocument();
  });

  it("shows error for unsupported version number", async () => {
    const user = userEvent.setup();
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    await simulateFileUpload(user, unknownVersionPayload);

    await waitFor(() => {
      expect(screen.getByText(/unsupported version/i)).toBeInTheDocument();
    });
  });

  it("shows error for non-agent agentType (mcp_gateway)", async () => {
    const user = userEvent.setup();
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    await simulateFileUpload(user, gatewayPayload);

    await waitFor(() => {
      expect(
        screen.getByText(/only internal agents can be imported/i),
      ).toBeInTheDocument();
    });
  });

  it("shows missing required fields error when agentType is absent", async () => {
    const user = userEvent.setup();
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    await user.click(screen.getByRole("button", { name: /paste json/i }));

    const textarea = screen.getByLabelText(/paste agent json here/i);
    await user.click(textarea);
    await user.paste(
      JSON.stringify({
        version: "1",
        agent: { name: "Partial Agent" },
      }),
    );

    await user.click(screen.getByRole("button", { name: /parse json/i }));

    await waitFor(() => {
      expect(screen.getByText("Invalid Configuration")).toBeInTheDocument();
      expect(
        screen.getByText(/missing required fields .*agent\.agentType/i),
      ).toBeInTheDocument();
    });
  });

  it("shows missing required fields error when preview arrays are absent", async () => {
    const user = userEvent.setup();
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    await user.click(screen.getByRole("button", { name: /paste json/i }));

    const textarea = screen.getByLabelText(/paste agent json here/i);
    await user.click(textarea);
    await user.paste(
      JSON.stringify({
        version: "1",
        agent: { name: "Partial Agent", agentType: "agent" },
      }),
    );

    await user.click(screen.getByRole("button", { name: /parse json/i }));

    await waitFor(() => {
      expect(screen.getByText("Invalid Configuration")).toBeInTheDocument();
      expect(screen.getByText(/missing required fields/i)).toHaveTextContent(
        "tools",
      );
    });
  });

  it("'Back' button returns from preview to idle state", async () => {
    const user = userEvent.setup();
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    await simulateFileUpload(user, validPayloadJson);
    await waitFor(() => screen.getByText("Ready to Import"));

    await user.click(screen.getByRole("button", { name: /back/i }));

    // Should be back to the file picker
    expect(screen.getByText(/drag and drop/i)).toBeInTheDocument();
    expect(screen.queryByText("Ready to Import")).not.toBeInTheDocument();
  });

  it("calls mutate and closes dialog after import", async () => {
    const onOpenChange = vi.fn();
    const user = userEvent.setup();
    render(<ImportAgentDialog open onOpenChange={onOpenChange} />);

    await simulateFileUpload(user, validPayloadJson);
    await waitFor(() => screen.getByRole("button", { name: /import agent/i }));

    await user.click(screen.getByRole("button", { name: /import agent/i }));

    expect(mutate).toHaveBeenCalledOnce();
    // Dialog stays open to show success state
    expect(onOpenChange).not.toHaveBeenCalledWith(false);

    await waitFor(() => {
      expect(screen.getByText(/import complete/i)).toBeInTheDocument();
    });

    // Done closes the dialog
    await user.click(screen.getByRole("button", { name: /done/i }));
    expect(onOpenChange).toHaveBeenCalledWith(false);
  });

  it("renders backend warnings after import", async () => {
    const user = userEvent.setup();
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    mutate.mockImplementation((_payload, options) => {
      options?.onSuccess?.({
        agent: { id: "agent-123", name: "Warn Agent (imported)" },
        warnings: [
          {
            type: "tool",
            name: "missing_tool",
            message: 'Tool "missing_tool" could not be resolved.',
          },
        ],
      });
    });

    await simulateFileUpload(user, validPayloadJson);
    await waitFor(() => screen.getByRole("button", { name: /import agent/i }));

    await user.click(screen.getByRole("button", { name: /import agent/i }));

    await waitFor(() => {
      expect(screen.getByText(/warn agent \(imported\)/i)).toBeInTheDocument();
      expect(screen.getByText(/missing_tool/i)).toBeInTheDocument();
    });
  });

  it("shows backend error message when mutation fails", async () => {
    const user = userEvent.setup();
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    mutate.mockImplementation((_payload, options) => {
      options?.onError?.({
        error: { message: "Invalid import payload: bad field" },
      });
    });

    await simulateFileUpload(user, validPayloadJson);
    await waitFor(() => screen.getByRole("button", { name: /import agent/i }));

    await user.click(screen.getByRole("button", { name: /import agent/i }));

    await waitFor(() => {
      expect(screen.getByText("Invalid Configuration")).toBeInTheDocument();
      expect(
        screen.getByText(/invalid import payload: bad field/i),
      ).toBeInTheDocument();
    });
  });

  it("calls onSuccess callback with agent data and warning count", async () => {
    const onSuccess = vi.fn();
    const user = userEvent.setup();
    render(
      <ImportAgentDialog open onOpenChange={vi.fn()} onSuccess={onSuccess} />,
    );

    await simulateFileUpload(user, validPayloadJson);
    await waitFor(() => screen.getByRole("button", { name: /import agent/i }));

    await user.click(screen.getByRole("button", { name: /import agent/i }));

    await waitFor(() => {
      expect(onSuccess).toHaveBeenCalledWith(
        { id: "agent-123", name: "Test Import Agent (imported)" },
        0,
      );
    });
  });

  it("'Cancel' button calls onOpenChange(false)", async () => {
    const onOpenChange = vi.fn();
    const user = userEvent.setup();
    render(<ImportAgentDialog open onOpenChange={onOpenChange} />);

    await user.click(screen.getByRole("button", { name: /cancel/i }));

    expect(onOpenChange).toHaveBeenCalledWith(false);
  });

  it("parses pasted JSON and shows preview in paste mode", async () => {
    const user = userEvent.setup();
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    // Switch to paste mode
    await user.click(screen.getByRole("button", { name: /paste json/i }));

    // Type valid JSON into the textarea
    const textarea = screen.getByLabelText(/paste agent json here/i);
    await user.click(textarea);
    await user.paste(
      JSON.stringify({
        ...validPayload,
        agent: { ...validPayload.agent, name: "Pasted Agent" },
      }),
    );

    // Parse JSON button should be enabled now
    const parseBtn = screen.getByRole("button", { name: /parse json/i });
    expect(parseBtn).not.toBeDisabled();

    await user.click(parseBtn);

    await waitFor(() => {
      expect(screen.getByText("Ready to Import")).toBeInTheDocument();
      expect(screen.getByText("Pasted Agent")).toBeInTheDocument();
    });
  });

  it("shows error when a non-JSON file is dropped", async () => {
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    const dropzone = screen
      .getByText(/drag and drop/i)
      .closest("label") as HTMLLabelElement;
    const file = new File(["not a json file"], "image.png", {
      type: "image/png",
    });

    fireEvent.drop(dropzone, {
      dataTransfer: {
        files: [file],
      },
    });

    await waitFor(() => {
      expect(
        screen.getByText("Only .json files are accepted."),
      ).toBeInTheDocument();
    });
  });

  it("shows agent preview when a valid JSON file is dropped", async () => {
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    const dropzone = screen
      .getByText(/drag and drop/i)
      .closest("label") as HTMLLabelElement;
    const file = new File([validPayloadJson], "agent-export.json", {
      type: "application/json",
    });

    fireEvent.drop(dropzone, {
      dataTransfer: {
        files: [file],
      },
    });

    await waitFor(() => {
      expect(screen.getByText("Ready to Import")).toBeInTheDocument();
      expect(screen.getByText("Test Import Agent")).toBeInTheDocument();
    });
  });

  it("handles drag events properly to toggle styles", async () => {
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    const dropzone = screen
      .getByText(/drag and drop/i)
      .closest("label") as HTMLLabelElement;

    // Initially not active
    expect(dropzone).not.toHaveClass("border-primary");

    // Enter
    fireEvent.dragEnter(dropzone);
    expect(dropzone).toHaveClass("border-primary");

    // Leave
    fireEvent.dragLeave(dropzone);
    expect(dropzone).not.toHaveClass("border-primary");
  });

  it("handles file read error gracefully", async () => {
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    // Mock FileReader to fail
    const readAsTextMock = vi
      .spyOn(FileReader.prototype, "readAsText")
      .mockImplementation(function (this: FileReader) {
        if (this.onerror) {
          this.onerror(
            new ProgressEvent("error") as unknown as ProgressEvent<FileReader>,
          );
        }
      });

    const file = new File(["some content"], "agent.json", {
      type: "application/json",
    });
    const dropzone = screen
      .getByText(/drag and drop/i)
      .closest("label") as HTMLLabelElement;
    fireEvent.drop(dropzone, { dataTransfer: { files: [file] } });

    await waitFor(() => {
      expect(
        screen.getByText("Failed to read the file. Please try again."),
      ).toBeInTheDocument();
    });

    readAsTextMock.mockRestore();
  });

  it("shows error alert for invalid JSON in paste mode", async () => {
    const user = userEvent.setup();
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    await user.click(screen.getByRole("button", { name: /paste json/i }));

    const textarea = screen.getByLabelText(/paste agent json here/i);
    await user.click(textarea);
    await user.paste("{ invalid }");

    await user.click(screen.getByRole("button", { name: /parse json/i }));

    await waitFor(() => {
      expect(screen.getByText("Invalid Configuration")).toBeInTheDocument();
      expect(screen.getByText(/invalid json file/i)).toBeInTheDocument();
    });
  });

  it("handles file change with no files selected gracefully", async () => {
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);
    const input = document.querySelector(
      'input[type="file"]',
    ) as HTMLInputElement;

    fireEvent.change(input, { target: { files: null } });

    expect(screen.getByText(/drag and drop/i)).toBeInTheDocument();
  });

  it("handles drop event with no files gracefully", async () => {
    render(<ImportAgentDialog open onOpenChange={vi.fn()} />);

    const dropzone = screen
      .getByText(/drag and drop/i)
      .closest("label") as HTMLLabelElement;

    fireEvent.drop(dropzone, {
      dataTransfer: {
        files: [],
      },
    });

    expect(screen.getByText(/drag and drop/i)).toBeInTheDocument();
  });
});
