import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
import { afterAll, afterEach, beforeAll, describe, expect, it } from "vitest";
import { KnowledgeBase } from "../../src/knowledge/index.js";
import { RecipeLibrary } from "../../src/recipes/loader.js";
import { TouchDesignerClient } from "../../src/td-client/touchDesignerClient.js";
import { duplicateNetworkImpl } from "../../src/tools/layer2/duplicateNetwork.js";
import { getTdPerformanceImpl } from "../../src/tools/layer3/getTdPerformance.js";
import { getTdTopologyImpl } from "../../src/tools/layer3/getTdTopology.js";
import type { ToolContext } from "../../src/tools/types.js";
import { silentLogger } from "../../src/utils/logger.js";
import { makeTdServer, TD_BASE } from "../helpers/tdMock.js";

const server = makeTdServer();
beforeAll(() => server.listen({ onUnhandledRequest: "error" }));
afterEach(() => server.resetHandlers());
afterAll(() => server.close());

function makeCtx(): ToolContext {
  return {
    client: new TouchDesignerClient({ baseUrl: TD_BASE, timeoutMs: 2000 }),
    knowledge: new KnowledgeBase(),
    recipes: new RecipeLibrary(),
    logger: silentLogger,
  };
}

function textOf(result: CallToolResult): string {
  return result.content
    .filter((c): c is { type: "text"; text: string } => c.type === "text")
    .map((c) => c.text)
    .join("\n");
}

describe("analysis tool handlers", () => {
  describe("get_td_performance", () => {
    it("reports cook times on structuredContent without error", async () => {
      const result = await getTdPerformanceImpl(makeCtx(), {
        root_path: "/project1",
        target_fps: 60,
        recursive: true,
      });
      expect(result.isError).toBeFalsy();
      // Summary text stays small; full data travels on structuredContent.
      expect(textOf(result)).toContain("/project1");
      const data = JSON.stringify(result.structuredContent);
      expect(data).toContain("/project1/noise1");
      expect(data).toContain("totalCookMs");
      expect(data).toContain("frameBudgetMs");
    });
  });

  describe("get_td_topology", () => {
    it("reports nodes/connections on structuredContent without error", async () => {
      const result = await getTdTopologyImpl(makeCtx(), { root_path: "/project1" });
      expect(result.isError).toBeFalsy();
      expect(textOf(result)).toContain("node(s)");
      const data = JSON.stringify(result.structuredContent);
      expect(data).toContain("nodeCount");
      expect(data).toContain("connectionCount");
      expect(data).toContain("/project1/noise1");
    });
  });

  describe("duplicate_network", () => {
    it("duplicates a node even when exec returns a null result", async () => {
      const result = await duplicateNetworkImpl(makeCtx(), {
        source_path: "/project1/noise1",
      });
      expect(result.isError).toBeFalsy();
      // Mock /api/exec returns { result: null }, so the copy path falls back to stdout.
      expect(textOf(result)).toContain("Duplicated /project1/noise1");
    });
  });
});