import { Text, Markdown, type Component } from "@earendil-works/pi-tui";
import { keyHint, getMarkdownTheme } from "@earendil-works/pi-coding-agent";

// ---------------------------------------------------------------------------
// Types — match the shape pi passes to renderResult
// ---------------------------------------------------------------------------

export interface ToolResultLike {
  content: Array<{ type: string; text?: string }>;
  details?: Record<string, unknown>;
}

export interface RenderOptions {
  expanded: boolean;
  isPartial: boolean;
}

// ---------------------------------------------------------------------------
// Shared collapsible markdown renderer
// ---------------------------------------------------------------------------

/**
 * Default `renderResult` body for the six TinyFish tools.
 *
 * - isPartial → one-line "Running..." placeholder (avoids re-rendering big
 *   markdown on every SSE event for agent-run).
 * - !expanded → one-line summary + "ctrl+o to expand" hint. Summary is the
 *   pre-formatted string the caller passes (it should already include the
 *   relevant count / status from `details`).
 * - expanded → full markdown from `content`. Markdown component handles
 *   wrapping and theme colors.
 *
 * Tools that need fully custom rendering (e.g. progress bars) should write
 * their own renderResult and skip this helper.
 */
export function renderCollapsibleMarkdown(
  result: ToolResultLike,
  options: RenderOptions,
  theme: { fg: (color: string, text: string) => string },
  summary: string,
): Component {
  if (options.isPartial) {
    return new Text(theme.fg("warning", "⏳ Running..."), 0, 0);
  }

  if (!options.expanded) {
    const hint = keyHint("app.tools.expand", "to expand");
    return new Text(`${theme.fg("success", summary)}  ${theme.fg("dim", `(${hint})`)}`, 0, 0);
  }

  const text = result.content
    .filter((c): c is { type: "text"; text: string } => c.type === "text" && typeof c.text === "string")
    .map((c) => c.text)
    .join("\n");

  return new Markdown(text, 0, 0, getMarkdownTheme());
}