import { beforeEach, describe, vi } from "vitest";
import { __testing as discordThreadBindingTesting } from "../../../../extensions/discord/src/monitor/thread-bindings.manager.js";
import { __testing as feishuThreadBindingTesting } from "../../../../extensions/feishu/src/thread-bindings.js";
import { resetMatrixThreadBindingsForTests } from "../../../../extensions/matrix/api.js";
import { __testing as telegramThreadBindingTesting } from "../../../../extensions/telegram/src/thread-bindings.js";
import { __testing as sessionBindingTesting } from "../../../infra/outbound/session-binding-service.js";
import {
  actionContractRegistry,
  directoryContractRegistry,
  pluginContractRegistry,
  sessionBindingContractRegistry,
  setupContractRegistry,
  statusContractRegistry,
  surfaceContractRegistry,
  threadingContractRegistry,
} from "./registry.js";
import {
  installChannelActionsContractSuite,
  installChannelDirectoryContractSuite,
  installChannelPluginContractSuite,
  installChannelSetupContractSuite,
  installChannelStatusContractSuite,
  installChannelSurfaceContractSuite,
  installChannelThreadingContractSuite,
  installSessionBindingContractSuite,
} from "./suites.js";

vi.mock("../../../../extensions/matrix/src/matrix/send.js", async () => {
  const actual = await vi.importActual<
    typeof import("../../../../extensions/matrix/src/matrix/send.js")
  >("../../../../extensions/matrix/src/matrix/send.js");
  return {
    ...actual,
    sendMessageMatrix: vi.fn(
      async (_to: string, _message: string, opts?: { threadId?: string }) => ({
        messageId: opts?.threadId ? "$reply" : "$root",
        roomId: "!room:example",
      }),
    ),
  };
});

for (const entry of pluginContractRegistry) {
  describe(`${entry.id} plugin contract`, () => {
    installChannelPluginContractSuite({
      plugin: entry.plugin,
    });
  });
}

for (const entry of actionContractRegistry) {
  describe(`${entry.id} actions contract`, () => {
    installChannelActionsContractSuite({
      plugin: entry.plugin,
      cases: entry.cases as never,
      unsupportedAction: entry.unsupportedAction as never,
    });
  });
}

for (const entry of setupContractRegistry) {
  describe(`${entry.id} setup contract`, () => {
    installChannelSetupContractSuite({
      plugin: entry.plugin,
      cases: entry.cases as never,
    });
  });
}

for (const entry of statusContractRegistry) {
  describe(`${entry.id} status contract`, () => {
    installChannelStatusContractSuite({
      plugin: entry.plugin,
      cases: entry.cases as never,
    });
  });
}

for (const entry of surfaceContractRegistry) {
  for (const surface of entry.surfaces) {
    describe(`${entry.id} ${surface} surface contract`, () => {
      installChannelSurfaceContractSuite({
        plugin: entry.plugin,
        surface,
      });
    });
  }
}

for (const entry of threadingContractRegistry) {
  describe(`${entry.id} threading contract`, () => {
    installChannelThreadingContractSuite({
      plugin: entry.plugin,
    });
  });
}

for (const entry of directoryContractRegistry) {
  describe(`${entry.id} directory contract`, () => {
    installChannelDirectoryContractSuite({
      plugin: entry.plugin,
      coverage: entry.coverage,
      cfg: entry.cfg,
      accountId: entry.accountId,
    });
  });
}

describe("session binding contract registry", () => {
  beforeEach(async () => {
    sessionBindingTesting.resetSessionBindingAdaptersForTests();
    discordThreadBindingTesting.resetThreadBindingsForTests();
    feishuThreadBindingTesting.resetFeishuThreadBindingsForTests();
    resetMatrixThreadBindingsForTests();
    await telegramThreadBindingTesting.resetTelegramThreadBindingsForTests();
  });

  for (const entry of sessionBindingContractRegistry) {
    describe(`${entry.id} session binding contract`, () => {
      installSessionBindingContractSuite({
        expectedCapabilities: entry.expectedCapabilities,
        getCapabilities: entry.getCapabilities,
        bindAndResolve: entry.bindAndResolve,
        unbindAndVerify: entry.unbindAndVerify,
        cleanup: entry.cleanup,
      });
    });
  }
});