@integrity-labs/agt-cli 0.25.1 → 0.25.2

package/dist/{claude-pair-runtime-ZWE5572F.js → claude-pair-runtime-Q5CZLM2R.js}

@@ -100,7 +100,7 @@ async function spawnPairSession(session) {
     return { ok: true };
   } catch {
   }
-  const { resolveClaudeBinary } = await import("./persistent-session-3FQVLVJM.js");
+  const { resolveClaudeBinary } = await import("./persistent-session-MUJA45LT.js");
   const claudeBin = resolveClaudeBinary();
   const pairEnv = {
     ...process.env,
@@ -373,4 +373,4 @@ export {
   startClaudePair,
   submitClaudePairCode
 };
-//# sourceMappingURL=claude-pair-runtime-ZWE5572F.js.map
+//# sourceMappingURL=claude-pair-runtime-Q5CZLM2R.js.map

package/dist/lib/manager-worker.js

@@ -13,7 +13,7 @@ import {
   provisionOrientHook,
   provisionStopHook,
   requireHost
-} from "../chunk-AAK2IU2Z.js";
+} from "../chunk-U4DAVDAZ.js";
 import {
   findTaskByTemplate,
   getProjectDir as getProjectDir2,
@@ -50,7 +50,7 @@ import {
   stopAllSessionsAndWait,
   stopPersistentSession,
   takeZombieDetection
-} from "../chunk-4FSQVXWB.js";
+} from "../chunk-M2RBM3M5.js";
 import {
   KANBAN_CHECK_COMMAND,
   appendDmFooter,
@@ -68,10 +68,12 @@ import {
   parseDeliveryTarget,
   parseTranscriptUsage,
   parseUsageBanner,
+  probeHttpProvider,
   resolveChannels,
+  resolveConnectivityProbe,
   resolveDmTarget,
   wrapScheduledTaskPrompt
-} from "../chunk-4K3ERUGE.js";
+} from "../chunk-VWOXS4YP.js";
 
 // src/lib/manager-worker.ts
 import { createHash as createHash3 } from "crypto";
@@ -691,6 +693,223 @@ function shouldRespawnForModelChange(input) {
   return previousModel !== primaryModel;
 }
 
+// src/lib/connectivity-probe-executor.ts
+async function executeConnectivityProbe(target, deps = {}) {
+  const descriptor = resolveConnectivityProbe({
+    definitionId: target.definitionId,
+    sourceType: target.sourceType,
+    authType: target.authType
+  });
+  if (!descriptor.readOnly) {
+    throw new Error(`Refusing non-read-only probe for ${target.definitionId}`);
+  }
+  switch (descriptor.kind) {
+    case "http_provider":
+      return probeHttpProvider(target.definitionId, target.credentials, deps.fetchImpl ?? fetch);
+    case "composio_account":
+      if (!deps.composioProbe) return null;
+      return deps.composioProbe(target.definitionId, target.credentials);
+    case "mcp_tools_list":
+      if (!deps.mcpProbe) return null;
+      return deps.mcpProbe({
+        serverKey: target.mcpServerKey ?? target.definitionId,
+        definitionId: target.definitionId
+      });
+    case "cli_command":
+      if (!deps.runCli) return null;
+      return deps.runCli(target.cliBinary ?? target.definitionId, descriptor.cliArgs ?? ["--version"]);
+    case "builtin":
+      return { status: "ok", message: `${target.definitionId}: built-in` };
+    case "unsupported":
+    default:
+      return null;
+  }
+}
+
+// src/lib/connectivity-probe-runner.ts
+var DEFAULT_INTERVAL_MS = 60 * 60 * 1e3;
+var DEFAULT_MAX_PER_RUN = 25;
+function isDue(lastCheck, now, intervalMs) {
+  if (!lastCheck) return true;
+  const t = Date.parse(lastCheck);
+  if (Number.isNaN(t)) return true;
+  return now - t >= intervalMs;
+}
+async function runConnectivityProbes(integrations, options = {}) {
+  const now = (options.now?.() ?? /* @__PURE__ */ new Date()).getTime();
+  const intervalMs = options.intervalMs ?? DEFAULT_INTERVAL_MS;
+  const maxPerRun = options.maxPerRun ?? DEFAULT_MAX_PER_RUN;
+  const probeDeps = options.probeDeps ?? {};
+  const execute = options.executeProbe ?? executeConnectivityProbe;
+  const due = integrations.filter((i) => isDue(i.last_connectivity_check_at, now, intervalMs)).sort((a, b) => {
+    const ta = a.last_connectivity_check_at ? Date.parse(a.last_connectivity_check_at) : 0;
+    const tb = b.last_connectivity_check_at ? Date.parse(b.last_connectivity_check_at) : 0;
+    return ta - tb;
+  });
+  const batch = due.slice(0, maxPerRun);
+  const reports = [];
+  let skipped = 0;
+  for (const integ of batch) {
+    const target = {
+      definitionId: integ.definition_id,
+      sourceType: integ.source_type ?? null,
+      authType: integ.auth_type ?? null,
+      credentials: integ.credentials ?? {},
+      cliBinary: integ.cli_binary,
+      mcpServerKey: integ.mcp_server_key
+    };
+    let outcome;
+    try {
+      outcome = await execute(target, probeDeps);
+    } catch (err) {
+      console.warn(`[connectivity-probe] ${integ.definition_id} threw: ${err.message}`);
+      outcome = null;
+    }
+    if (!outcome) {
+      skipped += 1;
+      continue;
+    }
+    reports.push({
+      integration_id: integ.id,
+      scope: integ.scope,
+      status: outcome.status,
+      ...outcome.message ? { message: outcome.message } : {}
+    });
+  }
+  return { reports, due: due.length, probed: batch.length, skipped };
+}
+
+// src/lib/mcp-probe-client.ts
+var MCP_ACCEPT = "application/json, text/event-stream";
+var DEFAULT_TIMEOUT_MS = 1e4;
+async function parseRpc(res, expectedId) {
+  const ct = res.headers.get("content-type") ?? "";
+  if (ct.includes("text/event-stream")) {
+    const text = await res.text();
+    let dataLines = [];
+    for (const rawLine of text.split(/\r?\n/)) {
+      if (rawLine.startsWith("data:")) {
+        dataLines.push(rawLine.slice(5).trimStart());
+        continue;
+      }
+      if (rawLine === "" && dataLines.length > 0) {
+        try {
+          const msg2 = JSON.parse(dataLines.join("\n"));
+          if (("result" in msg2 || "error" in msg2) && msg2["id"] === expectedId) return msg2;
+        } catch {
+        }
+        dataLines = [];
+      }
+    }
+    return null;
+  }
+  const msg = await res.json().catch(() => null);
+  if (msg && ("result" in msg || "error" in msg) && msg["id"] === expectedId) return msg;
+  return msg;
+}
+function httpStatusOutcome(status, step) {
+  if (status === 401 || status === 403) {
+    return { status: "down", message: `MCP ${step} unauthorized (${status}) \u2014 reconnect required` };
+  }
+  if (status >= 500) {
+    return { status: "transient_error", message: `MCP ${step} returned ${status}` };
+  }
+  return { status: "down", message: `MCP ${step} returned ${status}` };
+}
+async function probeMcpHttp(config2, fetchImpl = fetch) {
+  const timeoutMs = config2.timeoutMs ?? DEFAULT_TIMEOUT_MS;
+  const baseHeaders = {
+    ...config2.headers ?? {},
+    "Content-Type": "application/json",
+    Accept: MCP_ACCEPT
+  };
+  try {
+    const initRes = await fetchImpl(config2.url, {
+      method: "POST",
+      headers: baseHeaders,
+      body: JSON.stringify({
+        jsonrpc: "2.0",
+        id: 1,
+        method: "initialize",
+        params: {
+          protocolVersion: "2025-03-26",
+          capabilities: {},
+          clientInfo: { name: "augmented-connectivity-probe", version: "1.0.0" }
+        }
+      }),
+      signal: AbortSignal.timeout(timeoutMs)
+    });
+    if (!initRes.ok) return httpStatusOutcome(initRes.status, "initialize");
+    const sessionId = initRes.headers.get("mcp-session-id");
+    await parseRpc(initRes, 1);
+    const sessionHeaders = { ...baseHeaders, ...sessionId ? { "Mcp-Session-Id": sessionId } : {} };
+    const initializedRes = await fetchImpl(config2.url, {
+      method: "POST",
+      headers: sessionHeaders,
+      body: JSON.stringify({ jsonrpc: "2.0", method: "notifications/initialized" }),
+      signal: AbortSignal.timeout(5e3)
+    });
+    if (!initializedRes.ok) return httpStatusOutcome(initializedRes.status, "initialized");
+    await initializedRes.text().catch(() => "");
+    const listRes = await fetchImpl(config2.url, {
+      method: "POST",
+      headers: sessionHeaders,
+      body: JSON.stringify({ jsonrpc: "2.0", id: 2, method: "tools/list" }),
+      signal: AbortSignal.timeout(timeoutMs)
+    });
+    if (!listRes.ok) return httpStatusOutcome(listRes.status, "tools/list");
+    const rpc = await parseRpc(listRes, 2);
+    if (rpc && "error" in rpc) {
+      const err = rpc["error"];
+      return { status: "down", message: `MCP tools/list error: ${err?.message ?? "unknown"}` };
+    }
+    const result = rpc?.["result"];
+    const toolCount = Array.isArray(result?.tools) ? result.tools.length : void 0;
+    return {
+      status: "ok",
+      message: toolCount !== void 0 ? `${toolCount} tools` : "reachable",
+      ...toolCount !== void 0 ? { details: { toolCount } } : {}
+    };
+  } catch (err) {
+    const isAbort = err?.name === "TimeoutError" || err?.name === "AbortError";
+    return {
+      status: "transient_error",
+      message: isAbort ? `MCP handshake timed out after ${timeoutMs / 1e3}s` : `MCP handshake failed: ${err.message}`
+    };
+  }
+}
+
+// src/lib/cli-probe.ts
+import { execFile } from "child_process";
+var DEFAULT_TIMEOUT_MS2 = 8e3;
+function runCliProbe(binary, args, opts = {}) {
+  const timeoutMs = opts.timeoutMs ?? DEFAULT_TIMEOUT_MS2;
+  return new Promise((resolve) => {
+    execFile(
+      binary,
+      args,
+      { timeout: timeoutMs, env: opts.env ?? process.env, windowsHide: true },
+      (err, stdout) => {
+        if (!err) {
+          const firstLine = String(stdout).split(/\r?\n/, 1)[0]?.trim();
+          resolve({ status: "ok", message: firstLine ? `${binary}: ${firstLine}` : `${binary}: ok` });
+          return;
+        }
+        const e = err;
+        if (e.code === "ENOENT") {
+          resolve({ status: "down", message: `${binary} not found on PATH (not installed)` });
+          return;
+        }
+        if (e.killed || e.signal === "SIGTERM") {
+          resolve({ status: "transient_error", message: `${binary} probe timed out after ${timeoutMs / 1e3}s` });
+          return;
+        }
+        resolve({ status: "down", message: `${binary} exited non-zero: ${e.message}` });
+      }
+    );
+  });
+}
+
 // src/lib/usage-banner-monitor.ts
 var MIN_CHECK_INTERVAL_MS = 6e4;
 var PANE_TAIL_LINES_FOR_BANNER = 200;
@@ -1377,9 +1596,9 @@ var GatewayClientPool = class extends EventEmitter {
 import { readFile, readdir } from "fs/promises";
 import { homedir as homedir2, platform } from "os";
 import { join as join3 } from "path";
-import { execFile } from "child_process";
+import { execFile as execFile2 } from "child_process";
 import { promisify } from "util";
-var execFileAsync = promisify(execFile);
+var execFileAsync = promisify(execFile2);
 var EXPIRING_SOON_MS = 48 * 60 * 60 * 1e3;
 async function detectClaudeAuth() {
   if (process.env["ANTHROPIC_API_KEY"] || process.env["ANTHROPIC_AUTH_TOKEN"]) {
@@ -2712,6 +2931,55 @@ function projectMcpKeys(_codeName, projectDir) {
     return null;
   }
 }
+function readMcpHttpServerConfig(projectDir, serverKey) {
+  try {
+    const raw = readFileSync5(join6(projectDir, ".mcp.json"), "utf-8");
+    const servers = JSON.parse(raw).mcpServers ?? {};
+    const entry = servers[serverKey];
+    if (entry && typeof entry.url === "string" && (entry.type === "http" || entry.type === void 0)) {
+      return { url: entry.url, ...entry.headers ? { headers: entry.headers } : {} };
+    }
+    return null;
+  } catch {
+    return null;
+  }
+}
+async function runAgentConnectivityProbes(agent, integrations, projectDir) {
+  if (integrations.length === 0) return;
+  const probeDeps = {
+    fetchImpl: fetch,
+    runCli: (binary, args) => runCliProbe(binary, args),
+    mcpProbe: async (target) => {
+      const cfg = readMcpHttpServerConfig(projectDir, target.serverKey);
+      if (!cfg) {
+        return { status: "transient_error", message: `MCP server '${target.serverKey}' not resolvable from .mcp.json` };
+      }
+      return probeMcpHttp(cfg);
+    }
+  };
+  const intervalSec = Number(process.env.AGT_CONNECTIVITY_PROBE_INTERVAL_SECONDS) || 3600;
+  const result = await runConnectivityProbes(
+    integrations.map((i) => ({
+      id: i.id,
+      definition_id: i.definition_id,
+      scope: i.scope,
+      auth_type: i.auth_type,
+      source_type: i.source_type ?? null,
+      credentials: i.credentials,
+      last_connectivity_check_at: i.last_connectivity_check_at ?? null
+    })),
+    { probeDeps, intervalMs: intervalSec * 1e3 }
+  );
+  if (result.reports.length > 0) {
+    await api.post("/host/integration-connectivity", {
+      agent_id: agent.agent_id,
+      results: result.reports
+    });
+    log(
+      `Connectivity probe for '${agent.code_name}': probed=${result.probed} reported=${result.reports.length} due=${result.due}`
+    );
+  }
+}
 function stopPersistentSessionAndForgetMcpBaseline(codeName, breakerReason) {
   cancelPendingSessionRestart(codeName);
   stopPersistentSession(codeName, log);
@@ -2862,7 +3130,7 @@ var cachedFrameworkVersion = null;
 var lastVersionCheckAt = 0;
 var VERSION_CHECK_INTERVAL_MS = 5 * 60 * 1e3;
 var lastResponsivenessProbeAt = 0;
-var agtCliVersion = true ? "0.25.1" : "dev";
+var agtCliVersion = true ? "0.25.2" : "dev";
 function resolveBrewPath(execFileSync4) {
   try {
     const out = execFileSync4("which", ["brew"], { timeout: 5e3 }).toString().trim();
@@ -3821,7 +4089,7 @@ async function pollCycle() {
     }
     try {
       const { detectHostSecurity } = await import("../host-security-6PDFG7F5.js");
-      const { collectDiagnostics } = await import("../persistent-session-3FQVLVJM.js");
+      const { collectDiagnostics } = await import("../persistent-session-MUJA45LT.js");
       const diagCodeNames = [...persistentSessionAgents];
       const agentDiagnostics = diagCodeNames.length > 0 ? collectDiagnostics(diagCodeNames) : void 0;
       let tailscaleHostname;
@@ -3879,7 +4147,7 @@ async function pollCycle() {
       const {
         collectResponsivenessProbes,
         getResponsivenessIntervalMs
-      } = await import("../responsiveness-probe-M7PUFD5V.js");
+      } = await import("../responsiveness-probe-2TBRHC5T.js");
       const probeIntervalMs = getResponsivenessIntervalMs();
       if (now - lastResponsivenessProbeAt > probeIntervalMs) {
         const probeCodeNames = [...persistentSessionAgents];
@@ -4789,6 +5057,14 @@ async function processAgent(agent, agentStates) {
         log(`OAuth token refresh failed for '${agent.code_name}/${integration.definition_id}': ${err.message}`);
       }
     }
+    if (process.env.AGT_CONNECTIVITY_PROBE_ENABLED === "true") {
+      try {
+        const probeProjectDir = join6(homedir4(), ".augmented", agent.code_name, "project");
+        await runAgentConnectivityProbes(agent, integrations, probeProjectDir);
+      } catch (err) {
+        log(`Connectivity probe failed for '${agent.code_name}': ${err.message}`);
+      }
+    }
     if (integrations.length > 0) {
       const intHash = computeIntegrationsHash(integrations);
       const prevIntHash = knownIntegrationHashes.get(agent.agent_id);
@@ -7875,7 +8151,7 @@ async function processClaudePairSessions(agents) {
     killPairSession,
     pairTmuxSession,
     finalizeClaudePairOnboarding
-  } = await import("../claude-pair-runtime-ZWE5572F.js");
+  } = await import("../claude-pair-runtime-Q5CZLM2R.js");
   for (const pairId of pendingResp.cancelled_pair_ids ?? []) {
     log(`[claude-pair] sweeping orphan tmux session for pair ${pairId.slice(0, 8)}`);
     const killed = await killPairSession(pairTmuxSession(pairId));