import assert from "node:assert/strict";
import test from "node:test";

import {
  extractPromptText,
  extractToolIdentity,
  parsePromptText,
  parseSystemPromptSections,
} from "./agentSessionTranscriptHelpers.ts";

const HEX = "a".repeat(64);
const HEX_UPPER = "A".repeat(64);

test("parsePromptText returns the empty/Prompt fallback for whitespace-only input", () => {
  // The early `sections.length === 0` branch only fires when there are no
  // section bodies at all (e.g. empty/whitespace input).
  const result = parsePromptText("   ");
  assert.deepEqual(result, {
    sections: [],
    userText: "",
    userTitle: "Prompt",
    userPubkey: null,
    userEventId: null,
  });
});

test("parsePromptText wraps header-less free text in a single Prompt section", () => {
  // Free text with no `[header]` becomes one "Prompt" section. Since no
  // section is a "Buzz event", there is no event content to surface, so
  // userText is empty and the title falls through to "Buzz event".
  const result = parsePromptText("just some free text");
  assert.deepEqual(
    result.sections.map((s) => s.title),
    ["Prompt"],
  );
  assert.equal(result.sections[0].body, "just some free text");
  assert.equal(result.userText, "");
  assert.equal(result.userTitle, "Buzz event");
  assert.equal(result.userPubkey, null);
  assert.equal(result.userEventId, null);
});

test("parsePromptText extracts event id, content, hex pubkey, and a title-cased kind", () => {
  const text = [
    "[System]",
    "system preamble here",
    "",
    "[Buzz event: @mention]",
    `Event ID: ${HEX_UPPER}`,
    "Channel: demo",
    `From: Wes (hex: ${HEX})`,
    "Content: hello @Brain please look",
  ].join("\n");

  const result = parsePromptText(text);

  assert.equal(result.userText, "hello @Brain please look");
  assert.equal(result.userPubkey, HEX);
  assert.equal(result.userEventId, HEX);
  // titleCase capitalizes after word boundaries but leaves the leading "@"
  // (a non-word char) in place: "@mention" -> "@Mention".
  assert.equal(result.userTitle, "@Mention");
  // Both headers become sections.
  assert.deepEqual(
    result.sections.map((s) => s.title),
    ["System", "Buzz event: @mention"],
  );
});

test("parsePromptText preserves multiline event content in the user bubble text", () => {
  const text = [
    "[Buzz event: @mention]",
    "Event ID: event-1",
    "Channel: agents",
    `From: tho (hex: ${HEX})`,
    "Time: 2026-06-15T17:15:00Z",
    "Content: @Ned",
    "",
    "- remove that stray cherry pick if it's not adding value here",
    "- help me understand what that e2eBridge change does",
    "- we'd want the e2e seed path as a separate pull request",
    'Tags: [["h","agents"]]',
    "Parsed: mentions=[Ned]",
  ].join("\n");

  const result = parsePromptText(text);

  assert.equal(
    result.userText,
    [
      "@Ned",
      "",
      "- remove that stray cherry pick if it's not adding value here",
      "- help me understand what that e2eBridge change does",
      "- we'd want the e2e seed path as a separate pull request",
    ].join("\n"),
  );
  assert.equal(result.userPubkey, HEX);
  assert.equal(result.userEventId, null);
});

test("parsePromptText lowercases the extracted hex pubkey", () => {
  const text = [
    "[Buzz event: dm]",
    `From: Someone (hex: ${HEX_UPPER})`,
    "Content: hi",
  ].join("\n");

  const result = parsePromptText(text);
  assert.equal(result.userPubkey, HEX);
});

test("parsePromptText yields a null pubkey when From has no hex", () => {
  const text = ["[Buzz event: note]", "From: Someone", "Content: hi"].join(
    "\n",
  );

  const result = parsePromptText(text);
  assert.equal(result.userPubkey, null);
  assert.equal(result.userText, "hi");
  assert.equal(result.userTitle, "Note");
});

test("parsePromptText defaults the title to 'Buzz event' when no kind is present", () => {
  const text = ["[Buzz event]", "Content: x"].join("\n");
  const result = parsePromptText(text);
  assert.equal(result.userTitle, "Buzz event");
});

test("parsePromptText leading text before a header becomes a Prompt section", () => {
  const text = ["preamble line", "[Other]", "body"].join("\n");
  const result = parsePromptText(text);
  assert.deepEqual(
    result.sections.map((s) => s.title),
    ["Prompt", "Other"],
  );
});

test("extractPromptText joins text blocks from params.prompt", () => {
  const payload = {
    params: {
      prompt: [{ text: "line one" }, { text: "line two" }],
    },
  };
  assert.equal(extractPromptText(payload), "line one\nline two");
});

test("extractPromptText handles plain string blocks", () => {
  const payload = { params: { prompt: ["a", "b"] } };
  assert.equal(extractPromptText(payload), "a\nb");
});

test("extractPromptText returns empty string when prompt is missing or not an array", () => {
  assert.equal(extractPromptText({}), "");
  assert.equal(extractPromptText({ params: { prompt: "nope" } }), "");
});

test("extractToolIdentity ignores Buzz tool names that only appear in file contents", () => {
  const identity = extractToolIdentity({
    sessionUpdate: "tool_call_update",
    toolCallId: "read-file-1",
    status: "completed",
    title: "read_file",
    kind: "read_file",
    rawInput: {
      path: "desktop/src/features/agents/ui/agentSessionToolCatalog.ts",
    },
    content: {
      text: 'const BUZZ_READ_TOOLS = new Set(["get_feed", "get_event"]);',
    },
  });

  assert.deepEqual(identity, {
    title: "read_file",
    toolName: "read_file",
    buzzToolName: null,
  });
});

test("extractToolIdentity still recognizes explicit Buzz tool fields", () => {
  const identity = extractToolIdentity({
    sessionUpdate: "tool_call",
    title: "Tool call",
    toolName: "get_feed",
    rawInput: { limit: 50 },
  });

  assert.deepEqual(identity, {
    title: "Tool call",
    toolName: "get_feed",
    buzzToolName: "get_feed",
  });
});

test("parseSystemPromptSections splits both prompts into Base and System", () => {
  const framed = "[Base]\nbase text\n\n[System]\npersona text";
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Base", body: "base text" },
    { title: "System", body: "persona text" },
  ]);
});

test("parseSystemPromptSections yields one Base section for a base-only frame", () => {
  const sections = parseSystemPromptSections("[Base]\nbase text");
  assert.deepEqual(sections, [{ title: "Base", body: "base text" }]);
});

test("parseSystemPromptSections yields one System section for a persona-only frame", () => {
  const sections = parseSystemPromptSections("[System]\npersona text");
  assert.deepEqual(sections, [{ title: "System", body: "persona text" }]);
});

test("parseSystemPromptSections keeps embedded bracket lines literal in bodies", () => {
  // A persona that itself contains a [Context]-like line must NOT split into a
  // spurious sub-section — the body is read literally after the first boundary.
  const framed = "[Base]\nbase\n\n[System]\nrule one\n[Context]\nrule two";
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Base", body: "base" },
    { title: "System", body: "rule one\n[Context]\nrule two" },
  ]);
});

test("parseSystemPromptSections degrades to a labeled Base when [System] header is elided", () => {
  // Oversize trim can drop the [System] header mid-string. Without a boundary
  // the whole value stays under a correctly-labeled Base — no missing label,
  // no inflated count, just a truncated body.
  const elided = "[Base]\nbase text …[elided 900000 bytes]… persona tail";
  const sections = parseSystemPromptSections(elided);
  assert.deepEqual(sections, [
    { title: "Base", body: "base text …[elided 900000 bytes]… persona tail" },
  ]);
});

test("parseSystemPromptSections returns no sections for empty input", () => {
  assert.deepEqual(parseSystemPromptSections(""), []);
  assert.deepEqual(parseSystemPromptSections("   "), []);
});

// ── [Agent Memory — core] section tests ──────────────────────────────────────

test("parseSystemPromptSections extracts core as its own section after Base+System", () => {
  const framed =
    "[Base]\nbase text\n\n[System]\npersona text\n\n[Agent Memory — core]\nmy memories";
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Base", body: "base text" },
    { title: "System", body: "persona text" },
    { title: "Core Memory", body: "my memories" },
  ]);
});

test("parseSystemPromptSections extracts core as its own section after Base only", () => {
  const framed = "[Base]\nbase text\n\n[Agent Memory — core]\nmy memories";
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Base", body: "base text" },
    { title: "Core Memory", body: "my memories" },
  ]);
});

test("parseSystemPromptSections extracts core as its own section after System only", () => {
  const framed = "[System]\npersona text\n\n[Agent Memory — core]\nmy memories";
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "System", body: "persona text" },
    { title: "Core Memory", body: "my memories" },
  ]);
});

test("parseSystemPromptSections returns only a core section when no Base/System present", () => {
  const framed = "[Agent Memory — core]\nmy memories";
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [{ title: "Core Memory", body: "my memories" }]);
});

test("parseSystemPromptSections keeps an embedded core-like line literal when a real appended core follows", () => {
  // The persona body contains a line that looks like the header. The actual
  // appended core block comes last — only the LAST boundary should split.
  const framed = [
    "[Base]",
    "base text",
    "",
    "[System]",
    "persona preamble",
    "[Agent Memory — core]",
    "this is NOT the core section — it is inside the persona body",
    "",
    "[Agent Memory — core]",
    "this IS the appended core",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Base", body: "base text" },
    {
      title: "System",
      body: "persona preamble\n[Agent Memory — core]\nthis is NOT the core section — it is inside the persona body",
    },
    {
      title: "Core Memory",
      body: "this IS the appended core",
    },
  ]);
});

test("parseSystemPromptSections keeps exact core header literal when only a single newline precedes it (no-core persona)", () => {
  // A no-core [System] persona that contains the exact header text on its own
  // line (preceded by only a single \n, not the double-newline appended
  // separator) must NOT be extracted as a Core Memory section.
  const framed = [
    "[System]",
    "persona preamble",
    "[Agent Memory — core]",
    "this is persona text, not a real core block",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    {
      title: "System",
      body: "persona preamble\n[Agent Memory — core]\nthis is persona text, not a real core block",
    },
  ]);
});

test("parseSystemPromptSections pins the realistic Workspace+Base+System+Core harness shape", () => {
  // The real Buzz harness emits [Workspace] content before [Base]. The parser
  // folds [Workspace] into the Base section (existing unchanged behavior);
  // core is extracted as a distinct "Core Memory" section last.
  const framed = [
    "[Workspace]",
    "You are operating inside the Buzz platform.",
    "",
    "[Base]",
    "You are an assistant.",
    "",
    "[System]",
    "Custom persona instructions.",
    "",
    "[Agent Memory — core]",
    "I am Duncan.",
    "## Lessons Learned",
    "Always tag on handoff.",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    {
      title: "Base",
      body: "[Workspace]\nYou are operating inside the Buzz platform.\n\n[Base]\nYou are an assistant.",
    },
    { title: "System", body: "Custom persona instructions." },
    {
      title: "Core Memory",
      body: "I am Duncan.\n## Lessons Learned\nAlways tag on handoff.",
    },
  ]);
});

// ── Channel Canvas extraction ─────────────────────────────────────────────────

test("parseSystemPromptSections pins the full Base+System+Core+Canvas harness shape", () => {
  // with_canvas() appends "\n\n[Channel Canvas]\n{metadata}" after the core block.
  // render_canvas_section() emits the revision event ID, last-modified timestamp,
  // and fetch command — never the canvas body. All four sections are extracted in order.
  const framed = [
    "[Base]",
    "You are an assistant.",
    "",
    "[System]",
    "Persona instructions.",
    "",
    "[Agent Memory — core]",
    "I am Duncan.",
    "",
    "[Channel Canvas]",
    "Canvas revision (event ID): a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2",
    "Last modified: 2026-07-11T10:00:00Z",
    "Fetch current content with: buzz canvas get --channel 94a444a4-c0a3-5966-ab05-530c6ddc2301",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Base", body: "You are an assistant." },
    { title: "System", body: "Persona instructions." },
    { title: "Core Memory", body: "I am Duncan." },
    {
      title: "Channel Canvas",
      body: "Canvas revision (event ID): a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2\nLast modified: 2026-07-11T10:00:00Z\nFetch current content with: buzz canvas get --channel 94a444a4-c0a3-5966-ab05-530c6ddc2301",
    },
  ]);
});

test("parseSystemPromptSections extracts canvas when no Base/System/Core present (canvas-only)", () => {
  const framed = ["[Channel Canvas]", "Canvas only."].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Channel Canvas", body: "Canvas only." },
  ]);
});

test("parseSystemPromptSections extracts Core+Canvas with no Base/System", () => {
  const framed = [
    "[Agent Memory — core]",
    "Core content.",
    "",
    "[Channel Canvas]",
    "Canvas content.",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Core Memory", body: "Core content." },
    { title: "Channel Canvas", body: "Canvas content." },
  ]);
});

test("parseSystemPromptSections extracts Base+Canvas when no System or Core", () => {
  const framed = [
    "[Base]",
    "Base content.",
    "",
    "[Channel Canvas]",
    "Canvas content.",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Base", body: "Base content." },
    { title: "Channel Canvas", body: "Canvas content." },
  ]);
});

test("parseSystemPromptSections omits canvas section when no canvas present (no regression)", () => {
  // Existing Base+System+Core shape must still produce exactly three sections.
  const framed = [
    "[Base]",
    "Base.",
    "",
    "[System]",
    "System.",
    "",
    "[Agent Memory — core]",
    "Core.",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Base", body: "Base." },
    { title: "System", body: "System." },
    { title: "Core Memory", body: "Core." },
  ]);
});

test("parseSystemPromptSections keeps embedded canvas-like line literal when only a single newline precedes it (no-canvas persona)", () => {
  // A [Channel Canvas] header inside a persona body preceded by a single \n
  // (not the double-newline appended separator) must NOT be extracted as a
  // Canvas section — same last-occurrence guard as the Core extraction.
  const framed = [
    "[System]",
    "persona preamble",
    "[Channel Canvas]",
    "this is persona text, not a real canvas block",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    {
      title: "System",
      body: "persona preamble\n[Channel Canvas]\nthis is persona text, not a real canvas block",
    },
  ]);
});

test("parseSystemPromptSections keeps an embedded canvas-like line literal when a real appended canvas follows", () => {
  // If the persona body contains "[Channel Canvas]\n..." with only a single
  // preceding newline AND a real appended canvas (double-newline boundary)
  // follows, the LAST-occurrence guard picks the appended one as the real
  // boundary, leaving the embedded one in the System body.
  const framed = [
    "[System]",
    "persona preamble",
    "[Channel Canvas]",
    "this is inside the persona body — NOT the real canvas",
    "",
    "[Channel Canvas]",
    "this IS the appended canvas",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    {
      title: "System",
      body: "persona preamble\n[Channel Canvas]\nthis is inside the persona body — NOT the real canvas",
    },
    { title: "Channel Canvas", body: "this IS the appended canvas" },
  ]);
});

// ── Team Instructions extraction ───────────────────────────────────────────

test("parseSystemPromptSections extracts Team Instructions as its own section after System (Base+System+Team)", () => {
  // compose_prompt() appends "\n\n---\n# Team Instructions\n{instructions}" to the persona body.
  // The canonical delimiter must split System from Team Instructions.
  const framed = [
    "[Base]",
    "You are a helpful assistant.",
    "",
    "[System]",
    "You are Agent X.",
    "",
    "---",
    "# Team Instructions",
    "Always respond in markdown.",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Base", body: "You are a helpful assistant." },
    { title: "System", body: "You are Agent X." },
    { title: "Team Instructions", body: "Always respond in markdown." },
  ]);
});

test("parseSystemPromptSections extracts Team Instructions after System-only (no Base)", () => {
  // When [Base] is absent the [System] header starts the input.
  const framed = [
    "[System]",
    "You are Agent Y.",
    "",
    "---",
    "# Team Instructions",
    "Keep responses concise.",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "System", body: "You are Agent Y." },
    { title: "Team Instructions", body: "Keep responses concise." },
  ]);
});

test("parseSystemPromptSections extracts Team Instructions with Core Memory and Channel Canvas (full 5-section shape)", () => {
  // Full production-shaped system prompt: Base → System → Team Instructions → Core Memory → Channel Canvas.
  // compose_prompt() produces the canonical delimiter; with_core() and with_canvas() append their frames.
  const framed = [
    "[Base]",
    "You are a helpful AI assistant running in Buzz.",
    "",
    "[System]",
    "You are Observer Agent. You coordinate multi-agent workflows.",
    "",
    "---",
    "# Team Instructions",
    "Always tag on handoff.",
    "Never expand scope without approval.",
    "",
    "[Agent Memory — core]",
    "I am Observer Agent.",
    "## Lessons Learned",
    "Always tag on handoff.",
    "",
    "[Channel Canvas]",
    "Canvas revision (event ID): a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2",
    "Last modified: 2026-07-11T10:00:00Z",
    "Fetch current content with: buzz canvas get --channel 94a444a4-c0a3-5966-ab05-530c6ddc2301",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Base", body: "You are a helpful AI assistant running in Buzz." },
    {
      title: "System",
      body: "You are Observer Agent. You coordinate multi-agent workflows.",
    },
    {
      title: "Team Instructions",
      body: "Always tag on handoff.\nNever expand scope without approval.",
    },
    {
      title: "Core Memory",
      body: "I am Observer Agent.\n## Lessons Learned\nAlways tag on handoff.",
    },
    {
      title: "Channel Canvas",
      body: "Canvas revision (event ID): a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2\nLast modified: 2026-07-11T10:00:00Z\nFetch current content with: buzz canvas get --channel 94a444a4-c0a3-5966-ab05-530c6ddc2301",
    },
  ]);
});

test("parseSystemPromptSections does NOT split on a bare '---' without the '# Team Instructions' heading", () => {
  // A horizontal rule alone inside a persona body is kept literal.
  const framed = [
    "[System]",
    "Some text.",
    "",
    "---",
    "",
    "More text after a separator.",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    {
      title: "System",
      body: "Some text.\n\n---\n\nMore text after a separator.",
    },
  ]);
});

test("parseSystemPromptSections does NOT split on '# Team Instructions' with only a single preceding newline", () => {
  // The canonical delimiter requires \n\n---\n before the heading.
  // A single-newline variant is kept literal inside System.
  const framed = [
    "[System]",
    "Persona preamble.",
    "---",
    "# Team Instructions",
    "These look canonical but lack the double newline before ---.",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    {
      title: "System",
      body: "Persona preamble.\n---\n# Team Instructions\nThese look canonical but lack the double newline before ---.",
    },
  ]);
});

test("parseSystemPromptSections does NOT split on '# Team Instructions' heading without the '---' separator", () => {
  // Only the exact composed form \n\n---\n# Team Instructions\n triggers the split.
  const framed = [
    "[System]",
    "Persona text.",
    "",
    "# Team Instructions",
    "This is just a heading in the persona body.",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    {
      title: "System",
      body: "Persona text.\n\n# Team Instructions\nThis is just a heading in the persona body.",
    },
  ]);
});

test("parseSystemPromptSections non-team persona (no delimiter) is unaffected", () => {
  // A standard Base+System prompt without any team instructions must produce
  // exactly two sections — no Team Instructions row added.
  const framed = [
    "[Base]",
    "You are a helpful assistant.",
    "",
    "[System]",
    "You are a coding assistant.",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Base", body: "You are a helpful assistant." },
    { title: "System", body: "You are a coding assistant." },
  ]);
});

test("parseSystemPromptSections splits on the LAST occurrence of the canonical delimiter (embedded lookalike + real appended team suffix)", () => {
  // A persona body may itself contain the exact delimiter string verbatim
  // (e.g. an example or a quoted earlier instruction set). compose_prompt()
  // always APPENDS the real team instructions, so the LAST occurrence is the
  // authoritative producer boundary. The earlier embedded occurrence must stay
  // inside the System body.
  const framed = [
    "[System]",
    "Here is an example of team framing:",
    "",
    "---",
    "# Team Instructions",
    "These are fake — embedded in the persona prose.",
    "",
    "---",
    "# Team Instructions",
    "These are real — appended by compose_prompt().",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    {
      title: "System",
      body: "Here is an example of team framing:\n\n---\n# Team Instructions\nThese are fake — embedded in the persona prose.",
    },
    {
      title: "Team Instructions",
      body: "These are real — appended by compose_prompt().",
    },
  ]);
});

// ── Modern [Team Instructions] bracket-header extraction ─────────────────────

test("parseSystemPromptSections (modern) extracts bracket Team Instructions after Base+System", () => {
  // with_team() emits "\n\n[Team Instructions]\n{instructions}" as a top-level
  // bracket section — the same framing used by with_core() and with_canvas().
  const framed = [
    "[Base]",
    "You are a helpful assistant.",
    "",
    "[System]",
    "You are Agent X.",
    "",
    "[Team Instructions]",
    "Always respond in markdown.",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Base", body: "You are a helpful assistant." },
    { title: "System", body: "You are Agent X." },
    { title: "Team Instructions", body: "Always respond in markdown." },
  ]);
});

test("parseSystemPromptSections (modern) extracts bracket Team Instructions after System-only", () => {
  // [Base] absent; [Team Instructions] is a direct top-level bracket section.
  const framed = [
    "[System]",
    "You are Agent Y.",
    "",
    "[Team Instructions]",
    "Keep responses concise.",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "System", body: "You are Agent Y." },
    { title: "Team Instructions", body: "Keep responses concise." },
  ]);
});

test("parseSystemPromptSections (modern) handles bracket Team Instructions as start-of-string", () => {
  // with_team() also handles prompt=None → "[Team Instructions]\n{instructions}".
  const framed = ["[Team Instructions]", "Instructions only."].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Team Instructions", body: "Instructions only." },
  ]);
});

test("parseSystemPromptSections (modern) pins full 5-section shape: Base+System+Team+Core+Canvas", () => {
  // Production shape from with_team() + with_core() + with_canvas(): all five sections present.
  const framed = [
    "[Base]",
    "You are a helpful AI assistant running in Buzz.",
    "",
    "[System]",
    "You are Observer Agent. You coordinate multi-agent workflows.",
    "",
    "[Team Instructions]",
    "Always tag on handoff.",
    "Never expand scope without approval.",
    "",
    "[Agent Memory — core]",
    "I am Observer Agent.",
    "## Lessons Learned",
    "Always tag on handoff.",
    "",
    "[Channel Canvas]",
    "Canvas revision (event ID): a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2",
    "Last modified: 2026-07-11T10:00:00Z",
    "Fetch current content with: buzz canvas get --channel 94a444a4-c0a3-5966-ab05-530c6ddc2301",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    { title: "Base", body: "You are a helpful AI assistant running in Buzz." },
    {
      title: "System",
      body: "You are Observer Agent. You coordinate multi-agent workflows.",
    },
    {
      title: "Team Instructions",
      body: "Always tag on handoff.\nNever expand scope without approval.",
    },
    {
      title: "Core Memory",
      body: "I am Observer Agent.\n## Lessons Learned\nAlways tag on handoff.",
    },
    {
      title: "Channel Canvas",
      body: "Canvas revision (event ID): a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2c3d4e5f6a1b2\nLast modified: 2026-07-11T10:00:00Z\nFetch current content with: buzz canvas get --channel 94a444a4-c0a3-5966-ab05-530c6ddc2301",
    },
  ]);
});

test("parseSystemPromptSections (modern) does NOT split on bracket [Team Instructions] preceded by only a single newline", () => {
  // The inline marker is "\n\n[Team Instructions]\n" — a single preceding newline
  // must be kept literal inside System, same guard as canvas/core.
  const framed = [
    "[System]",
    "Persona preamble.",
    "[Team Instructions]",
    "This is persona text, not a real team block.",
  ].join("\n");
  const sections = parseSystemPromptSections(framed);
  assert.deepEqual(sections, [
    {
      title: "System",
      body: "Persona preamble.\n[Team Instructions]\nThis is persona text, not a real team block.",
    },
  ]);
});