import { describe, it, expect, vi } from "vitest";
import { createMockServer } from "../helpers.js";
import { registerAutocomplete } from "../../src/tools/autocomplete.js";
import { TorrentClawClient, ApiError } from "../../src/api-client.js";

function createMockClient(overrides: Partial<TorrentClawClient> = {}) {
  return {
    search: vi.fn(),
    autocomplete: vi.fn(),
    getPopular: vi.fn(),
    getRecent: vi.fn(),
    getWatchProviders: vi.fn(),
    getCredits: vi.fn(),
    getStats: vi.fn(),
    getTorrentDownloadUrl: vi.fn(),
    track: vi.fn(),
    submitScanRequest: vi.fn(),
    getScanStatus: vi.fn(),
    ...overrides,
  } as unknown as TorrentClawClient;
}

describe("autocomplete tool", () => {
  it("returns formatted suggestions", async () => {
    const client = createMockClient({
      autocomplete: vi.fn().mockResolvedValue({
        suggestions: [
          {
            id: 1,
            title: "Breaking Bad",
            year: 2008,
            contentType: "show",
            posterUrl: null,
          },
          {
            id: 2,
            title: "The Break-Up",
            year: 2006,
            contentType: "movie",
            posterUrl: null,
          },
        ],
      }),
    });
    const { server, getToolHandler } = createMockServer();
    registerAutocomplete(server, client);

    const handler = getToolHandler("autocomplete");
    const result = await handler({ query: "break" });

    expect(result.isError).toBeUndefined();
    expect(result.content[0].text).toContain("Breaking Bad");
    expect(result.content[0].text).toContain("The Break-Up");
    expect(result.content[0].text).toContain("[show]");
    expect(result.content[0].text).toContain("[movie]");
  });

  it("returns message when no suggestions found", async () => {
    const client = createMockClient({
      autocomplete: vi.fn().mockResolvedValue({ suggestions: [] }),
    });
    const { server, getToolHandler } = createMockServer();
    registerAutocomplete(server, client);

    const handler = getToolHandler("autocomplete");
    const result = await handler({ query: "zzzzz" });

    expect(result.content[0].text).toContain("No suggestions");
    expect(result.content[0].text).toContain("search_content");
  });

  it("returns isError on ApiError", async () => {
    const client = createMockClient({
      autocomplete: vi
        .fn()
        .mockRejectedValue(new ApiError(429, "Rate limited")),
    });
    const { server, getToolHandler } = createMockServer();
    registerAutocomplete(server, client);

    const handler = getToolHandler("autocomplete");
    const result = await handler({ query: "test" });

    expect(result.isError).toBe(true);
    expect(result.content[0].text).toContain("TorrentClaw API error (429)");
  });

  it("returns isError on generic error", async () => {
    const client = createMockClient({
      autocomplete: vi.fn().mockRejectedValue(new Error("Network timeout")),
    });
    const { server, getToolHandler } = createMockServer();
    registerAutocomplete(server, client);

    const handler = getToolHandler("autocomplete");
    const result = await handler({ query: "test" });

    expect(result.isError).toBe(true);
    expect(result.content[0].text).toContain("Request failed: Network timeout");
  });

  it("handles non-Error throw", async () => {
    const client = createMockClient({
      autocomplete: vi.fn().mockRejectedValue("string error"),
    });
    const { server, getToolHandler } = createMockServer();
    registerAutocomplete(server, client);

    const handler = getToolHandler("autocomplete");
    const result = await handler({ query: "test" });

    expect(result.isError).toBe(true);
    expect(result.content[0].text).toContain("Unknown error");
  });
});