viber-channel 0.5.2 → 0.6.0

package/viber-codex-supervisor.ts

@@ -0,0 +1,96 @@
+#!/usr/bin/env bun
+/**
+ * viber-codex-supervisor (#280 step-12, slice 4).
+ *
+ * Runs ONE supervisor process that launches and supervises many Codex agents, each
+ * in its own `viber-codex-bridge` child process. This is the local entry point;
+ * remote (server-driven) control of agents is a later slice.
+ *
+ * USAGE
+ *   viber-codex-supervisor --config <path.json>
+ *   VIBER_SUPERVISOR_CONFIG=<path.json> viber-codex-supervisor
+ *
+ * Config: see lib/supervisor_config.ts. Each agent defaults to `--await-invite`, so
+ * the agents come up idle and the web control plane adds them to conversations.
+ *
+ * Ctrl-C / SIGTERM stops every agent and exits.
+ */
+
+import { readFileSync } from "node:fs";
+import { createBridgeSpawn } from "./lib/bridge_spawn.ts";
+import { parseSupervisorConfig } from "./lib/supervisor_config.ts";
+import { type AgentSpec, Supervisor } from "./lib/supervisor.ts";
+
+export function configPathFromArgv(argv: string[]): string | undefined {
+	const eq = argv.find((a) => a.startsWith("--config="));
+	if (eq) return eq.slice("--config=".length);
+	const i = argv.indexOf("--config");
+	if (i !== -1) {
+		// `--config` was given explicitly: it MUST be followed by a real path. A
+		// missing value or a following flag is a usage error — fail loudly rather
+		// than silently falling back to the env var (CLI precedence wins).
+		const val = argv[i + 1];
+		if (!val || val.startsWith("--")) {
+			throw new Error("--config requires a path argument");
+		}
+		return val;
+	}
+	return process.env.VIBER_SUPERVISOR_CONFIG;
+}
+
+export function loadConfig(argv: string[]): AgentSpec[] {
+	const path = configPathFromArgv(argv);
+	if (!path) {
+		throw new Error(
+			"no config: pass --config <path.json> or set VIBER_SUPERVISOR_CONFIG",
+		);
+	}
+	let raw: unknown;
+	try {
+		raw = JSON.parse(readFileSync(path, "utf-8"));
+	} catch (err) {
+		const msg = err instanceof Error ? err.message : String(err);
+		throw new Error(`cannot read/parse config ${path}: ${msg}`);
+	}
+	try {
+		return parseSupervisorConfig(raw);
+	} catch (err) {
+		// Name the file so a validation error points at the offending config.
+		const msg = err instanceof Error ? err.message : String(err);
+		throw new Error(`invalid config ${path}: ${msg}`);
+	}
+}
+
+async function main(argv: string[] = process.argv.slice(2)): Promise<void> {
+	const specs = loadConfig(argv);
+	// Launcher is configurable so the supervisor works both with the installed
+	// `viber-codex-bridge` bin (default) and from source (`bun viber-codex-bridge.ts`).
+	const command = process.env.VIBER_BRIDGE_COMMAND ?? "viber-codex-bridge";
+	const baseArgs = (process.env.VIBER_BRIDGE_BASE_ARGS ?? "").split(" ").filter(Boolean);
+	const supervisor = new Supervisor({ spawn: createBridgeSpawn({ command, baseArgs }) });
+
+	process.stderr.write(`[supervisor] launching ${specs.length} agent(s)\n`);
+	for (const spec of specs) supervisor.start(spec);
+
+	let shuttingDown = false;
+	const shutdown = (sig: NodeJS.Signals): void => {
+		if (shuttingDown) return;
+		shuttingDown = true;
+		process.stderr.write(`[supervisor] ${sig} — stopping all agents\n`);
+		supervisor.stopAll();
+		// Give children a moment to exit on SIGTERM, then leave.
+		setTimeout(() => process.exit(0), 500);
+	};
+	process.on("SIGINT", () => shutdown("SIGINT"));
+	process.on("SIGTERM", () => shutdown("SIGTERM"));
+
+	// Keep the process alive; the supervisor drives everything via callbacks.
+	await new Promise<void>(() => {});
+}
+
+if (import.meta.main) {
+	main().catch((err) => {
+		process.stderr.write(`[supervisor] fatal: ${err instanceof Error ? err.message : String(err)}\n`);
+		process.exit(1);
+	});
+}