import fs from "node:fs/promises";
import os from "node:os";
import path from "node:path";
import { describe, expect, it, vi } from "vitest";

vi.mock("../infra/update-runner.js", () => ({
  runGatewayUpdate: vi.fn(async () => ({
    status: "ok",
    mode: "git",
    root: "/repo",
    steps: [],
    durationMs: 12,
  })),
}));

import {
  connectOk,
  installGatewayTestHooks,
  onceMessage,
  startServerWithClient,
} from "./test-helpers.js";

installGatewayTestHooks();

async function waitForSignal(check: () => boolean, timeoutMs = 2000) {
  const start = Date.now();
  while (Date.now() - start < timeoutMs) {
    if (check()) return;
    await new Promise((resolve) => setTimeout(resolve, 10));
  }
  throw new Error("timeout");
}

describe("gateway update.run", () => {
  it("writes sentinel and schedules restart", async () => {
    const sigusr1 = vi.fn();
    process.on("SIGUSR1", sigusr1);

    const { server, ws } = await startServerWithClient();
    await connectOk(ws);

    const id = "req-update";
    ws.send(
      JSON.stringify({
        type: "req",
        id,
        method: "update.run",
        params: {
          sessionKey: "agent:main:whatsapp:dm:+15555550123",
          restartDelayMs: 0,
        },
      }),
    );
    const res = await onceMessage<{ ok: boolean; payload?: unknown }>(
      ws,
      (o) => o.type === "res" && o.id === id,
    );
    expect(res.ok).toBe(true);

    await waitForSignal(() => sigusr1.mock.calls.length > 0);
    expect(sigusr1).toHaveBeenCalled();

    const sentinelPath = path.join(os.homedir(), ".clawdbot", "restart-sentinel.json");
    const raw = await fs.readFile(sentinelPath, "utf-8");
    const parsed = JSON.parse(raw) as {
      payload?: { kind?: string; stats?: { mode?: string } };
    };
    expect(parsed.payload?.kind).toBe("update");
    expect(parsed.payload?.stats?.mode).toBe("git");

    ws.close();
    await server.close();
    process.off("SIGUSR1", sigusr1);
  });
});