import { expect, test, describe } from "bun:test";
import { readFileSync } from "fs";
import { join } from "path";
import {
  formatTurnsFromData,
  groupTurnsNaturally,
  formatGroupedContent,
  detectContentType,
  formatResultContent,
  formatToolWithResult,
  type Turn,
  type ToolUse,
  type ToolResult,
} from "../src/entrypoints/format-turns";

describe("detectContentType", () => {
  test("detects JSON objects", () => {
    expect(detectContentType('{"key": "value"}')).toBe("json");
    expect(detectContentType('{"number": 42}')).toBe("json");
  });

  test("detects JSON arrays", () => {
    expect(detectContentType("[1, 2, 3]")).toBe("json");
    expect(detectContentType('["a", "b"]')).toBe("json");
  });

  test("detects Python code", () => {
    expect(detectContentType("def hello():\n    pass")).toBe("python");
    expect(detectContentType("import os")).toBe("python");
    expect(detectContentType("from math import pi")).toBe("python");
  });

  test("detects JavaScript code", () => {
    expect(detectContentType("function test() {}")).toBe("javascript");
    expect(detectContentType("const x = 5")).toBe("javascript");
    expect(detectContentType("let y = 10")).toBe("javascript");
    expect(detectContentType("const fn = () => console.log()")).toBe(
      "javascript",
    );
  });

  test("detects bash/shell content", () => {
    expect(detectContentType("/usr/bin/test")).toBe("bash");
    expect(detectContentType("Error: command not found")).toBe("bash");
    expect(detectContentType("ls -la")).toBe("bash");
    expect(detectContentType("$ echo hello")).toBe("bash");
  });

  test("detects diff format", () => {
    expect(detectContentType("@@ -1,3 +1,3 @@")).toBe("diff");
    expect(detectContentType("+++ file.txt")).toBe("diff");
    expect(detectContentType("--- file.txt")).toBe("diff");
  });

  test("detects HTML/XML", () => {
    expect(detectContentType("<div>hello</div>")).toBe("html");
    expect(detectContentType("<xml>content</xml>")).toBe("html");
  });

  test("detects markdown", () => {
    expect(detectContentType("- List item")).toBe("markdown");
    expect(detectContentType("* List item")).toBe("markdown");
    expect(detectContentType("```code```")).toBe("markdown");
  });

  test("defaults to text", () => {
    expect(detectContentType("plain text")).toBe("text");
    expect(detectContentType("just some words")).toBe("text");
  });
});

describe("formatResultContent", () => {
  test("handles empty content", () => {
    expect(formatResultContent("")).toBe("*(No output)*\n\n");
    expect(formatResultContent(null)).toBe("*(No output)*\n\n");
    expect(formatResultContent(undefined)).toBe("*(No output)*\n\n");
  });

  test("formats short text without code blocks", () => {
    const result = formatResultContent("success");
    expect(result).toBe("**→** success\n\n");
  });

  test("formats long text with code blocks", () => {
    const longText =
      "This is a longer piece of text that should be formatted in a code block because it exceeds the short text threshold";
    const result = formatResultContent(longText);
    expect(result).toContain("**Result:**");
    expect(result).toContain("```text");
    expect(result).toContain(longText);
  });

  test("pretty prints JSON content", () => {
    const jsonContent = '{"key": "value", "number": 42}';
    const result = formatResultContent(jsonContent);
    expect(result).toContain("```json");
    expect(result).toContain('"key": "value"');
    expect(result).toContain('"number": 42');
  });

  test("truncates very long content", () => {
    const veryLongContent = "A".repeat(4000);
    const result = formatResultContent(veryLongContent);
    expect(result).toContain("...");
    // Should not contain the full long content
    expect(result.length).toBeLessThan(veryLongContent.length);
  });

  test("handles type:text structure", () => {
    const structuredContent = [{ type: "text", text: "Hello world" }];
    const result = formatResultContent(JSON.stringify(structuredContent));
    expect(result).toBe("**→** Hello world\n\n");
  });

  test("strips ANSI color codes from terminal output", () => {
    // Test bold yellow warning (the issue reported: [1;33m)
    const coloredOutput = "\x1B[1;33mWarning: something happened\x1B[0m";
    const result = formatResultContent(coloredOutput);
    expect(result).toBe("**→** Warning: something happened\n\n");
    expect(result).not.toContain("\x1B");
    expect(result).not.toContain("[1;33m");
  });

  test("strips ANSI codes from longer output in code blocks", () => {
    const longColoredOutput =
      "\x1B[32m✓\x1B[0m Test 1 passed\n" +
      "\x1B[32m✓\x1B[0m Test 2 passed\n" +
      "\x1B[31m✗\x1B[0m Test 3 failed\n" +
      "Some additional output to make it longer";
    const result = formatResultContent(longColoredOutput);
    expect(result).toContain("✓ Test 1 passed");
    expect(result).toContain("✗ Test 3 failed");
    expect(result).not.toContain("\x1B");
  });
});

describe("formatToolWithResult", () => {
  test("formats tool with parameters and result", () => {
    const toolUse: ToolUse = {
      type: "tool_use",
      name: "read_file",
      input: { file_path: "/path/to/file.txt" },
      id: "tool_123",
    };

    const toolResult: ToolResult = {
      type: "tool_result",
      tool_use_id: "tool_123",
      content: "File content here",
      is_error: false,
    };

    const result = formatToolWithResult(toolUse, toolResult);

    expect(result).toContain("### 🔧 `read_file`");
    expect(result).toContain("**Parameters:**");
    expect(result).toContain('"file_path": "/path/to/file.txt"');
    expect(result).toContain("**→** File content here");
  });

  test("formats tool with error result", () => {
    const toolUse: ToolUse = {
      type: "tool_use",
      name: "failing_tool",
      input: { param: "value" },
    };

    const toolResult: ToolResult = {
      type: "tool_result",
      content: "Permission denied",
      is_error: true,
    };

    const result = formatToolWithResult(toolUse, toolResult);

    expect(result).toContain("### 🔧 `failing_tool`");
    expect(result).toContain("❌ **Error:** `Permission denied`");
  });

  test("formats tool without parameters", () => {
    const toolUse: ToolUse = {
      type: "tool_use",
      name: "simple_tool",
    };

    const result = formatToolWithResult(toolUse);

    expect(result).toContain("### 🔧 `simple_tool`");
    expect(result).not.toContain("**Parameters:**");
  });

  test("handles unknown tool name", () => {
    const toolUse: ToolUse = {
      type: "tool_use",
    };

    const result = formatToolWithResult(toolUse);

    expect(result).toContain("### 🔧 `unknown_tool`");
  });
});

describe("groupTurnsNaturally", () => {
  test("groups system initialization", () => {
    const data: Turn[] = [
      {
        type: "system",
        subtype: "init",
        tools: [{ name: "tool1" }, { name: "tool2" }],
      },
    ];

    const result = groupTurnsNaturally(data);

    expect(result).toHaveLength(1);
    expect(result[0]?.type).toBe("system_init");
    expect(result[0]?.tools_count).toBe(2);
  });

  test("groups assistant actions with tool calls", () => {
    const data: Turn[] = [
      {
        type: "assistant",
        message: {
          content: [
            { type: "text", text: "I'll help you" },
            {
              type: "tool_use",
              id: "tool_123",
              name: "read_file",
              input: { file_path: "/test.txt" },
            },
          ],
          usage: { input_tokens: 100, output_tokens: 50 },
        },
      },
      {
        type: "user",
        message: {
          content: [
            {
              type: "tool_result",
              tool_use_id: "tool_123",
              content: "file content",
              is_error: false,
            },
          ],
        },
      },
    ];

    const result = groupTurnsNaturally(data);

    expect(result).toHaveLength(1);
    expect(result[0]?.type).toBe("assistant_action");
    expect(result[0]?.text_parts).toEqual(["I'll help you"]);
    expect(result[0]?.tool_calls).toHaveLength(1);
    expect(result[0]?.tool_calls?.[0]?.tool_use.name).toBe("read_file");
    expect(result[0]?.tool_calls?.[0]?.tool_result?.content).toBe(
      "file content",
    );
    expect(result[0]?.usage).toEqual({ input_tokens: 100, output_tokens: 50 });
  });

  test("groups user messages", () => {
    const data: Turn[] = [
      {
        type: "user",
        message: {
          content: [{ type: "text", text: "Please help me" }],
        },
      },
    ];

    const result = groupTurnsNaturally(data);

    expect(result).toHaveLength(1);
    expect(result[0]?.type).toBe("user_message");
    expect(result[0]?.text_parts).toEqual(["Please help me"]);
  });

  test("groups final results", () => {
    const data: Turn[] = [
      {
        type: "result",
        cost_usd: 0.1234,
        duration_ms: 5000,
        result: "Task completed",
      },
    ];

    const result = groupTurnsNaturally(data);

    expect(result).toHaveLength(1);
    expect(result[0]?.type).toBe("final_result");
    expect(result[0]?.data).toEqual(data[0]!);
  });
});

describe("formatGroupedContent", () => {
  test("formats system initialization", () => {
    const groupedContent = [
      {
        type: "system_init",
        tools_count: 3,
      },
    ];

    const result = formatGroupedContent(groupedContent);

    expect(result).toContain("## Claude Code Report");
    expect(result).toContain("## 🚀 System Initialization");
    expect(result).toContain("**Available Tools:** 3 tools loaded");
  });

  test("formats assistant actions", () => {
    const groupedContent = [
      {
        type: "assistant_action",
        text_parts: ["I'll help you with that"],
        tool_calls: [
          {
            tool_use: {
              type: "tool_use",
              name: "test_tool",
              input: { param: "value" },
            },
            tool_result: {
              type: "tool_result",
              content: "result",
              is_error: false,
            },
          },
        ],
        usage: { input_tokens: 100, output_tokens: 50 },
      },
    ];

    const result = formatGroupedContent(groupedContent);

    expect(result).toContain("I'll help you with that");
    expect(result).toContain("### 🔧 `test_tool`");
    expect(result).toContain("*Token usage: 100 input, 50 output*");
  });

  test("formats user messages", () => {
    const groupedContent = [
      {
        type: "user_message",
        text_parts: ["Help me please"],
      },
    ];

    const result = formatGroupedContent(groupedContent);

    expect(result).toContain("## 👤 User");
    expect(result).toContain("Help me please");
  });

  test("formats final results", () => {
    const groupedContent = [
      {
        type: "final_result",
        data: {
          type: "result",
          cost_usd: 0.1234,
          duration_ms: 5678,
          result: "Success!",
        } as Turn,
      },
    ];

    const result = formatGroupedContent(groupedContent);

    expect(result).toContain("## ✅ Final Result");
    expect(result).toContain("Success!");
    expect(result).toContain("**Cost:** $0.1234");
    expect(result).toContain("**Duration:** 5.7s");
  });
});

describe("formatTurnsFromData", () => {
  test("handles empty data", () => {
    const result = formatTurnsFromData([]);
    expect(result).toBe("## Claude Code Report\n\n");
  });

  test("formats complete conversation", () => {
    const data: Turn[] = [
      {
        type: "system",
        subtype: "init",
        tools: [{ name: "tool1" }],
      },
      {
        type: "assistant",
        message: {
          content: [
            { type: "text", text: "I'll help you" },
            {
              type: "tool_use",
              id: "tool_123",
              name: "read_file",
              input: { file_path: "/test.txt" },
            },
          ],
        },
      },
      {
        type: "user",
        message: {
          content: [
            {
              type: "tool_result",
              tool_use_id: "tool_123",
              content: "file content",
              is_error: false,
            },
          ],
        },
      },
      {
        type: "result",
        cost_usd: 0.05,
        duration_ms: 2000,
        result: "Done",
      },
    ];

    const result = formatTurnsFromData(data);

    expect(result).toContain("## Claude Code Report");
    expect(result).toContain("## 🚀 System Initialization");
    expect(result).toContain("I'll help you");
    expect(result).toContain("### 🔧 `read_file`");
    expect(result).toContain("## ✅ Final Result");
    expect(result).toContain("Done");
  });
});

describe("integration tests", () => {
  test("formats real conversation data correctly", () => {
    // Load the sample JSON data
    const jsonPath = join(__dirname, "fixtures", "sample-turns.json");
    const expectedPath = join(
      __dirname,
      "fixtures",
      "sample-turns-expected-output.md",
    );

    const jsonData = JSON.parse(readFileSync(jsonPath, "utf-8"));
    const expectedOutput = readFileSync(expectedPath, "utf-8").trim();

    // Format the data using our function
    const actualOutput = formatTurnsFromData(jsonData).trim();

    // Compare the outputs
    expect(actualOutput).toBe(expectedOutput);
  });
});