@gajae-code/coding-agent 0.11.3 → 0.11.5

package/src/sdk/bus/telegram-daemon-control.ts

@@ -9,7 +9,9 @@
  */
 
 import * as fs from "node:fs";
+import * as os from "node:os";
 import * as path from "node:path";
+import { Process } from "@gajae-code/natives";
 import type { Settings } from "../../config/settings";
 import type {
 	BuiltInDaemonController,
@@ -22,18 +24,31 @@ import type {
 } from "../../daemon/control-types";
 import { OWNERSHIP_MISMATCH_MESSAGE, ownershipMismatchRecovery } from "../../daemon/operator-contract";
 import { resolveGjcRuntimeSpawnInfo } from "../../daemon/runtime";
+import { isProcessIncarnation } from "../broker/process-incarnation";
 import { getNotificationConfig, isTelegramConfigured, tokenFingerprint } from "./config";
+import { exactUnlinkNotificationFile, readNotificationEndpointFile } from "./notification-service";
 import {
+	confirmTelegramDaemonSpawn,
+	type DaemonState,
 	daemonPaths,
+	hasSafeDaemonStateShape,
+	isCurrentCompatibleOwner,
 	isFreshLiveOwner,
+	isSignalableMatchingOwner,
 	readDaemonRoots,
 	readDaemonState,
 	spawnTelegramDaemonOwner,
 	type TelegramDaemonDeps,
 	type TelegramDaemonFs,
+	type TelegramSpawnOwnerResult,
 } from "./telegram-daemon";
 
-const nodeFs: TelegramDaemonFs = fs.promises as unknown as TelegramDaemonFs;
+const nodeFs: TelegramDaemonFs = {
+	...(fs.promises as unknown as TelegramDaemonFs),
+	readEndpointFile: readNotificationEndpointFile,
+	exactUnlink: async (file, identity) =>
+		exactUnlinkNotificationFile(file, identity, `.gjc-delete-daemon-transition-${crypto.randomUUID()}.json`),
+};
 const DEFAULT_GRACEFUL_TIMEOUT_MS = 8_000;
 const DEFAULT_KILL_TIMEOUT_MS = 3_000;
 const DEFAULT_WAIT_STEP_MS = 25;
@@ -92,11 +107,41 @@ export async function clearTelegramControlRequest(
 	await fsImpl.unlink(file).catch(() => undefined);
 }
 
+export interface DaemonProcessReference {
+	incarnation: string;
+	signalRoot(signal: NodeJS.Signals): void;
+}
+
+function defaultProcessReference(pid: number, platform = os.platform()): DaemonProcessReference | undefined {
+	// Process.fromPid on Darwin still resolves identity and then signals a numeric
+	// PID. A process can exit and its PID be reused in that interval, so daemon
+	// control must not use it as privileged signaling authority.
+	if (platform === "darwin") return undefined;
+	try {
+		const processRef = Process.fromPid(pid);
+		if (!processRef || !isProcessIncarnation(processRef.incarnation)) return undefined;
+		return {
+			incarnation: processRef.incarnation,
+			signalRoot: signal => {
+				const nativeSignal = os.constants.signals[signal];
+				if (nativeSignal === undefined) throw new Error(`Unsupported signal: ${signal}`);
+				const rootProcess = processRef as typeof processRef & { signalRoot(signal: number): boolean };
+				if (!rootProcess.signalRoot(nativeSignal)) throw new Error("Pinned process is already gone");
+			},
+		};
+	} catch {
+		return undefined;
+	}
+}
+
 export interface TelegramDaemonControlDeps {
 	fs?: TelegramDaemonFs;
 	now?: () => number;
 	pidAlive?: (pid: number) => boolean;
-	sendSignal?: (pid: number, signal: NodeJS.Signals) => void;
+	processReference?: (pid: number) => DaemonProcessReference | undefined;
+	pidIncarnation?: (pid: number) => string | undefined;
+	/** Test seam for platform-specific default stable-process authority. */
+	platform?: NodeJS.Platform;
 	spawn?: TelegramDaemonDeps["spawn"];
 	execPath?: string;
 	/**
@@ -110,8 +155,14 @@ export interface TelegramDaemonControlDeps {
 	randomId?: () => string;
 	sleep?: (ms: number) => Promise<void>;
 	waitStepMs?: number;
+	/** Bounded startup-readiness timeout; injectable for deterministic controller tests. */
+	readinessTimeoutMs?: number;
 }
 
+export type TelegramGenerationReloadResult =
+	| { outcome: "ready"; operation: DaemonOperationResult }
+	| { outcome: "failed"; operation: DaemonOperationResult };
+
 function defaultPidAlive(pid: number): boolean {
 	try {
 		process.kill(pid, 0);
@@ -121,20 +172,12 @@ function defaultPidAlive(pid: number): boolean {
 	}
 }
 
-function defaultSendSignal(pid: number, signal: NodeJS.Signals): void {
-	try {
-		process.kill(pid, signal);
-	} catch {
-		// Best-effort: the process may already be gone.
-	}
-}
-
 export class TelegramDaemonController implements BuiltInDaemonController {
 	readonly kind = "telegram" as const;
 	private readonly fsImpl: TelegramDaemonFs;
-	private readonly now: () => number;
 	private readonly pidAlive: (pid: number) => boolean;
-	private readonly sendSignal: (pid: number, signal: NodeJS.Signals) => void;
+	private readonly now: () => number;
+	private readonly processReference: (pid: number) => DaemonProcessReference | undefined;
 	private readonly waitStepMs: number;
 
 	constructor(
@@ -144,7 +187,7 @@ export class TelegramDaemonController implements BuiltInDaemonController {
 		this.fsImpl = deps.fs ?? nodeFs;
 		this.now = deps.now ?? Date.now;
 		this.pidAlive = deps.pidAlive ?? defaultPidAlive;
-		this.sendSignal = deps.sendSignal ?? defaultSendSignal;
+		this.processReference = deps.processReference ?? (pid => defaultProcessReference(pid, deps.platform));
 		this.waitStepMs = deps.waitStepMs ?? DEFAULT_WAIT_STEP_MS;
 	}
 
@@ -167,18 +210,19 @@ export class TelegramDaemonController implements BuiltInDaemonController {
 		}
 		const state = await readDaemonState(this.settings, this.fsImpl);
 		const roots = await readDaemonRoots(this.settings, this.fsImpl);
-		const live =
-			state !== undefined &&
-			isFreshLiveOwner({
-				state,
-				now: this.now(),
-				tokenFingerprint: tokenFingerprint(cfg.botToken as string),
-				chatId: cfg.chatId as string,
-				pidAlive: this.pidAlive,
-			});
-		let health: DaemonHealth = "stopped";
-		if (live) health = "running";
-		else if (state && state.stoppedAt === undefined) health = "stale";
+		const health: DaemonHealth =
+			!state || state.stoppedAt !== undefined || !this.pidAlive(state.pid)
+				? "stopped"
+				: isCurrentCompatibleOwner({
+							state,
+							now: this.now(),
+							tokenFingerprint: tokenFingerprint(cfg.botToken as string),
+							chatId: cfg.chatId as string,
+							pidAlive: this.pidAlive,
+							pidIncarnation: this.deps.pidIncarnation,
+						})
+					? "running"
+					: "stale";
 		return {
 			kind: this.kind,
 			configured: true,
@@ -198,6 +242,7 @@ export class TelegramDaemonController implements BuiltInDaemonController {
 			fs: this.deps.fs,
 			now: this.deps.now,
 			pidAlive: this.deps.pidAlive,
+			pidIncarnation: this.deps.pidIncarnation,
 			pid: this.deps.ownerPid ?? process.ppid,
 			spawn: this.deps.spawn,
 			execPath: this.deps.execPath,
@@ -205,6 +250,32 @@ export class TelegramDaemonController implements BuiltInDaemonController {
 		};
 	}
 
+	private async spawnAndWait(
+		roots: string[],
+		token: string,
+		chatId: string,
+	): Promise<{ spawned: TelegramSpawnOwnerResult; ready: boolean }> {
+		const spawned = await spawnTelegramDaemonOwner(
+			{ settings: this.settings, roots, tokenFingerprint: token, chatId },
+			this.spawnDeps(),
+		);
+		const ready = await confirmTelegramDaemonSpawn({
+			settings: this.settings,
+			spawned,
+			tokenFingerprint: token,
+			chatId,
+			pid: this.deps.ownerPid ?? process.ppid,
+			fs: this.fsImpl,
+			now: this.now,
+			pidAlive: this.pidAlive,
+			pidIncarnation: this.deps.pidIncarnation,
+			sleep: this.deps.sleep,
+			waitStepMs: this.waitStepMs,
+			timeoutMs: this.deps.readinessTimeoutMs,
+		});
+		return { spawned, ready };
+	}
+
 	private sleep(ms: number): Promise<void> {
 		if (this.deps.sleep) return this.deps.sleep(ms);
 		return new Promise(resolve => setTimeout(resolve, ms));
@@ -217,14 +288,67 @@ export class TelegramDaemonController implements BuiltInDaemonController {
 	 */
 	private async waitForPidDeath(pid: number, timeoutMs: number): Promise<boolean> {
 		if (!this.pidAlive(pid)) return true;
-		const deadline = this.now() + timeoutMs;
-		while (this.now() < deadline) {
-			await this.sleep(this.waitStepMs);
+		const timeout = Math.max(timeoutMs, 0);
+		const waitStepMs = Math.max(this.waitStepMs, 1);
+		const deadline = this.now() + timeout;
+		const maxPolls = Math.ceil(timeout / waitStepMs);
+		for (let poll = 0; poll < maxPolls && this.now() < deadline; poll++) {
+			await this.sleep(waitStepMs);
 			if (!this.pidAlive(pid)) return true;
 		}
 		return !this.pidAlive(pid);
 	}
 
+	private async signalCapturedOwner(
+		captured: DaemonState,
+		tokenFingerprint: string,
+		chatId: string,
+		signal: NodeJS.Signals,
+	): Promise<"signaled" | "already_gone" | "ownership_changed"> {
+		const current = await readDaemonState(this.settings, this.fsImpl);
+		// The signal is a privileged action: immediately before sending it, prove the
+		// complete state record is still the exact captured owner. A malformed record,
+		// configuration mutation, or successor using a reused PID is never signalable.
+		if (
+			!hasSafeDaemonStateShape(current) ||
+			current.ownerId !== captured.ownerId ||
+			current.acquisitionId !== captured.acquisitionId ||
+			current.pid !== captured.pid ||
+			current.generation !== captured.generation ||
+			current.incarnation !== captured.incarnation ||
+			current.tokenFingerprint !== captured.tokenFingerprint ||
+			current.chatId !== captured.chatId ||
+			current.tokenFingerprint !== tokenFingerprint ||
+			current.chatId !== chatId
+		)
+			return "ownership_changed";
+		// A matching captured owner that exited between the request and recheck has
+		// completed the handoff. A live owner with changed or unavailable provenance
+		// remains ambiguous and must not be treated as stopped.
+		if (!this.pidAlive(captured.pid)) return "already_gone";
+		if (
+			!isSignalableMatchingOwner({
+				state: current,
+				tokenFingerprint,
+				chatId,
+				pidAlive: this.pidAlive,
+				pidIncarnation: this.deps.pidIncarnation,
+			})
+		)
+			return "ownership_changed";
+		const processRef = this.processReference(captured.pid);
+		// A stable native process reference (pidfd/handle/start-time reference) closes
+		// the exit-and-reuse window between ordinary provenance checks and signaling.
+		// Its identity must still be the exact persisted incarnation before use.
+		if (!processRef || processRef.incarnation !== captured.incarnation) return "ownership_changed";
+		try {
+			processRef.signalRoot(signal);
+		} catch {
+			return "already_gone";
+		}
+		return "signaled";
+	}
+
 	private result(
 		action: "stop" | "reload",
 		ok: boolean,
@@ -241,6 +365,13 @@ export class TelegramDaemonController implements BuiltInDaemonController {
 		return this.stopOrReload("reload", opts);
 	}
 
+	async reloadForGenerationUpgrade(opts: DaemonOperationOptions = {}): Promise<TelegramGenerationReloadResult> {
+		const operation = await this.stopOrReload("reload", opts);
+		if (!operation.ok) return { outcome: "failed", operation };
+		const after = await this.status();
+		return after.health === "running" ? { outcome: "ready", operation } : { outcome: "failed", operation };
+	}
+
 	async stop(opts: DaemonOperationOptions = {}): Promise<DaemonOperationResult> {
 		return this.stopOrReload("stop", opts);
 	}
@@ -259,19 +390,55 @@ export class TelegramDaemonController implements BuiltInDaemonController {
 		const gracefulTimeoutMs = opts.gracefulTimeoutMs ?? DEFAULT_GRACEFUL_TIMEOUT_MS;
 		const killTimeoutMs = opts.killTimeoutMs ?? DEFAULT_KILL_TIMEOUT_MS;
 
-		// No running owner.
-		if (before.health !== "running") {
-			if (action === "stop") {
+		const state = await readDaemonState(this.settings, this.fsImpl);
+		const replaceableLiveOwner =
+			(action === "reload" &&
+				state !== undefined &&
+				isFreshLiveOwner({
+					state,
+					now: this.now(),
+					tokenFingerprint: fp,
+					chatId,
+					pidAlive: this.pidAlive,
+					pidIncarnation: this.deps.pidIncarnation,
+				}) &&
+				isSignalableMatchingOwner({
+					state,
+					tokenFingerprint: fp,
+					chatId,
+					pidAlive: this.pidAlive,
+					pidIncarnation: this.deps.pidIncarnation,
+				})) ||
+			// A physically-live matching owner whose heartbeat is stale (hung) may be
+			// past-TTL yet still holding the poller. Autostart/generation-upgrade reloads
+			// stay conservative and refuse it, but an explicit `reload --force` must be
+			// able to signal and replace it rather than deadlock behind the live PID.
+			(opts.force === true &&
+				isSignalableMatchingOwner({
+					state,
+					tokenFingerprint: fp,
+					chatId,
+					pidAlive: this.pidAlive,
+					pidIncarnation: this.deps.pidIncarnation,
+				}));
+		const stoppableLiveOwner =
+			action === "stop" &&
+			isSignalableMatchingOwner({
+				state,
+				tokenFingerprint: fp,
+				chatId,
+				pidAlive: this.pidAlive,
+				pidIncarnation: this.deps.pidIncarnation,
+			});
+		// A stale pre-generation owner may only be moved by reload; manual stop
+		// also targets a physically live matching legacy owner, but never spawns.
+		if (before.health !== "running" && !replaceableLiveOwner && !stoppableLiveOwner) {
+			if (action === "stop")
 				return this.result(action, true, "no running telegram daemon", before, before, warnings);
-			}
-			const spawnIfStopped = opts.spawnIfStopped ?? true;
-			if (!spawnIfStopped) {
+			if (!(opts.spawnIfStopped ?? true)) {
 				return this.result(action, true, "no running telegram daemon to reload", before, before, warnings);
 			}
-			const spawned = await spawnTelegramDaemonOwner(
-				{ settings: this.settings, roots, tokenFingerprint: fp, chatId },
-				this.spawnDeps(),
-			);
+			const { spawned, ready } = await this.spawnAndWait(roots, fp, chatId);
 			warnings.push(...spawned.warnings);
 			const after = await this.status();
 			if (spawned.result === "blocked") {
@@ -287,8 +454,10 @@ export class TelegramDaemonController implements BuiltInDaemonController {
 			}
 			return this.result(
 				action,
-				spawned.result === "owner_spawned",
-				`spawned fresh telegram daemon (${spawned.result})`,
+				spawned.result === "owner_spawned" && ready && after.health === "running",
+				ready
+					? `spawned fresh telegram daemon (${spawned.result})`
+					: "telegram daemon did not become ready after spawning",
 				before,
 				after,
 				warnings,
@@ -296,15 +465,48 @@ export class TelegramDaemonController implements BuiltInDaemonController {
 		}
 
 		// Running owner: capture identity, request cooperative stop, signal, wait.
-		const oldOwnerId = before.ownerId as string;
-		const oldPid = before.pid as number;
+		if (
+			!hasSafeDaemonStateShape(state) ||
+			!isSignalableMatchingOwner({
+				state,
+				tokenFingerprint: fp,
+				chatId,
+				pidAlive: this.pidAlive,
+				pidIncarnation: this.deps.pidIncarnation,
+			})
+		) {
+			return this.result(
+				action,
+				false,
+				"telegram daemon ownership changed; refusing to signal",
+				before,
+				before,
+				warnings,
+			);
+		}
+		const capturedOwner = state;
+		const oldOwnerId = capturedOwner.ownerId;
+		const oldPid = capturedOwner.pid;
 		const requestId = this.deps.randomId?.() ?? `${this.now().toString(36)}-${Math.random().toString(36).slice(2)}`;
 		await writeTelegramControlRequest(
 			this.settings,
 			{ version: 1, requestId, action, ownerId: oldOwnerId, pid: oldPid, createdAt: this.now() },
 			this.fsImpl,
 		);
-		if (this.pidAlive(oldPid)) this.sendSignal(oldPid, "SIGTERM");
+		if ((await this.signalCapturedOwner(capturedOwner, fp, chatId, "SIGTERM")) === "ownership_changed") {
+			await this.clearOwnRequest(requestId);
+			return this.result(
+				action,
+				false,
+				"telegram daemon ownership changed; refusing to signal",
+				before,
+				await this.status(),
+				warnings,
+			);
+		}
+		// "signaled" or "already_gone" (the owner already exited cooperatively): confirm
+		// death through the normal path — waitForPidDeath returns immediately for a
+		// dead pid, so stop succeeds and reload proceeds to spawn the replacement.
 
 		let dead = await this.waitForPidDeath(oldPid, gracefulTimeoutMs);
 		if (!dead) {
@@ -319,33 +521,36 @@ export class TelegramDaemonController implements BuiltInDaemonController {
 					tokenFingerprint: fp,
 					chatId,
 					pidAlive: this.pidAlive,
+					pidIncarnation: this.deps.pidIncarnation,
 				});
 			if (changedToLiveOwner) {
-				// A different, fresh-live owner already supersedes the old one. Do not
-				// kill it or spawn another; attach to the running daemon.
-				await this.clearOwnRequest(requestId, oldOwnerId);
+				await this.clearOwnRequest(requestId);
 				const after = await this.status();
-				warnings.push("ownership changed to a live owner; attached without spawning");
-				return this.result(
-					action,
-					true,
-					action === "reload"
-						? "a live owner already exists; attached instead of reloading"
-						: "another live owner already exists",
-					before,
-					after,
-					warnings,
-				);
+				if (!dead) {
+					return this.result(
+						action,
+						false,
+						"ownership changed before the captured daemon exited; refusing to signal or spawn",
+						before,
+						after,
+						warnings,
+					);
+				}
 			}
 			// No live replacement. Escalate to SIGKILL only with --force and only when
 			// the captured owner/pid still matches, so we never kill a different owner.
 			const stillSameOwner = current !== undefined && current.ownerId === oldOwnerId && current.pid === oldPid;
-			if (opts.force && stillSameOwner && this.pidAlive(oldPid)) {
-				this.sendSignal(oldPid, "SIGKILL");
-				dead = await this.waitForPidDeath(oldPid, killTimeoutMs);
+			if (opts.force && stillSameOwner) {
+				const killResult = await this.signalCapturedOwner(capturedOwner, fp, chatId, "SIGKILL");
+				// Kill only a still-live matching owner; an owner that exited between the
+				// graceful timeout and this recheck ("already_gone") is confirmed dead by
+				// waitForPidDeath, while a real ownership change stays fenced.
+				if (killResult === "signaled" || killResult === "already_gone") {
+					dead = await this.waitForPidDeath(oldPid, killTimeoutMs);
+				}
 			}
 			if (!dead) {
-				await this.clearOwnRequest(requestId, oldOwnerId);
+				await this.clearOwnRequest(requestId);
 				const after = await this.status();
 				const message = opts.force
 					? "old daemon did not exit after SIGKILL; refusing to spawn to avoid a Telegram 409 conflict"
@@ -355,22 +560,38 @@ export class TelegramDaemonController implements BuiltInDaemonController {
 		}
 
 		// Old pid is dead: safe to clear our request and proceed.
-		await this.clearOwnRequest(requestId, oldOwnerId);
+		await this.clearOwnRequest(requestId);
 
 		if (action === "stop") {
 			const after = await this.status();
 			return this.result(action, true, "stopped telegram daemon", before, after, warnings);
 		}
 
-		const spawned = await spawnTelegramDaemonOwner(
-			{ settings: this.settings, roots, tokenFingerprint: fp, chatId },
-			this.spawnDeps(),
-		);
+		const { spawned, ready } = await this.spawnAndWait(roots, fp, chatId);
 		warnings.push(...spawned.warnings);
 		const after = await this.status();
 		if (spawned.result === "attached") {
-			// A live owner already exists; attaching to it is a valid running end-state.
-			warnings.push("a live owner already exists; attached instead of spawning");
+			const attachedState = await readDaemonState(this.settings, this.fsImpl);
+			if (
+				!isCurrentCompatibleOwner({
+					state: attachedState,
+					now: this.now(),
+					tokenFingerprint: fp,
+					chatId,
+					pidAlive: this.pidAlive,
+					pidIncarnation: this.deps.pidIncarnation,
+				})
+			) {
+				return this.result(
+					action,
+					false,
+					"ownership moved to a non-ready or incompatible daemon; reload required",
+					before,
+					after,
+					warnings,
+				);
+			}
+			warnings.push("a ready current-compatible owner already exists; attached instead of spawning");
 		} else if (after.ownerId && after.ownerId === oldOwnerId) {
 			warnings.push("owner id unchanged after reload");
 		}
@@ -387,20 +608,17 @@ export class TelegramDaemonController implements BuiltInDaemonController {
 		}
 		return this.result(
 			action,
-			spawned.result !== "disabled",
-			`reloaded telegram daemon (${spawned.result})`,
+			(spawned.result === "owner_spawned" || spawned.result === "attached") && ready && after.health === "running",
+			ready ? `reloaded telegram daemon (${spawned.result})` : "telegram daemon did not become ready after spawning",
 			before,
 			after,
 			warnings,
 		);
 	}
 
-	/** Clear our own control request unless a newer owner-scoped request replaced it. */
-	private async clearOwnRequest(requestId: string, oldOwnerId: string): Promise<void> {
+	/** Clear only our exact control request; a successor request must survive. */
+	private async clearOwnRequest(requestId: string): Promise<void> {
 		const current = await readTelegramControlRequest(this.settings, this.fsImpl);
-		if (!current) return;
-		if (current.requestId === requestId || current.ownerId === oldOwnerId) {
-			await clearTelegramControlRequest(this.settings, current.requestId, this.fsImpl);
-		}
+		if (current?.requestId === requestId) await clearTelegramControlRequest(this.settings, requestId, this.fsImpl);
 	}
 }