@tonyclaw/agent-inspector 3.1.32 → 4.0.2

package/src/proxy/ecosystemTasks.ts

@@ -1,415 +0,0 @@
-import { Buffer } from "node:buffer";
-import { spawn, type ChildProcess } from "node:child_process";
-import { randomUUID } from "node:crypto";
-import type {
-  EcosystemExecutionPlan,
-  EcosystemTask,
-  EcosystemTaskAction,
-  EcosystemTaskStatus,
-} from "../lib/ecosystemContract";
-import { findEcosystemPackage, findEcosystemRecipe } from "./ecosystemRegistry";
-import {
-  npmCommand,
-  npmSpawnCommand,
-  signalProcessTree,
-  type CommandSpec,
-} from "./platformCommands";
-import { DEFAULT_RUNTIME_BUDGET_POLICY } from "../lib/resourceLimits";
-
-const MAX_TASKS = 50;
-const OUTPUT_LIMIT = 40_000;
-const INSTALL_TIMEOUT_MS = 120_000;
-const HELP_TIMEOUT_MS = 30_000;
-const RECIPE_TIMEOUT_MS = 90_000;
-const SHUTDOWN_GRACE_MS = 2_000;
-
-const tasks = new Map<string, EcosystemTask>();
-type ActiveTask = {
-  child: ChildProcess;
-  closed: Promise<void>;
-  resolveClosed: () => void;
-  timeout: ReturnType<typeof setTimeout>;
-};
-
-const activeTasks = new Map<string, ActiveTask>();
-let acceptingTasks = true;
-let shutdownForceTimer: ReturnType<typeof setTimeout> | null = null;
-
-function activeTaskLimit(): number {
-  const configured = Number(process.env["ECOSYSTEM_TASK_ACTIVE_LIMIT"]);
-  if (Number.isSafeInteger(configured) && configured > 0) return configured;
-  return DEFAULT_RUNTIME_BUDGET_POLICY.queues.backgroundWorkers;
-}
-
-function nowIso(): string {
-  return new Date().toISOString();
-}
-
-function trimOutput(output: string): string {
-  if (output.length <= OUTPUT_LIMIT) return output;
-  return output.slice(output.length - OUTPUT_LIMIT);
-}
-
-function appendTaskOutput(task: EcosystemTask, chunk: string): void {
-  task.output = trimOutput(`${task.output}${chunk}`);
-}
-
-function rememberTask(task: EcosystemTask): void {
-  tasks.set(task.id, task);
-  while (tasks.size > MAX_TASKS) {
-    const oldest = tasks.keys().next().value;
-    if (typeof oldest !== "string") return;
-    tasks.delete(oldest);
-  }
-}
-
-function packageActionArgs(action: EcosystemTaskAction, npmName: string): string[] {
-  switch (action) {
-    case "install":
-    case "upgrade":
-      return ["install", "-g", npmName];
-    case "help":
-      return ["exec", "--yes", npmName, "--", "--help"];
-    case "runner-presets":
-      return ["exec", "--yes", "@tonyclaw/eval-harness", "--", "runner-presets"];
-    case "recipe":
-      return ["exec", "--yes", npmName, "--", "--help"];
-  }
-}
-
-function recipeActionArgs(recipeId: string): string[] | null {
-  const execPrefix = ["exec", "--yes", "@tonyclaw/eval-harness", "--"];
-  switch (recipeId) {
-    case "inspector-mcp-smoke":
-      return [...execPrefix, "check-inspector-mcp"];
-    case "session-evidence-smoke":
-      return [
-        ...execPrefix,
-        "inspector-smoke",
-        "--title",
-        "TonyClaw Lab smoke",
-        "--latest-log-limit",
-        "5",
-      ];
-    case "runner-presets-json":
-      return [...execPrefix, "runner-presets", "--format", "json"];
-    case "local-smoke-batch":
-      return null;
-  }
-
-  return null;
-}
-
-function packageActionTimeout(action: EcosystemTaskAction): number {
-  switch (action) {
-    case "install":
-    case "upgrade":
-      return INSTALL_TIMEOUT_MS;
-    case "help":
-    case "runner-presets":
-      return HELP_TIMEOUT_MS;
-    case "recipe":
-      return RECIPE_TIMEOUT_MS;
-  }
-}
-
-function taskCommand(args: readonly string[]): string {
-  return `${npmCommand()} ${args.join(" ")}`;
-}
-
-export function getEcosystemPackageExecutionPlan(
-  packageId: string,
-  action: EcosystemTaskAction,
-): EcosystemExecutionPlan | null {
-  if (action === "recipe") return null;
-  const definition =
-    action === "runner-presets"
-      ? findEcosystemPackage("eval-harness")
-      : findEcosystemPackage(packageId);
-  if (definition === null) return null;
-  if (action === "runner-presets" && packageId !== "eval-harness") return null;
-
-  const args = packageActionArgs(action, definition.npmName);
-  const mutatesGlobalPackages = action === "install" || action === "upgrade";
-  return {
-    action,
-    targetId: definition.id,
-    targetName: definition.name,
-    command: taskCommand(args),
-    source: `npm registry: ${definition.npmName}`,
-    timeoutMs: packageActionTimeout(action),
-    scope: mutatesGlobalPackages
-      ? "Global npm package installation"
-      : "Ephemeral npm exec process in the Inspector service account",
-  };
-}
-
-export function getEcosystemRecipeExecutionPlan(recipeId: string): EcosystemExecutionPlan | null {
-  const recipe = findEcosystemRecipe(recipeId);
-  if (recipe === null || !recipe.runnable) return null;
-  const definition = findEcosystemPackage(recipe.packageId);
-  if (definition === null) return null;
-  const args = recipeActionArgs(recipe.id);
-  if (args === null) return null;
-
-  return {
-    action: "recipe",
-    targetId: recipe.id,
-    targetName: recipe.title,
-    command: taskCommand(args),
-    source: `Built-in allowlisted recipe using ${definition.npmName}`,
-    timeoutMs: RECIPE_TIMEOUT_MS,
-    scope: "Ephemeral npm exec process in the Inspector service account",
-  };
-}
-
-function completeTask(
-  task: EcosystemTask,
-  status: Exclude<EcosystemTaskStatus, "queued" | "running">,
-  exitCode: number | null,
-  error: string | null,
-): void {
-  task.status = status;
-  task.exitCode = exitCode;
-  task.error = error;
-  task.finishedAt = nowIso();
-}
-
-function finishActiveTask(taskId: string): void {
-  const active = activeTasks.get(taskId);
-  if (active === undefined) return;
-  clearTimeout(active.timeout);
-  activeTasks.delete(taskId);
-  active.resolveClosed();
-  if (activeTasks.size === 0 && shutdownForceTimer !== null) {
-    clearTimeout(shutdownForceTimer);
-    shutdownForceTimer = null;
-  }
-}
-
-function failTaskForShutdown(task: EcosystemTask): void {
-  if (task.status !== "running" && task.status !== "queued") return;
-  const error = "Runtime shutdown terminated the ecosystem command";
-  appendTaskOutput(task, `\n${error}.\n`);
-  completeTask(task, "failed", null, error);
-}
-
-function forceActiveTaskTrees(): void {
-  for (const [taskId, active] of activeTasks) {
-    const pid = active.child.pid;
-    if (pid === undefined) continue;
-    signalProcessTree(pid, "SIGKILL");
-    if (process.platform === "win32") {
-      const task = tasks.get(taskId);
-      if (task !== undefined) failTaskForShutdown(task);
-      finishActiveTask(taskId);
-    }
-  }
-}
-
-function launchTask(task: EcosystemTask, spawnCommand: CommandSpec, timeoutMs: number): void {
-  const child = spawn(spawnCommand.command, spawnCommand.args, {
-    detached: process.platform !== "win32",
-    windowsHide: true,
-    shell: false,
-    stdio: ["ignore", "pipe", "pipe"],
-  });
-  let resolveClosed = (): void => {};
-  const closed = new Promise<void>((resolve) => {
-    resolveClosed = resolve;
-  });
-  const timeout = setTimeout(() => {
-    const pid = child.pid;
-    if (pid !== undefined) signalProcessTree(pid, "SIGKILL");
-    appendTaskOutput(task, `\nTimed out after ${String(timeoutMs / 1000)}s.\n`);
-  }, timeoutMs);
-  activeTasks.set(task.id, { child, closed, resolveClosed, timeout });
-
-  task.status = "running";
-  task.startedAt = nowIso();
-  appendTaskOutput(task, `$ ${task.command}\n`);
-
-  child.stdout?.on("data", (chunk: Buffer) => {
-    appendTaskOutput(task, chunk.toString("utf8"));
-  });
-  child.stderr?.on("data", (chunk: Buffer) => {
-    appendTaskOutput(task, chunk.toString("utf8"));
-  });
-  child.on("error", (error) => {
-    appendTaskOutput(task, `\n${error.message}\n`);
-    if (task.status !== "failed") completeTask(task, "failed", null, error.message);
-    finishActiveTask(task.id);
-  });
-  child.on("close", (code) => {
-    if (task.status !== "failed") {
-      if (code === 0) {
-        completeTask(task, "succeeded", code, null);
-      } else {
-        completeTask(task, "failed", code, `Command exited with code ${String(code ?? "unknown")}`);
-      }
-    }
-    finishActiveTask(task.id);
-  });
-}
-
-export function getEcosystemTask(taskId: string): EcosystemTask | null {
-  return tasks.get(taskId) ?? null;
-}
-
-export function listEcosystemTasks(): EcosystemTask[] {
-  return [...tasks.values()].sort((left, right) => right.createdAt.localeCompare(left.createdAt));
-}
-
-export function startEcosystemTask(
-  packageId: string,
-  action: EcosystemTaskAction,
-): EcosystemTask | null {
-  if (!acceptingTasks) return null;
-  if (activeTasks.size >= activeTaskLimit()) return null;
-  if (action === "recipe") return null;
-  const definition =
-    action === "runner-presets"
-      ? findEcosystemPackage("eval-harness")
-      : findEcosystemPackage(packageId);
-  if (definition === null) return null;
-  if (action === "runner-presets" && packageId !== "eval-harness") return null;
-
-  const args = packageActionArgs(action, definition.npmName);
-  const task: EcosystemTask = {
-    id: randomUUID(),
-    packageId: definition.id,
-    npmName: definition.npmName,
-    recipeId: null,
-    action,
-    status: "queued",
-    command: taskCommand(args),
-    output: "",
-    createdAt: nowIso(),
-    startedAt: null,
-    finishedAt: null,
-    exitCode: null,
-    error: null,
-  };
-  rememberTask(task);
-
-  launchTask(task, npmSpawnCommand(args), packageActionTimeout(action));
-
-  return task;
-}
-
-export function startEcosystemRecipeTask(recipeId: string): EcosystemTask | null {
-  if (!acceptingTasks) return null;
-  if (activeTasks.size >= activeTaskLimit()) return null;
-  const recipe = findEcosystemRecipe(recipeId);
-  if (recipe === null) return null;
-  if (!recipe.runnable) return null;
-  const definition = findEcosystemPackage(recipe.packageId);
-  if (definition === null) return null;
-  const args = recipeActionArgs(recipe.id);
-  if (args === null) return null;
-
-  const task: EcosystemTask = {
-    id: randomUUID(),
-    packageId: definition.id,
-    npmName: definition.npmName,
-    recipeId: recipe.id,
-    action: "recipe",
-    status: "queued",
-    command: taskCommand(args),
-    output: "",
-    createdAt: nowIso(),
-    startedAt: null,
-    finishedAt: null,
-    exitCode: null,
-    error: null,
-  };
-  rememberTask(task);
-
-  launchTask(task, npmSpawnCommand(args), RECIPE_TIMEOUT_MS);
-
-  return task;
-}
-
-/** Stop admitting commands and begin terminating every tracked child process tree. */
-export function beginEcosystemTaskShutdown(): void {
-  acceptingTasks = false;
-  for (const [taskId, active] of activeTasks) {
-    const task = tasks.get(taskId);
-    if (task !== undefined) failTaskForShutdown(task);
-    clearTimeout(active.timeout);
-    const pid = active.child.pid;
-    if (pid !== undefined) {
-      signalProcessTree(pid, process.platform === "win32" ? "SIGKILL" : "SIGTERM");
-      if (process.platform === "win32") finishActiveTask(taskId);
-    }
-  }
-  if (activeTasks.size > 0 && shutdownForceTimer === null) {
-    shutdownForceTimer = setTimeout(() => {
-      shutdownForceTimer = null;
-      forceActiveTaskTrees();
-    }, SHUTDOWN_GRACE_MS);
-    shutdownForceTimer.unref();
-  }
-}
-
-/** Wait until children signalled by beginEcosystemTaskShutdown have closed. */
-export async function waitForEcosystemTasksToClose(): Promise<void> {
-  await Promise.all([...activeTasks.values()].map(async (active) => await active.closed));
-}
-
-/** Immediate deadline fallback used by the runtime shutdown coordinator. */
-export function forceCloseEcosystemTasks(): void {
-  acceptingTasks = false;
-  if (shutdownForceTimer !== null) {
-    clearTimeout(shutdownForceTimer);
-    shutdownForceTimer = null;
-  }
-  for (const taskId of activeTasks.keys()) {
-    const task = tasks.get(taskId);
-    if (task !== undefined) failTaskForShutdown(task);
-  }
-  forceActiveTaskTrees();
-}
-
-export function _startEcosystemTaskForTests(
-  spawnCommand: CommandSpec,
-  timeoutMs = RECIPE_TIMEOUT_MS,
-): EcosystemTask | null {
-  if (!acceptingTasks) return null;
-  if (activeTasks.size >= activeTaskLimit()) return null;
-  const task: EcosystemTask = {
-    id: randomUUID(),
-    packageId: "test-package",
-    npmName: "test-package",
-    recipeId: null,
-    action: "help",
-    status: "queued",
-    command: `${spawnCommand.command} ${spawnCommand.args.join(" ")}`,
-    output: "",
-    createdAt: nowIso(),
-    startedAt: null,
-    finishedAt: null,
-    exitCode: null,
-    error: null,
-  };
-  rememberTask(task);
-  launchTask(task, spawnCommand, timeoutMs);
-  return task;
-}
-
-export function _resetEcosystemTasksForTests(): void {
-  forceCloseEcosystemTasks();
-  tasks.clear();
-  activeTasks.clear();
-  acceptingTasks = true;
-}
-
-export function _getEcosystemTaskAdmissionHealthForTests(): {
-  activeTasks: number;
-  activeTaskLimit: number;
-} {
-  return {
-    activeTasks: activeTasks.size,
-    activeTaskLimit: activeTaskLimit(),
-  };
-}