agent-transport-system 0.7.55 → 0.7.57

package/dist/ats.js

@@ -27,7 +27,7 @@ import wrapAnsi from "wrap-ansi";
 import { Box, Container, Editor, Key, ProcessTerminal, TUI, Text, getEditorKeybindings, matchesKey } from "@mariozechner/pi-tui";
 
 //#region package.json
-var version = "0.7.55";
+var version = "0.7.57";
 var package_default = {
 	$schema: "https://www.schemastore.org/package.json",
 	name: "agent-transport-system",
@@ -15223,6 +15223,7 @@ const PROVIDER_CONVERSATION_PROOF_SYNC_RETRY_INTERVAL_MS = 6e4;
 const RUNTIME_STATE_PERSIST_MIN_INTERVAL_MS = 6e4;
 const DEFAULT_RECONNECT_DELAY_MS = 1e3;
 const MAX_RECONNECT_DELAY_MS = 1e4;
+const DAEMON_STREAM_CONNECT_TIMEOUT_MS = 15e3;
 const REGISTER_RESPONSE_TIMEOUT_MS = 1e4;
 const HEARTBEAT_RESPONSE_TIMEOUT_MS = 15e3;
 const HEARTBEAT_CONSECUTIVE_FAILURES_BEFORE_CLOSE = 3;
@@ -19738,8 +19739,8 @@ function isPidAlive$4(pid) {
 	}
 }
 function isRuntimeLeaseHeartbeatStale(runtimeLease) {
-	if (!runtimeLease.heartbeatAt) return false;
-	const heartbeatMs = Date.parse(runtimeLease.heartbeatAt);
+	const freshnessTimestamp = runtimeLease.heartbeatAt ?? runtimeLease.updatedAt;
+	const heartbeatMs = Date.parse(freshnessTimestamp);
 	if (Number.isNaN(heartbeatMs)) return true;
 	return Date.now() - heartbeatMs > DAEMON_HEARTBEAT_STALE_MS;
 }
@@ -39544,12 +39545,29 @@ function buildDaemonAgentOverviewPayload(command) {
 //#region src/transport/ws.ts
 async function connectWebSocket(url, input) {
 	return await new Promise((resolve, reject) => {
+		let settled = false;
 		const ws = new WebSocket(url, { headers: input?.headers });
+		const openTimeoutMs = normalizeOpenTimeoutMs(input?.openTimeoutMs);
+		const openTimeout = openTimeoutMs === null ? null : setTimeout(() => {
+			if (settled) return;
+			settled = true;
+			ws.terminate();
+			reject(/* @__PURE__ */ new Error(`websocket open timeout after ${openTimeoutMs}ms`));
+		}, openTimeoutMs);
 		const closePromise = new Promise((r) => {
 			ws.on("close", () => r());
 			ws.on("error", () => r());
 		});
+		ws.on("close", (code, reason) => {
+			if (settled) return;
+			settled = true;
+			if (openTimeout) clearTimeout(openTimeout);
+			reject(/* @__PURE__ */ new Error(`websocket closed before open (code=${String(code)}, reason=${reason.toString("utf8")})`));
+		});
 		ws.on("open", () => {
+			if (settled) return;
+			settled = true;
+			if (openTimeout) clearTimeout(openTimeout);
 			resolve({
 				send: (data) => ws.send(data),
 				close: (code, reason) => ws.close(code, reason),
@@ -39569,10 +39587,19 @@ async function connectWebSocket(url, input) {
 			});
 		});
 		ws.on("error", (err) => {
+			if (settled) return;
+			settled = true;
+			if (openTimeout) clearTimeout(openTimeout);
 			reject(err);
 		});
 	});
 }
+function normalizeOpenTimeoutMs(value) {
+	if (value === void 0 || value === null) return null;
+	if (typeof value !== "number" || !Number.isFinite(value)) return null;
+	const normalized = Math.trunc(value);
+	return normalized > 0 ? normalized : null;
+}
 function toUtf8(value) {
 	if (typeof value === "string") return value;
 	if (value instanceof Buffer) return value.toString("utf8");
@@ -40720,13 +40747,15 @@ function createDaemonRuntimeStateTracker(input) {
 	let lastRuntimeStatePersistAtMs = Date.now();
 	const touchRuntimeState = async (updates, options) => {
 		const nowAt = nowIsoString();
+		const previousHeartbeatAt = runtimeState.heartbeatAt;
 		runtimeState = {
 			...runtimeState,
 			...updates,
 			updatedAt: updates.updatedAt ?? nowAt
 		};
+		const receivedFirstHeartbeat = previousHeartbeatAt === null && typeof updates.heartbeatAt === "string" && updates.heartbeatAt.length > 0;
 		if (!shouldPersistDaemonRuntimeState({
-			forcePersist: options?.forcePersist === true,
+			forcePersist: options?.forcePersist === true || receivedFirstHeartbeat,
 			nowMs: Date.now(),
 			lastPersistedAtMs: lastRuntimeStatePersistAtMs,
 			updates
@@ -46065,7 +46094,7 @@ async function runDaemonSocketSession(input) {
 				text: `heartbeat failed (${String(consecutiveHeartbeatFailures)}/${String(HEARTBEAT_CONSECUTIVE_FAILURES_BEFORE_CLOSE)}): ${toErrorMessage$29(error)}`
 			});
 			if (shouldCloseSocket) {
-				input.socket.close(1011, "heartbeat_failed");
+				input.socket.terminate();
 				return;
 			}
 			updateHeartbeatInterval({
@@ -46926,10 +46955,13 @@ const runDaemonServiceLoop = async (input) => {
 		}
 		while (!stopRequested) try {
 			const wsUrl = toDaemonStreamUrl(baseUrl);
-			const ws = await connectWebSocket(wsUrl, { headers: selectedAtsProfile ? await buildAtsProfileRequestHeadersWithAuth({
-				atsProfile: selectedAtsProfile,
-				requestUrl: wsUrl
-			}) : await buildRequestAuthHeaders({ requestUrl: wsUrl }) });
+			const ws = await connectWebSocket(wsUrl, {
+				headers: selectedAtsProfile ? await buildAtsProfileRequestHeadersWithAuth({
+					atsProfile: selectedAtsProfile,
+					requestUrl: wsUrl
+				}) : await buildRequestAuthHeaders({ requestUrl: wsUrl }),
+				openTimeoutMs: DAEMON_STREAM_CONNECT_TIMEOUT_MS
+			});
 			activeSocket = ws;
 			emitRunLine({
 				presenter: input.presenter,