import { join } from "node:path";
import { describe, expect, it } from "vitest";
import { SessionStore } from "../src/sessions/store.js";
import { silentLogger, tempRoot } from "./helpers.js";

function makeStore(root: string) {
  return new SessionStore(join(root, "sessions.json"), join(root, "messages"), silentLogger);
}

describe("SessionStore", () => {
  it("creates, lists, and owns sessions", async () => {
    const store = makeStore(tempRoot());
    const session = await store.create({ uid: "1", team_id: "t1" }, { title: "hello" });
    expect((await store.get(session.session_id))?.title).toBe("hello");
    expect((await store.getOwned({ uid: "1", team_id: "t1" }, session.session_id))?.session_id).toBe(
      session.session_id,
    );
    expect(await store.getOwned({ uid: "2", team_id: "t1" }, session.session_id)).toBeUndefined();
    expect((await store.list({ uid: "1" })).total).toBe(1);
    expect((await store.list({ uid: "2" })).total).toBe(0);
  });

  it("appends messages, updates metadata, and auto-titles from the first user message", async () => {
    const store = makeStore(tempRoot());
    const session = await store.create({ uid: "1", team_id: "t1" });
    await store.append(session.session_id, {
      role: "user",
      mode: "agent",
      content: "帮我做一张云南海报",
      media: [],
      status: "ok",
    });
    await store.append(session.session_id, {
      role: "assistant",
      mode: "image",
      content: "已完成",
      media: [{ type: "image", url: "https://img/1.png" }],
      status: "ok",
    });

    const meta = await store.get(session.session_id);
    expect(meta?.message_count).toBe(2);
    expect(meta?.last_mode).toBe("image");
    expect(meta?.title).toBe("帮我做一张云南海报");
    const messages = await store.messages(session.session_id);
    expect(messages).toHaveLength(2);
    expect(messages[1]?.media[0]?.url).toBe("https://img/1.png");
  });

  it("persists the index and paginates", async () => {
    const root = tempRoot();
    const store = makeStore(root);
    for (let index = 0; index < 3; index += 1) {
      await store.create({ uid: "1", team_id: "t1" }, { title: `s${index}` });
    }
    const reloaded = makeStore(root);
    expect((await reloaded.list({ uid: "1" })).total).toBe(3);
    expect((await reloaded.list({ uid: "1", limit: 2 })).sessions).toHaveLength(2);
    expect((await reloaded.list({ uid: "1", limit: 2, offset: 2 })).sessions).toHaveLength(1);
  });
});