import { beforeEach, describe, expect, it, vi } from "vitest";
import type { SkillCommandSpec } from "../../agents/skills.js";
import type { SessionEntry } from "../../config/sessions.js";
import type { TemplateContext } from "../templating.js";
import { clearInlineDirectives } from "./get-reply-directives-utils.js";
import { stripInlineStatus } from "./reply-inline.js";
import { buildTestCtx } from "./test-ctx.js";
import type { TypingController } from "./typing.js";

const handleCommandsMock = vi.fn();
const getChannelPluginMock = vi.fn();
const createOpenClawToolsMock = vi.fn();
const buildStatusReplyMock = vi.fn();

let handleInlineActions: typeof import("./get-reply-inline-actions.js").handleInlineActions;
type HandleInlineActionsInput = Parameters<
  typeof import("./get-reply-inline-actions.js").handleInlineActions
>[0];

async function loadFreshInlineActionsModuleForTest() {
  vi.resetModules();
  vi.doMock("./commands.runtime.js", () => ({
    handleCommands: (...args: unknown[]) => handleCommandsMock(...args),
    buildStatusReply: (...args: unknown[]) => buildStatusReplyMock(...args),
  }));
  vi.doMock("../../agents/openclaw-tools.runtime.js", () => ({
    createOpenClawTools: (...args: unknown[]) => createOpenClawToolsMock(...args),
  }));
  vi.doMock("../../channels/plugins/index.js", async (importOriginal) => {
    const actual = await importOriginal<typeof import("../../channels/plugins/index.js")>();
    return {
      ...actual,
      getChannelPlugin: (...args: unknown[]) => getChannelPluginMock(...args),
    };
  });
  ({ handleInlineActions } = await import("./get-reply-inline-actions.js"));
}

const createTypingController = (): TypingController => ({
  onReplyStart: async () => {},
  startTypingLoop: async () => {},
  startTypingOnText: async () => {},
  refreshTypingTtl: () => {},
  isActive: () => false,
  markRunComplete: () => {},
  markDispatchIdle: () => {},
  cleanup: vi.fn(),
});

const createHandleInlineActionsInput = (params: {
  ctx: ReturnType<typeof buildTestCtx>;
  typing: TypingController;
  cleanedBody: string;
  command?: Partial<HandleInlineActionsInput["command"]>;
  overrides?: Partial<Omit<HandleInlineActionsInput, "ctx" | "sessionCtx" | "typing" | "command">>;
}): HandleInlineActionsInput => {
  const baseCommand: HandleInlineActionsInput["command"] = {
    surface: "whatsapp",
    channel: "whatsapp",
    channelId: "whatsapp",
    ownerList: [],
    senderIsOwner: false,
    isAuthorizedSender: false,
    senderId: undefined,
    abortKey: "whatsapp:+999",
    rawBodyNormalized: params.cleanedBody,
    commandBodyNormalized: params.cleanedBody,
    from: "whatsapp:+999",
    to: "whatsapp:+999",
  };
  return {
    ctx: params.ctx,
    sessionCtx: params.ctx as unknown as TemplateContext,
    cfg: {},
    agentId: "main",
    sessionKey: "s:main",
    workspaceDir: "/tmp",
    isGroup: false,
    typing: params.typing,
    allowTextCommands: false,
    inlineStatusRequested: false,
    command: {
      ...baseCommand,
      ...params.command,
    },
    directives: clearInlineDirectives(params.cleanedBody),
    cleanedBody: params.cleanedBody,
    elevatedEnabled: false,
    elevatedAllowed: false,
    elevatedFailures: [],
    defaultActivation: () => "always",
    resolvedThinkLevel: undefined,
    resolvedVerboseLevel: undefined,
    resolvedReasoningLevel: "off",
    resolvedElevatedLevel: "off",
    resolveDefaultThinkingLevel: async () => "off",
    provider: "openai",
    model: "gpt-4o-mini",
    contextTokens: 0,
    abortedLastRun: false,
    sessionScope: "per-sender",
    ...params.overrides,
  };
};

async function expectInlineActionSkipped(params: {
  ctx: ReturnType<typeof buildTestCtx>;
  typing: TypingController;
  cleanedBody: string;
  command?: Partial<HandleInlineActionsInput["command"]>;
  overrides?: Partial<Omit<HandleInlineActionsInput, "ctx" | "sessionCtx" | "typing" | "command">>;
}) {
  const result = await handleInlineActions(createHandleInlineActionsInput(params));
  expect(result).toEqual({ kind: "reply", reply: undefined });
  expect(params.typing.cleanup).toHaveBeenCalled();
  expect(handleCommandsMock).not.toHaveBeenCalled();
}

describe("handleInlineActions", () => {
  beforeEach(async () => {
    handleCommandsMock.mockReset();
    handleCommandsMock.mockResolvedValue({ shouldContinue: true, reply: undefined });
    getChannelPluginMock.mockReset();
    createOpenClawToolsMock.mockReset();
    buildStatusReplyMock.mockReset();
    buildStatusReplyMock.mockResolvedValue({ text: "status" });
    createOpenClawToolsMock.mockReturnValue([]);
    getChannelPluginMock.mockImplementation((channelId?: string) =>
      channelId === "whatsapp" ? { commands: { skipWhenConfigEmpty: true } } : undefined,
    );
    await loadFreshInlineActionsModuleForTest();
  });

  it("skips whatsapp replies when config is empty and From !== To", async () => {
    const typing = createTypingController();

    const ctx = buildTestCtx({
      From: "whatsapp:+999",
      To: "whatsapp:+123",
      Body: "hi",
    });
    await expectInlineActionSkipped({
      ctx,
      typing,
      cleanedBody: "hi",
      command: { to: "whatsapp:+123" },
    });
  });

  it("forwards agentDir into handleCommands", async () => {
    const typing = createTypingController();

    handleCommandsMock.mockResolvedValue({ shouldContinue: false, reply: { text: "done" } });

    const ctx = buildTestCtx({
      Body: "/status",
      CommandBody: "/status",
    });
    const agentDir = "/tmp/inline-agent";

    const result = await handleInlineActions(
      createHandleInlineActionsInput({
        ctx,
        typing,
        cleanedBody: "/status",
        command: {
          isAuthorizedSender: true,
          senderId: "sender-1",
          abortKey: "sender-1",
        },
        overrides: {
          cfg: { commands: { text: true } },
          agentDir,
        },
      }),
    );

    expect(result).toEqual({ kind: "reply", reply: { text: "done" } });
    expect(handleCommandsMock).toHaveBeenCalledTimes(1);
    expect(handleCommandsMock).toHaveBeenCalledWith(
      expect.objectContaining({
        agentDir,
      }),
    );
  });

  it("does not run command handlers after replying to an inline status-only turn", async () => {
    const typing = createTypingController();
    const ctx = buildTestCtx({
      Body: "/status",
      CommandBody: "/status",
    });

    const result = await handleInlineActions(
      createHandleInlineActionsInput({
        ctx,
        typing,
        cleanedBody: stripInlineStatus("/status").cleaned,
        command: {
          isAuthorizedSender: true,
          rawBodyNormalized: "/status",
          commandBodyNormalized: "/status",
        },
        overrides: {
          allowTextCommands: true,
          inlineStatusRequested: true,
        },
      }),
    );

    expect(result).toEqual({ kind: "reply", reply: undefined });
    expect(buildStatusReplyMock).toHaveBeenCalledTimes(1);
    expect(handleCommandsMock).not.toHaveBeenCalled();
    expect(typing.cleanup).toHaveBeenCalled();
  });

  it("does not continue into the agent after a mention-wrapped inline status-only turn", async () => {
    const typing = createTypingController();
    const ctx = buildTestCtx({
      Body: "<@123> /status",
      CommandBody: "<@123> /status",
      Provider: "discord",
      Surface: "discord",
      ChatType: "channel",
      WasMentioned: true,
    });

    const result = await handleInlineActions(
      createHandleInlineActionsInput({
        ctx,
        typing,
        cleanedBody: "<@123>",
        command: {
          surface: "discord",
          channel: "discord",
          channelId: "discord",
          isAuthorizedSender: true,
          rawBodyNormalized: "<@123> /status",
          commandBodyNormalized: "<@123> /status",
        },
        overrides: {
          allowTextCommands: true,
          inlineStatusRequested: true,
          isGroup: true,
        },
      }),
    );

    expect(result).toEqual({ kind: "reply", reply: undefined });
    expect(buildStatusReplyMock).toHaveBeenCalledTimes(1);
    expect(handleCommandsMock).not.toHaveBeenCalled();
    expect(typing.cleanup).toHaveBeenCalled();
  });

  it("skips stale queued messages that are at or before the /stop cutoff", async () => {
    const typing = createTypingController();
    const sessionEntry: SessionEntry = {
      sessionId: "session-1",
      updatedAt: Date.now(),
      abortCutoffMessageSid: "42",
      abortedLastRun: true,
    };
    const sessionStore = { "s:main": sessionEntry };
    const ctx = buildTestCtx({
      Body: "old queued message",
      CommandBody: "old queued message",
      MessageSid: "41",
    });

    await expectInlineActionSkipped({
      ctx,
      typing,
      cleanedBody: "old queued message",
      command: {
        rawBodyNormalized: "old queued message",
        commandBodyNormalized: "old queued message",
      },
      overrides: {
        sessionEntry,
        sessionStore,
      },
    });
  });

  it("clears /stop cutoff when a newer message arrives", async () => {
    const typing = createTypingController();
    const sessionEntry: SessionEntry = {
      sessionId: "session-2",
      updatedAt: Date.now(),
      abortCutoffMessageSid: "42",
      abortedLastRun: true,
    };
    const sessionStore = { "s:main": sessionEntry };
    handleCommandsMock.mockResolvedValue({ shouldContinue: false, reply: { text: "ok" } });
    const ctx = buildTestCtx({
      Body: "new message",
      CommandBody: "new message",
      MessageSid: "43",
    });

    const result = await handleInlineActions(
      createHandleInlineActionsInput({
        ctx,
        typing,
        cleanedBody: "new message",
        command: {
          rawBodyNormalized: "new message",
          commandBodyNormalized: "new message",
        },
        overrides: {
          sessionEntry,
          sessionStore,
        },
      }),
    );

    expect(result).toEqual({
      kind: "continue",
      directives: clearInlineDirectives("new message"),
      abortedLastRun: false,
    });
    expect(sessionStore["s:main"]?.abortCutoffMessageSid).toBeUndefined();
    expect(sessionStore["s:main"]?.abortCutoffTimestamp).toBeUndefined();
    expect(handleCommandsMock).not.toHaveBeenCalled();
  });

  it("rewrites Claude bundle markdown commands into a native agent prompt", async () => {
    const typing = createTypingController();
    handleCommandsMock.mockResolvedValue({ shouldContinue: false, reply: { text: "done" } });
    const ctx = buildTestCtx({
      Body: "/office_hours build me a deployment plan",
      CommandBody: "/office_hours build me a deployment plan",
    });
    const skillCommands: SkillCommandSpec[] = [
      {
        name: "office_hours",
        skillName: "office-hours",
        description: "Office hours",
        promptTemplate: "Act as an engineering advisor.\n\nFocus on:\n$ARGUMENTS",
        sourceFilePath: "/tmp/plugin/commands/office-hours.md",
      },
    ];

    const result = await handleInlineActions(
      createHandleInlineActionsInput({
        ctx,
        typing,
        cleanedBody: "/office_hours build me a deployment plan",
        command: {
          isAuthorizedSender: true,
          rawBodyNormalized: "/office_hours build me a deployment plan",
          commandBodyNormalized: "/office_hours build me a deployment plan",
        },
        overrides: {
          allowTextCommands: true,
          cfg: { commands: { text: true } },
          skillCommands,
        },
      }),
    );

    expect(result).toEqual({ kind: "reply", reply: { text: "done" } });
    expect(ctx.Body).toBe(
      "Act as an engineering advisor.\n\nFocus on:\nbuild me a deployment plan",
    );
    expect(handleCommandsMock).toHaveBeenCalledWith(
      expect.objectContaining({
        ctx: expect.objectContaining({
          Body: "Act as an engineering advisor.\n\nFocus on:\nbuild me a deployment plan",
        }),
      }),
    );
  });

  it("passes requesterAgentIdOverride into inline tool runtimes", async () => {
    const typing = createTypingController();
    const toolExecute = vi.fn(async () => ({ text: "spawned" }));
    createOpenClawToolsMock.mockReturnValue([
      {
        name: "sessions_spawn",
        execute: toolExecute,
      },
    ]);

    const ctx = buildTestCtx({
      Body: "/spawn_subagent investigate",
      CommandBody: "/spawn_subagent investigate",
    });
    const skillCommands: SkillCommandSpec[] = [
      {
        name: "spawn_subagent",
        skillName: "spawn-subagent",
        description: "Spawn a subagent",
        dispatch: {
          kind: "tool",
          toolName: "sessions_spawn",
          argMode: "raw",
        },
        sourceFilePath: "/tmp/plugin/commands/spawn-subagent.md",
      },
    ];

    const result = await handleInlineActions(
      createHandleInlineActionsInput({
        ctx,
        typing,
        cleanedBody: "/spawn_subagent investigate",
        command: {
          isAuthorizedSender: true,
          senderId: "sender-1",
          senderIsOwner: true,
          abortKey: "sender-1",
          rawBodyNormalized: "/spawn_subagent investigate",
          commandBodyNormalized: "/spawn_subagent investigate",
        },
        overrides: {
          cfg: { commands: { text: true } },
          agentId: "named-worker",
          allowTextCommands: true,
          skillCommands,
        },
      }),
    );

    expect(result).toEqual({ kind: "reply", reply: { text: "✅ Done." } });
    expect(createOpenClawToolsMock).toHaveBeenCalledWith(
      expect.objectContaining({
        requesterAgentIdOverride: "named-worker",
      }),
    );
    expect(toolExecute).toHaveBeenCalled();
  });
});