import type { TranscriptEntry } from "@paperclipai/adapter-utils";

function safeJsonParse(text: string): unknown {
  try {
    return JSON.parse(text);
  } catch {
    return null;
  }
}

function asRecord(value: unknown): Record<string, unknown> | null {
  if (typeof value !== "object" || value === null || Array.isArray(value)) return null;
  return value as Record<string, unknown>;
}

function asString(value: unknown, fallback = ""): string {
  return typeof value === "string" ? value : fallback;
}

function extractErrorText(value: unknown): string {
  if (typeof value === "string") return value;
  const record = asRecord(value);
  if (!record) return "";
  return asString(record.message) || asString(record.detail) || asString(record.code);
}

export function parseGrokStdoutLine(line: string, ts: string): TranscriptEntry[] {
  const parsed = asRecord(safeJsonParse(line));
  if (!parsed) {
    return [{ kind: "stdout", ts, text: line }];
  }

  const type = asString(parsed.type).trim();

  if (type === "thought") {
    const text = asString(parsed.data);
    return text ? [{ kind: "thinking", ts, text, delta: true }] : [];
  }

  if (type === "text") {
    const text = asString(parsed.data);
    return text ? [{ kind: "assistant", ts, text, delta: true }] : [];
  }

  if (type === "error") {
    const text = asString(parsed.data) || asString(parsed.message) || extractErrorText(parsed.error);
    return text ? [{ kind: "stderr", ts, text }] : [{ kind: "stderr", ts, text: "Grok error" }];
  }

  if (type === "end") {
    const stopReason = asString(parsed.stopReason).trim();
    const sessionId = asString(parsed.sessionId).trim();
    const parts = [
      stopReason ? `stop_reason=${stopReason}` : "",
      sessionId ? `session=${sessionId}` : "",
    ].filter(Boolean);
    return [{ kind: "system", ts, text: parts.join(" ") || "run completed" }];
  }

  return [{ kind: "system", ts, text: `event: ${type || "unknown"}` }];
}