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

import type { ConversationRecord, WorkspaceRecord } from "../src/types/index.ts";
import { initialWorkbenchConversationIdFromLocation, nextExplicitWorkbenchConversationId, resolveWorkbenchActiveConversationId } from "../src/stores/workbench-projection.ts";
import { createWorkbenchServerState, reduceWorkbenchServerState, selectActiveConversation, selectActiveMessages, selectTraceAuthorityById } from "../src/stores/workbench-server-state.ts";

test("R5 active conversation comes only from route or explicit UI selection", () => {
  assert.equal(resolveWorkbenchActiveConversationId({ switchingConversationId: null, explicitConversationId: "cnv_route" }), "cnv_route");
  assert.equal(resolveWorkbenchActiveConversationId({ switchingConversationId: "cnv_switching", explicitConversationId: "cnv_route" }), "cnv_switching");
  assert.equal(resolveWorkbenchActiveConversationId({ switchingConversationId: null, explicitConversationId: null }), null);
});

test("R5 initial route parsing ignores workspace fallback", () => {
  assert.equal(initialWorkbenchConversationIdFromLocation({ pathname: "/workbench/sessions/cnv_route" }), "cnv_route");
  assert.equal(initialWorkbenchConversationIdFromLocation({ pathname: "/workbench" }), null);
  assert.equal(nextExplicitWorkbenchConversationId("cnv_existing", "not-a-conversation"), "cnv_existing");
});

test("R5 late workspace and trace snapshots update cache without changing active projection", () => {
  const workspace: WorkspaceRecord = { workspaceId: "wsp_r5", projectId: "prj_hwpod_workbench", selectedConversationId: "cnv_late" };
  const routeConversation: ConversationRecord = { conversationId: "cnv_route", sessionId: "ses_route", messages: [{ id: "msg_route", messageId: "msg_route", role: "user", title: "用户", text: "route prompt", status: "sent", createdAt: "2026-06-17T00:00:00.000Z", conversationId: "cnv_route" }] };
  const lateConversation: ConversationRecord = { conversationId: "cnv_late", sessionId: "ses_late", messages: [{ id: "msg_late", messageId: "msg_late", role: "agent", title: "Code Agent", text: "late reply", status: "completed", createdAt: "2026-06-17T00:00:01.000Z", conversationId: "cnv_late", traceId: "trc_late" }] };
  let state = createWorkbenchServerState();
  const active = "cnv_route";

  state = reduceWorkbenchServerState(state, { type: "workspace.snapshot", projectId: "prj_hwpod_workbench", workspace });
  state = reduceWorkbenchServerState(state, { type: "conversation.detail", conversation: lateConversation });
  state = reduceWorkbenchServerState(state, { type: "trace.snapshot", traceId: "trc_late", trace: { traceId: "trc_late", status: "completed", events: [{ seq: 1, type: "late" }] } });

  assert.equal(selectActiveConversation(state, active), null);
  assert.deepEqual(selectActiveMessages(state, active), []);
  assert.equal(selectTraceAuthorityById(state).trc_late?.traceId, "trc_late");

  state = reduceWorkbenchServerState(state, { type: "conversation.detail", conversation: routeConversation });
  assert.equal(selectActiveConversation(state, active)?.conversationId, "cnv_route");
  assert.deepEqual(selectActiveMessages(state, active).map((message) => message.id), ["msg_route"]);
});