AgentHub/client/src/components/conversation/MessageList.tsx

/**
 * @file MessageList.tsx
 * @description Renders the chronological message stream of a Claude Code
 * transcript: alternating user / assistant rows with collapsible thinking
 * blocks, inline ToolCallBlocks for tool_use / tool_result pairs, and
 * MarkdownContent for prose. Used by ConversationView as the main body of
 * the Conversation tab on the Session detail page.
 * @author Nguyễn Ngọc Trí Vĩ <vinnt@smartgift.vn>
 */
/* =============================================================================
 * MODULE_GUIDE — extended in-file reference (comments only; safe to read, never executed)
 * =============================================================================
 * **Purpose:** Renders Claude transcript rows (user, assistant, tool calls) inside Session Detail with markdown, syntax highlighting, and TUI-style segments.
 *
 * ## Design constraints
 * - Local-first: no telemetry leaves the machine unless the user configures webhooks.
 * - Fail-safe hooks path on the server must never block Claude Code; UI mirrors that
 *   philosophy by degrading gracefully (empty states, stale badges, reconnect loops).
 * - Destructive flows stay behind explicit confirmation modals and server-side gates.
 * - Internationalization: user-visible strings belong in i18n JSON, not literals here.
 *
 * ## Remote data & SSH
 * Remote Data Sources let operators aggregate multiple machines. SSH entries describe
 * how to reach a peer dashboard; the global data scope (`dataScope.ts`) narrows every
 * scoped GET via `?sources=`. Health checks and import history surface in Settings.
 *
 * ## Internal dependencies
 * - `../../lib/types`
 * - `./ToolCallBlock`
 * - `./MarkdownContent`
 * - `../../lib/format`
 * - `./tuiSegments`
 *
 * ## Public surface
 * - `MessageList` — exported API; see TSDoc on the symbol for behavior.
 *
 * ## Testing pointers
 * - Prefer colocated `__tests__` with Vitest + Testing Library for UI.
 * - Server contract changes require `npm run test:server` and OpenAPI sync.
 *
 * ## Related docs
 * - `ARCHITECTURE.md` — hooks → API → SQLite → WebSocket → UI pipeline.
 * - `docs/API.md` — REST reference.
 * - `skills/file-headers/` — mandatory `@author` header policy.
 * ============================================================================= */
/* -----------------------------------------------------------------------------
 * EXPORT CATALOG — quick index of symbols defined below (documentation only).
 * -----------------------------------------------------------------------------
 * **MessageList**
 *   Part of this module's public contract. Downstream imports should treat
 *   the signature and return type as stable unless release notes say otherwise.
 *   When behavior changes, update the `@file` overview and relevant tests.
 *
 * ----------------------------------------------------------------------------- */

import { memo, useState, useMemo } from "react";
import { useTranslation } from "react-i18next";
import i18n from "../../i18n";
import {
  ChevronDown,
  ChevronRight,
  Bot,
  User,
  Brain,
  ScrollText,
  Terminal,
  Info,
  AlertTriangle,
  Pencil,
  Workflow,
  Cog,
} from "lucide-react";
import type { TranscriptMessage, TranscriptContent, TranscriptSender } from "../../lib/types";

/** Per-sender visual treatment for a transcript row. A JSONL `type:"user"` line
 *  is not always the human (tool results, harness task-notifications, the
 *  orchestrator's task to a subagent) — each sender gets its own label, icon,
 *  and accent so attribution is unambiguous. */
const SENDER_STYLES: Record<
  TranscriptSender,
  { icon: typeof User; avatarRing: string; accentBar: string; headerText: string }
> = {
  user: {
    icon: User,
    avatarRing: "bg-syntax-builtin/15 text-syntax-builtin ring-1 ring-syntax-builtin/30",
    accentBar: "before:bg-syntax-builtin/40",
    headerText: "text-syntax-builtin",
  },
  assistant: {
    icon: Bot,
    avatarRing: "bg-syntax-keyword/15 text-syntax-keyword ring-1 ring-syntax-keyword/30",
    accentBar: "before:bg-syntax-keyword/40",
    headerText: "text-syntax-keyword",
  },
  orchestrator: {
    icon: Workflow,
    avatarRing: "bg-syntax-string/15 text-syntax-string ring-1 ring-syntax-string/30",
    accentBar: "before:bg-syntax-string/40",
    headerText: "text-syntax-string",
  },
  system: {
    icon: Cog,
    avatarRing: "bg-surface-4 text-fg-secondary ring-1 ring-border-light/30",
    accentBar: "before:bg-surface-4/40",
    headerText: "text-fg-secondary",
  },
  tool: {
    icon: Terminal,
    avatarRing: "bg-status-warning/15 text-status-warning ring-1 ring-status-warning/30",
    accentBar: "before:bg-status-warning/40",
    headerText: "text-status-warning",
  },
};
import { ToolCallBlock } from "./ToolCallBlock";
import { MarkdownContent } from "./MarkdownContent";
import { fmt, formatModelName } from "../../lib/format";
import { parseTuiSegments, stripAnsi, hasTuiTags, type TuiSegment } from "./tuiSegments";

interface MessageListProps {
  messages: TranscriptMessage[];
  loading: boolean;
}

/** Build a map from tool_use id → tool_result for matching */
function buildToolResultMap(messages: TranscriptMessage[]): Map<string, TranscriptContent> {
  const map = new Map<string, TranscriptContent>();
  for (const msg of messages) {
    if (msg.type !== "user") continue;
    for (const c of msg.content) {
      if (c.type === "tool_result" && c.id) {
        map.set(c.id, c);
      }
    }
  }
  return map;
}

/** Detect if text is skill loading content (starts with "Base directory for this skill:") */
function isSkillContent(text: string): boolean {
  return text.startsWith("Base directory for this skill:");
}

/** Detect if text is a task notification (contains <task-notification> tag) */
function isTaskNotification(text: string): boolean {
  return text.includes("<task-notification>") || text.includes("<task-id>");
}

/** Format a timestamp as compact local time (e.g. "14:23:01"). */
function formatLocalTime(iso: string): string {
  try {
    return new Date(iso).toLocaleTimeString();
  } catch {
    return "";
  }
}

/** Centered marker for a session rename (/rename, `claude -n`, picker Ctrl+R).
 *  These TUI-only commands write no conversation turn, so without this they're
 *  invisible in the transcript. */
function SessionEventRow({ title, timestamp }: { title?: string; timestamp: string | null }) {
  return (
    <div className="flex items-center justify-center py-1">
      <div className="inline-flex items-center gap-2 text-[11px] text-fg-secondary bg-surface-2/70 border border-surface-3 rounded-full px-3 py-1 max-w-full">
        <Pencil className="w-3 h-3 text-violet-700/70 dark:text-violet-300/70 flex-shrink-0" />
        <span className="text-fg-muted">Renamed session →</span>
        <span className="text-fg-secondary font-medium truncate">{title || "(untitled)"}</span>
        {timestamp && (
          <span className="text-[10px] text-fg-muted font-mono flex-shrink-0">
            {formatLocalTime(timestamp)}
          </span>
        )}
      </div>
    </div>
  );
}

/** Compact pill for /command invocations parsed out of TUI markup. */
function CommandPill({ display }: { display: string }) {
  return (
    <div className="inline-flex items-center gap-2 text-sm text-status-success font-mono bg-status-success/10 border border-status-success/20 rounded-md px-3 py-1.5 max-w-full">
      <span className="text-status-success/70">›</span>
      <span className="break-all">{display}</span>
    </div>
  );
}

/** Terminal-style fenced block for stdout/stderr captured from local commands. */
function TerminalBlock({ text, stream }: { text: string; stream: "stdout" | "stderr" }) {
  const cleaned = stripAnsi(text).replace(/^\n+|\n+$/g, "");
  const isErr = stream === "stderr";
  const accent = isErr
    ? "border-status-danger/30 bg-status-danger/30 text-status-danger/90"
    : "border-surface-3 bg-surface-4/60 text-fg-secondary";
  const labelColor = isErr ? "text-status-danger/80" : "text-fg-secondary";
  return (
    <div className={`rounded-lg border ${accent} overflow-hidden`}>
      <div
        className={`flex items-center gap-1.5 px-3 py-1 text-[10px] uppercase tracking-wider border-b border-current/10 ${labelColor}`}
      >
        <Terminal className="w-3 h-3" />
        <span>{stream}</span>
      </div>
      <pre className="px-3 py-2 text-xs font-mono whitespace-pre-wrap break-words leading-relaxed max-h-96 overflow-y-auto">
        {cleaned}
      </pre>
    </div>
  );
}

/** Subtle inline note for the local-command-caveat banner. */
function CaveatBlock({ text }: { text: string }) {
  return (
    <div className="flex items-start gap-2 rounded-md border border-status-warning/15 bg-status-warning/[0.05] px-3 py-1.5 text-[11px] text-status-warning/70">
      <Info className="w-3.5 h-3.5 mt-px flex-shrink-0 opacity-60" />
      <span className="leading-relaxed italic">{stripAnsi(text).trim()}</span>
    </div>
  );
}

/** Render a single segment produced by parseTuiSegments. */
function renderSegment(seg: TuiSegment, key: number): React.ReactNode {
  switch (seg.kind) {
    case "command":
      return <CommandPill key={key} display={seg.display} />;
    case "stdout":
      return <TerminalBlock key={key} text={seg.text} stream="stdout" />;
    case "stderr":
      return <TerminalBlock key={key} text={seg.text} stream="stderr" />;
    case "caveat":
      return <CaveatBlock key={key} text={seg.text} />;
    case "system-reminder":
      return (
        <CollapsibleBlock
          key={key}
          text={seg.text}
          icon={<AlertTriangle className="w-3.5 h-3.5 text-status-warning/70 flex-shrink-0" />}
          title={i18n.t("sessions:conversation.systemReminder")}
          borderClass="border-status-warning/20"
          bgClass="bg-status-warning/5"
          textClass="text-status-warning/80"
        />
      );
    case "persisted-output":
      return (
        <CollapsibleBlock
          key={key}
          text={seg.text}
          icon={<ScrollText className="w-3.5 h-3.5 text-syntax-keyword/70 flex-shrink-0" />}
          title={i18n.t("sessions:conversation.persistedOutput")}
          borderClass="border-syntax-keyword/20"
          bgClass="bg-syntax-keyword/5"
          textClass="text-syntax-keyword"
        />
      );
    case "text": {
      const cleaned = stripAnsi(seg.text);
      if (!cleaned.trim()) return null;
      return (
        <div key={key} className="min-w-0">
          <MarkdownContent text={cleaned} />
        </div>
      );
    }
  }
}

/** Generic collapsible content block */
function CollapsibleBlock({
  text,
  icon,
  title,
  borderClass,
  bgClass,
  textClass,
}: {
  text: string;
  icon: React.ReactNode;
  title: string;
  borderClass: string;
  bgClass: string;
  textClass: string;
}) {
  const [expanded, setExpanded] = useState(false);

  return (
    <div className={`rounded-lg border ${borderClass} ${bgClass} overflow-hidden`}>
      <button
        onClick={() => setExpanded(!expanded)}
        aria-expanded={expanded}
        className="w-full flex items-center gap-2 px-3 py-1.5 text-left hover:opacity-80 transition-colors"
      >
        {expanded ? (
          <ChevronDown className="w-3.5 h-3.5 opacity-60 flex-shrink-0" />
        ) : (
          <ChevronRight className="w-3.5 h-3.5 opacity-60 flex-shrink-0" />
        )}
        {icon}
        <span className={`text-xs ${textClass} truncate`}>{title}</span>
      </button>
      {expanded && (
        <div className="border-t border-current/10 px-3 py-2">
          <pre className="text-xs opacity-60 whitespace-pre-wrap break-words leading-relaxed max-h-96 overflow-y-auto">
            {text}
          </pre>
        </div>
      )}
    </div>
  );
}

/** A key per message that survives prepend (older history) and append (live
 *  tail): built from the message itself, with an occurrence counter for twins.
 *  The index was the key before, so expanding a block and then loading older
 *  messages moved the expansion onto another message. */
export function messageKeys(messages: TranscriptMessage[]): string[] {
  const seen = new Map<string, number>();
  return messages.map((m) => {
    const first = m.content[0];
    const sig = first
      ? `${first.type}:${first.id ?? (first.text ?? "").slice(0, 48)}`
      : (m.title ?? "");
    const base = `${m.timestamp ?? ""}|${m.type}|${m.sender ?? ""}|${sig}`;
    const n = seen.get(base) ?? 0;
    seen.set(base, n + 1);
    return n ? `${base}#${n}` : base;
  });
}

export function MessageList({ messages, loading }: MessageListProps) {
  const { t } = useTranslation("sessions");
  const toolResultMap = useMemo(() => buildToolResultMap(messages), [messages]);
  const keys = useMemo(() => messageKeys(messages), [messages]);

  if (loading) {
    return (
      <div className="flex items-center justify-center py-12 text-fg-muted text-sm">
        {t("conversation.loading")}
      </div>
    );
  }

  if (messages.length === 0) {
    return <div className="text-center py-12 text-fg-muted text-sm">{t("conversation.empty")}</div>;
  }

  return (
    <div className="space-y-3">
      {messages.map((msg, idx) => {
        // Session lifecycle markers (e.g. /rename) render as a centered chip,
        // not as a user/assistant row.
        if (msg.type === "session_event") {
          return <SessionEventRow key={keys[idx]} title={msg.title} timestamp={msg.timestamp} />;
        }
        // Skip user messages that are purely tool_result - they're rendered inside ToolCallBlock
        if (msg.type === "user" && !msg.content.some((c) => c.type === "text")) {
          return null;
        }
        return <MessageRow key={keys[idx]} msg={msg} toolResultMap={toolResultMap} />;
      })}
    </div>
  );
}

/** One user/assistant row. Memoized: a live tail append re-renders only the
 *  new rows; its thinking-block expansion is its own state, kept by key. */
const MessageRow = memo(function MessageRow({
  msg,
  toolResultMap,
}: {
  msg: TranscriptMessage;
  toolResultMap: ReturnType<typeof buildToolResultMap>;
}) {
  const { t } = useTranslation("sessions");
  const [expandedThinking, setExpandedThinking] = useState<Set<number>>(() => new Set());
  const isAssistant = msg.type === "assistant";
  // The true sender (classified server-side) drives the label + styling.
  // Falls back to the coarse type for older payloads without `sender`.
  const sender: TranscriptSender = msg.sender ?? (isAssistant ? "assistant" : "user");
  const style = SENDER_STYLES[sender] ?? SENDER_STYLES.user;
  const SenderIcon = style.icon;

  return (
    <div
      style={{ contentVisibility: "auto", containIntrinsicSize: "auto 120px" }}
      className={`relative flex gap-3 rounded-xl px-3 py-2.5 hover:bg-surface-2/30 transition-colors before:absolute before:left-0 before:top-3 before:bottom-3 before:w-0.5 before:rounded-full before:opacity-60 ${style.accentBar}`}
    >
      {/* Avatar */}
      <div
        className={`flex-shrink-0 w-8 h-8 rounded-full flex items-center justify-center mt-0.5 shadow-sm ${style.avatarRing}`}
      >
        <SenderIcon className="w-4 h-4" />
      </div>

      {/* Message body */}
      <div className="flex-1 min-w-0 space-y-2">
        {/* Header line */}
        <div className="flex items-center gap-2 flex-wrap">
          <span className={`text-xs font-semibold tracking-wide ${style.headerText}`}>
            {t(`conversation.sender.${sender}`)}
          </span>
          {msg.model && (
            <span className="text-[10px] text-fg-secondary font-mono bg-surface-3/60 border border-surface-3 rounded px-1.5 py-0.5">
              {formatModelName(msg.model)}
            </span>
          )}
          {msg.usage && (
            <span className="text-[10px] text-fg-muted font-mono inline-flex items-center gap-1">
              <span className="text-status-success/70">↓ {fmt(msg.usage.input_tokens)}</span>
              <span className="text-fg-muted">·</span>
              <span className="text-syntax-number">↑ {fmt(msg.usage.output_tokens)}</span>
            </span>
          )}
          {msg.timestamp && (
            <span className="text-[10px] text-fg-muted ml-auto font-mono">
              {formatLocalTime(msg.timestamp)}
            </span>
          )}
        </div>

        {/* Content blocks */}
        {msg.content.map((block, bIdx) => {
          if (block.type === "text" && block.text) {
            // Detect task notifications, collapsed by default
            if (isTaskNotification(block.text)) {
              return (
                <CollapsibleBlock
                  key={bIdx}
                  text={block.text}
                  icon={
                    <ScrollText className="w-3.5 h-3.5 text-syntax-property/70 flex-shrink-0" />
                  }
                  title={t("conversation.taskNotification")}
                  borderClass="border-syntax-property/20"
                  bgClass="bg-syntax-property/5"
                  textClass="text-syntax-property"
                />
              );
            }

            // Detect skill content, collapsed by default
            if (isSkillContent(block.text)) {
              const pathMatch = block.text.match(/^Base directory for this skill:\s*(\S+)/);
              const skillPath = pathMatch ? pathMatch[1]! : "Skill";
              return (
                <CollapsibleBlock
                  key={bIdx}
                  text={block.text}
                  icon={<ScrollText className="w-3.5 h-3.5 text-syntax-builtin/70 flex-shrink-0" />}
                  title={skillPath}
                  borderClass="border-syntax-builtin/20"
                  bgClass="bg-syntax-builtin/5"
                  textClass="text-syntax-builtin"
                />
              );
            }

            // Mixed TUI markup: caveat / command / stdout / stderr / system-reminder
            // can appear inline (sometimes interleaved with prose). Parse the text
            // into segments and render each with the appropriate visual treatment.
            if (hasTuiTags(block.text)) {
              const segments = parseTuiSegments(block.text);
              return (
                <div key={bIdx} className="space-y-2 min-w-0">
                  {segments.map((s, sIdx) => renderSegment(s, sIdx))}
                </div>
              );
            }

            return (
              <div key={bIdx} className="min-w-0">
                <MarkdownContent text={stripAnsi(block.text)} />
              </div>
            );
          }

          if (block.type === "thinking" && block.text) {
            const thinkKey = bIdx;
            const isExpanded = expandedThinking.has(thinkKey);
            return (
              <div
                key={bIdx}
                className="rounded-lg border border-status-warning/20 bg-status-warning/5 overflow-hidden"
              >
                <button
                  aria-expanded={isExpanded}
                  onClick={() =>
                    setExpandedThinking((prev) => {
                      const next = new Set(prev);
                      if (next.has(thinkKey)) next.delete(thinkKey);
                      else next.add(thinkKey);
                      return next;
                    })
                  }
                  className="w-full flex items-center gap-2 px-3 py-1.5 text-left hover:bg-status-warning/10 transition-colors"
                >
                  <ChevronRight
                    className={`w-3.5 h-3.5 text-status-warning/60 transition-transform duration-150 ${
                      isExpanded ? "rotate-90" : ""
                    }`}
                  />
                  <Brain className="w-3.5 h-3.5 text-status-warning/80" />
                  <span className="text-xs text-status-warning/90 font-medium">
                    {t("conversation.thinking")}
                  </span>
                  {!isExpanded && (
                    <span className="text-[10px] text-status-warning/40 font-mono ml-auto">
                      {t("conversation.chars", {
                        count: block.text.length,
                        formatted: block.text.length.toLocaleString(),
                      })}
                    </span>
                  )}
                </button>
                {isExpanded && (
                  <div className="border-t border-status-warning/10 px-3 py-2 text-status-warning/80">
                    <MarkdownContent text={block.text} dense />
                  </div>
                )}
              </div>
            );
          }

          if (block.type === "tool_use") {
            const matchedResult = block.id ? (toolResultMap.get(block.id) ?? null) : null;
            return <ToolCallBlock key={bIdx} toolUse={block} toolResult={matchedResult} />;
          }

          // tool_result blocks rendered inside ToolCallBlock, skip standalone
          return null;
        })}
      </div>
    </div>
  );
});