torrentclaw-mcp/src/formatters/content.ts

import type {
  SearchResponse,
  SearchResult,
  TorrentInfo,
  PopularResponse,
  PopularItem,
  RecentResponse,
  RecentItem,
} from "../types.js";

function formatSize(bytes: string | null): string {
  if (!bytes) return "?";
  const b = parseInt(bytes, 10);
  if (isNaN(b)) return "?";
  if (b >= 1_073_741_824) return `${(b / 1_073_741_824).toFixed(1)} GB`;
  if (b >= 1_048_576) return `${(b / 1_048_576).toFixed(0)} MB`;
  return `${(b / 1024).toFixed(0)} KB`;
}

function formatRating(imdb: string | null, tmdb: string | null): string {
  const parts: string[] = [];
  if (imdb) parts.push(`IMDb: ${imdb}`);
  if (tmdb) parts.push(`TMDB: ${tmdb}`);
  return parts.length > 0 ? parts.join(" | ") : "No ratings";
}

function truncate(text: string, max: number): string {
  if (text.length <= max) return text;
  return text.slice(0, max - 3) + "...";
}

function formatTorrent(t: TorrentInfo, compact?: boolean): string {
  const parts: string[] = [];
  if (t.quality) parts.push(t.quality);
  if (t.sourceType) parts.push(t.sourceType);
  if (t.codec) parts.push(t.codec);
  if (t.hdrType) parts.push(t.hdrType);
  const label = parts.length > 0 ? parts.join(" ") : "Unknown quality";
  const size = formatSize(t.sizeBytes);
  const seeds = `${t.seeders} seeders`;
  const score = t.qualityScore !== null ? `Score: ${t.qualityScore}` : null;

  let line = `  - ${label} (${size}) | ${seeds}`;
  if (score) line += ` | ${score}`;

  if (t.season != null) {
    const ep =
      t.episode != null ? `E${String(t.episode).padStart(2, "0")}` : "";
    line += ` | S${String(t.season).padStart(2, "0")}${ep}`;
  }

  line += `\n    Info hash: ${t.infoHash}`;
  if (compact) {
    line += `\n    Magnet: magnet:?xt=urn:btih:${t.infoHash}`;
  } else if (t.magnetUrl) {
    line += `\n    Magnet: ${t.magnetUrl}`;
  }
  if (t.torrentUrl) {
    line += `\n    Torrent: ${t.torrentUrl}`;
  }

  // Audio/subtitle track summary (from scanned torrents)
  if (t.audioTracks && t.audioTracks.length > 0) {
    const langs = t.audioTracks
      .map((a) => a.lang || "?")
      .filter((v, i, arr) => arr.indexOf(v) === i);
    line += `\n    Audio: ${langs.join(", ")}`;
    const codecs = t.audioTracks
      .map((a) => a.codec)
      .filter((v): v is string => v != null)
      .filter((v, i, arr) => arr.indexOf(v) === i);
    if (codecs.length > 0) line += ` (${codecs.join(", ")})`;
  }
  if (t.subtitleTracks && t.subtitleTracks.length > 0) {
    const langs = t.subtitleTracks
      .map((s) => s.lang || "?")
      .filter((v, i, arr) => arr.indexOf(v) === i);
    line += `\n    Subtitles: ${langs.join(", ")}`;
  }

  return line;
}

export interface FormatOptions {
  compact?: boolean;
}

function formatResult(
  r: SearchResult,
  index: number,
  opts?: FormatOptions,
): string {
  const lines: string[] = [];
  const yearStr = r.year ? ` (${r.year})` : "";
  lines.push(`${index}. ${r.title}${yearStr} [${r.contentType}]`);
  lines.push(`   ${formatRating(r.ratingImdb, r.ratingTmdb)}`);
  if (r.genres && r.genres.length > 0) {
    lines.push(`   Genres: ${r.genres.join(", ")}`);
  }
  if (r.overview) {
    lines.push(`   ${truncate(r.overview, 200)}`);
  }

  if (r.torrents.length > 0) {
    const top = r.torrents
      .sort((a, b) => (b.qualityScore ?? 0) - (a.qualityScore ?? 0))
      .slice(0, 5);
    lines.push(`   Torrents (${r.torrents.length} total, top ${top.length}):`);
    for (const t of top) {
      lines.push(formatTorrent(t, opts?.compact));
    }
  } else {
    lines.push("   No torrents available");
  }

  if (r.streaming) {
    const providers: string[] = [];
    if (r.streaming.flatrate.length > 0)
      providers.push(
        `Stream: ${r.streaming.flatrate.map((p) => p.name).join(", ")}`,
      );
    if (r.streaming.free.length > 0)
      providers.push(`Free: ${r.streaming.free.map((p) => p.name).join(", ")}`);
    if (providers.length > 0) lines.push(`   ${providers.join(" | ")}`);
  }

  lines.push(
    `   Content ID: ${r.id} — use with get_watch_providers(content_id=${r.id}) or get_credits(content_id=${r.id})`,
  );
  if (r.imdbId) lines.push(`   IMDb: ${r.imdbId}`);
  if (r.contentUrl) lines.push(`   URL: ${r.contentUrl}`);

  return lines.join("\n");
}

export function formatSearchResults(
  data: SearchResponse,
  opts?: FormatOptions,
): string {
  if (data.results.length === 0) {
    return "No results found. Try: (1) a shorter or alternate title, (2) removing filters like quality or year, (3) checking spelling. You can also try get_popular or get_recent to browse available content.";
  }

  const headerParts = [
    `Found ${data.total} results (page ${data.page}, showing ${data.results.length}):`,
  ];
  if (data.parsedSeason != null) {
    const ep =
      data.parsedEpisode != null
        ? `E${String(data.parsedEpisode).padStart(2, "0")}`
        : "";
    headerParts.push(
      `Detected season/episode: S${String(data.parsedSeason).padStart(2, "0")}${ep}`,
    );
  }

  const results = data.results.map((r, i) => formatResult(r, i + 1, opts));
  return [...headerParts, "", ...results].join("\n");
}

function formatPopularItem(item: PopularItem, index: number): string {
  const yearStr = item.year ? ` (${item.year})` : "";
  const rating = formatRating(item.ratingImdb, item.ratingTmdb);
  return `${index}. ${item.title}${yearStr} [${item.contentType}] — ${rating} — ${item.clickCount} clicks — ID: ${item.id}`;
}

export function formatPopularResults(data: PopularResponse): string {
  if (data.items.length === 0) {
    return "No popular content found.";
  }

  const header = `Popular content (${data.total} total, page ${data.page}):`;
  const hint =
    "(Use search_content with a title to get torrents and full details)";
  const items = data.items.map((item, i) => formatPopularItem(item, i + 1));
  return [header, hint, "", ...items].join("\n");
}

function formatRecentItem(item: RecentItem, index: number): string {
  const yearStr = item.year ? ` (${item.year})` : "";
  const rating = formatRating(item.ratingImdb, item.ratingTmdb);
  const date = new Date(item.createdAt).toLocaleDateString("en-US", {
    month: "short",
    day: "numeric",
    year: "numeric",
  });
  return `${index}. ${item.title}${yearStr} [${item.contentType}] — ${rating} — Added: ${date} — ID: ${item.id}`;
}

export function formatRecentResults(data: RecentResponse): string {
  if (data.items.length === 0) {
    return "No recent content found.";
  }

  const header = `Recently added content (${data.total} total, page ${data.page}):`;
  const hint =
    "(Use search_content with a title to get torrents and full details)";
  const items = data.items.map((item, i) => formatRecentItem(item, i + 1));
  return [header, hint, "", ...items].join("\n");
}