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

import {
  buildTaskFollowUpPrompt,
  formatTaskFollowUpConfidenceLabel,
  normalizeTaskFollowUpSessionId,
  resolveTaskFollowUpAvailability
} from "@/lib/openclaw/domains/task-follow-up";
import type { TaskRecord } from "@/lib/openclaw/types";

test("task follow-up prompt includes operator message, original mission, and latest result", () => {
  const prompt = buildTaskFollowUpPrompt({
    task: createTaskRecord({
      mission: "Research the release checklist and summarize the blockers.",
      metadata: {
        resultPreview: "The release is blocked by a failing smoke check."
      }
    }),
    operatorMessage: "Now fix the smoke check and rerun validation.",
    latestResult: "The release is blocked by a failing smoke check.",
    createdFiles: [{ path: "/workspace/reports/smoke.md", displayPath: "reports/smoke.md" }]
  });

  assert.match(prompt, /Operator follow-up:\nNow fix the smoke check and rerun validation\./);
  assert.match(prompt, /Original mission:\nResearch the release checklist/);
  assert.match(prompt, /Latest result:\nThe release is blocked by a failing smoke check\./);
  assert.match(prompt, /Existing output\/files:\nreports\/smoke\.md/);
});

test("task follow-up availability allows completed tasks with existing context", () => {
  const availability = resolveTaskFollowUpAvailability(createTaskRecord({ status: "completed" }));

  assert.equal(availability.available, true);
  assert.equal(availability.reason, null);
  assert.equal(availability.context.agentId, "agent-1");
  assert.equal(availability.context.sessionId, "session-1");
  assert.equal(availability.context.sessionKey, "agent:agent-1:explicit:session-1");
});

test("task follow-up availability rejects tasks without agent or session context", () => {
  const missingAgent = resolveTaskFollowUpAvailability(createTaskRecord({
    agentIds: [],
    primaryAgentId: undefined
  }));

  assert.equal(missingAgent.available, false);
  assert.equal(missingAgent.reason, "This task does not expose an OpenClaw agent to continue.");

  const missingSession = resolveTaskFollowUpAvailability(createTaskRecord({
    dispatchId: "dispatch-1",
    sessionIds: []
  }));

  assert.equal(missingSession.available, false);
  assert.equal(missingSession.reason, "This task does not expose an OpenClaw session to continue.");
});

test("task follow-up availability can use native task session metadata", () => {
  const availability = resolveTaskFollowUpAvailability(createTaskRecord({
    sessionIds: [],
    metadata: {
      provenance: "native-task",
      openClawTaskId: "task-native-1",
      openClawSessionKey: "agent:agent-1:explicit:session-native"
    }
  }));

  assert.equal(availability.available, true);
  assert.equal(availability.context.openClawTaskId, "task-native-1");
  assert.equal(availability.context.sessionId, "session-native");
  assert.equal(availability.context.sessionKey, "agent:agent-1:explicit:session-native");
  assert.equal(availability.context.confidence, "high");
});

test("task follow-up availability warns for runtime-derived continuation context", () => {
  const availability = resolveTaskFollowUpAvailability(createTaskRecord({
    dispatchId: undefined,
    metadata: {
      provenance: "runtime-derived"
    }
  }));

  assert.equal(availability.available, true);
  assert.equal(availability.context.provenance, "runtime-derived");
  assert.equal(availability.context.confidence, "medium");
  assert.match(availability.warning ?? "", /runtime-derived/);
});

test("task follow-up session ids normalize explicit session keys into plain session ids", () => {
  assert.equal(normalizeTaskFollowUpSessionId("agent:agent-1:explicit:session-1"), "session-1");
  assert.equal(normalizeTaskFollowUpSessionId(" session-plain "), "session-plain");
  assert.equal(normalizeTaskFollowUpSessionId(""), null);
});

test("task follow-up confidence labels are stable for card and composer UX", () => {
  assert.equal(formatTaskFollowUpConfidenceLabel("high"), "high");
  assert.equal(formatTaskFollowUpConfidenceLabel("medium"), "medium warning");
  assert.equal(formatTaskFollowUpConfidenceLabel("none"), "none disabled");
});

function createTaskRecord(overrides: Partial<TaskRecord> = {}): TaskRecord {
  const { metadata, ...restOverrides } = overrides;

  return {
    id: "task-1",
    key: "dispatch:task-1",
    title: "Release checklist",
    mission: "Review the release checklist.",
    subtitle: "Latest task result.",
    status: "completed",
    updatedAt: Date.now(),
    ageMs: 0,
    workspaceId: "workspace-1",
    primaryAgentId: "agent-1",
    primaryAgentName: "Main",
    primaryRuntimeId: "runtime-1",
    dispatchId: "dispatch-1",
    runtimeIds: ["runtime-1"],
    agentIds: ["agent-1"],
    sessionIds: ["session-1"],
    runIds: ["run-1"],
    runtimeCount: 1,
    updateCount: 1,
    liveRunCount: 0,
    artifactCount: 1,
    warningCount: 0,
    ...restOverrides,
    metadata: {
      ...metadata
    }
  };
}
