import type { Page } from "@playwright/test";
import type { AgentRuntimeAvailability } from "@/lib/types/agent-runtime";

export type AgentRuntimeAvailabilitySnapshot = AgentRuntimeAvailability;

type E2EStoreWindow = Window & {
  __KANDEV_E2E_STORE__?: {
    getState: () => {
      setAgentRuntime: (snapshot: AgentRuntimeAvailabilitySnapshot | null) => void;
    };
  };
};

export async function setAgentRuntimeAvailability(
  page: Page,
  snapshot: AgentRuntimeAvailabilitySnapshot | null,
): Promise<void> {
  await page.evaluate((nextSnapshot) => {
    const store = (window as E2EStoreWindow).__KANDEV_E2E_STORE__;
    if (!store) throw new Error("E2E store bridge missing");
    store.getState().setAgentRuntime(nextSnapshot);
  }, snapshot);
}

export async function stubAgentRuntimeRestart(page: Page): Promise<() => number> {
  let restartRequests = 0;

  await page.route("**/api/v1/system/restart-capability", (route) =>
    route.fulfill({
      status: 200,
      contentType: "application/json",
      body: JSON.stringify({ supported: true, mode: "supervisor", adapter: "supervisor" }),
    }),
  );
  await page.route("**/api/v1/system/info", (route) =>
    route.fulfill({
      status: 200,
      contentType: "application/json",
      body: JSON.stringify({ boot_id: "e2e-stable-boot" }),
    }),
  );
  await page.route("**/api/v1/system/restart", (route) => {
    restartRequests += 1;
    return route.fulfill({
      status: 200,
      contentType: "application/json",
      body: JSON.stringify({ accepted: true, message: "accepted" }),
    });
  });

  return () => restartRequests;
}