import { describe, expect, it } from "bun:test";

import {
  generateSessionTitle,
  shouldGenerateSessionTitle,
} from "../../src/routes/chatSession.js";
import { type OpencodeRuntimeAdapter } from "../../src/runtime/opencode.js";

describe("shouldGenerateSessionTitle", () => {
  it("allows auto-title generation for the first turn when the title was not edited", () => {
    expect(
      shouldGenerateSessionTitle({
        recentTurnCount: 0,
        isTitleManuallyEdited: false,
      }),
    ).toBe(true);
  });

  it("blocks auto-title generation after the user edits the title manually", () => {
    expect(
      shouldGenerateSessionTitle({
        recentTurnCount: 0,
        isTitleManuallyEdited: true,
      }),
    ).toBe(false);
  });

  it("only allows auto-title generation during the first two turns", () => {
    expect(
      shouldGenerateSessionTitle({
        recentTurnCount: 1,
        isTitleManuallyEdited: false,
      }),
    ).toBe(true);
    expect(
      shouldGenerateSessionTitle({
        recentTurnCount: 2,
        isTitleManuallyEdited: false,
      }),
    ).toBe(false);
  });
});

describe("generateSessionTitle", () => {
  it("uses the current user and assistant turn instead of reading wrapped runtime context", async () => {
    let titlePrompt = "";
    const runtime = {
      createSession: async () => ({ id: "title-session" }),
      prompt: async (_sessionId: string, prompt: string) => {
        titlePrompt = prompt;
      },
      waitForSessionIdle: async () => undefined,
      messages: async () => [
        {
          info: { role: "assistant" },
          parts: [{ type: "text", text: "标题：泵站压力异常排查。" }],
        },
      ],
      abortSession: async () => undefined,
    } as unknown as OpencodeRuntimeAdapter;

    const title = await generateSessionTitle(runtime, {
      sessionId: "chat-session",
      latestUserMessage: "检查一下三号泵站最近压力波动的原因",
      latestAssistantMessage: "三号泵站压力波动主要与夜间阀门开度变化有关。",
      fallbackTitle: "新对话",
    });

    expect(title).toBe("泵站压力异常排查");
    expect(titlePrompt).toContain("用户：检查一下三号泵站最近压力波动的原因");
    expect(titlePrompt).toContain("助手：三号泵站压力波动主要与夜间阀门开度变化有关。");
  });
});