#!/usr/bin/env npx tsx

/**
 * Regression: `paseo daemon stop` must only act on daemon ownership state and
 * must not discover/kill processes via home-scoped `ps` command heuristics.
 */

import assert from "node:assert";
import { spawn, type ChildProcess } from "node:child_process";
import { mkdtemp, rm } from "node:fs/promises";
import { tmpdir } from "node:os";
import { join } from "node:path";
import { $ } from "zx";

$.verbose = false;

const testEnv = {
  PASEO_LOCAL_SPEECH_AUTO_DOWNLOAD: process.env.PASEO_LOCAL_SPEECH_AUTO_DOWNLOAD ?? "0",
  PASEO_DICTATION_ENABLED: process.env.PASEO_DICTATION_ENABLED ?? "0",
  PASEO_VOICE_MODE_ENABLED: process.env.PASEO_VOICE_MODE_ENABLED ?? "0",
};

function sleep(ms: number): Promise<void> {
  return new Promise((resolve) => setTimeout(resolve, ms));
}

function isProcessRunning(pid: number): boolean {
  if (!Number.isInteger(pid) || pid <= 0) {
    return false;
  }

  try {
    process.kill(pid, 0);
    return true;
  } catch {
    return false;
  }
}

async function waitForRunning(pid: number, timeoutMs: number): Promise<void> {
  const deadline = Date.now() + timeoutMs;
  while (Date.now() < deadline) {
    if (isProcessRunning(pid)) {
      return;
    }
    await sleep(50);
  }
  throw new Error(`Process ${pid} did not become running in time`);
}

console.log("=== Daemon Stop Ownership Regression ===\n");

const paseoHome = await mkdtemp(join(tmpdir(), "paseo-stop-ownership-"));
let decoyProcess: ChildProcess | null = null;

try {
  console.log("Test 1: start decoy process with daemon-like command markers");

  decoyProcess = spawn(
    process.execPath,
    [
      "-e",
      // Keep the process alive long enough for stop command assertions.
      "setInterval(() => {}, 1000)",
      "supervisor-entrypoint.ts",
    ],
    {
      env: {
        ...process.env,
        PASEO_HOME: paseoHome,
      },
      stdio: "ignore",
      detached: process.platform !== "win32",
    },
  );
  decoyProcess.unref();

  const decoyPid = decoyProcess.pid;
  assert(Number.isInteger(decoyPid) && (decoyPid ?? 0) > 0, "decoy pid should exist");
  await waitForRunning(decoyPid!, 5000);
  console.log(`✓ decoy process started (${decoyPid})\n`);

  console.log("Test 2: daemon stop should report not_running and leave decoy untouched");

  const stopResult =
    await $`PASEO_HOME=${paseoHome} PASEO_LOCAL_SPEECH_AUTO_DOWNLOAD=${testEnv.PASEO_LOCAL_SPEECH_AUTO_DOWNLOAD} PASEO_DICTATION_ENABLED=${testEnv.PASEO_DICTATION_ENABLED} PASEO_VOICE_MODE_ENABLED=${testEnv.PASEO_VOICE_MODE_ENABLED} npx paseo daemon stop --home ${paseoHome} --json`.nothrow();
  assert.strictEqual(stopResult.exitCode, 0, `stop should succeed: ${stopResult.stderr}`);

  const parsed = JSON.parse(stopResult.stdout) as { action?: unknown };
  assert.strictEqual(
    parsed.action,
    "not_running",
    `stop should not target decoy process: ${stopResult.stdout}`,
  );
  assert(isProcessRunning(decoyPid!), "decoy process must remain alive after stop");

  console.log("✓ stop is ownership-driven and does not kill decoy process\n");
} finally {
  if (decoyProcess?.pid && isProcessRunning(decoyProcess.pid)) {
    try {
      process.kill(decoyProcess.pid, "SIGTERM");
    } catch {
      // ignore
    }
    await sleep(100);
    if (isProcessRunning(decoyProcess.pid)) {
      try {
        process.kill(decoyProcess.pid, "SIGKILL");
      } catch {
        // ignore
      }
    }
  }

  await $`PASEO_HOME=${paseoHome} npx paseo daemon stop --home ${paseoHome} --force`.nothrow();
  await rm(paseoHome, { recursive: true, force: true });
}

console.log("=== Daemon stop ownership regression test passed ===");