@haven_ai/connect 0.1.23-alpha.0 → 0.1.23-alpha.2

package/dist/index.cjs

@@ -1,5 +1,6 @@
 'use strict';
 
+var sdk = require('@haven_ai/sdk');
 var crypto = require('crypto');
 var ethers = require('ethers');
 var promises = require('fs/promises');
@@ -9,8 +10,8 @@ var child_process = require('child_process');
 var util = require('util');
 var yaml = require('yaml');
 var mcp = require('@haven_ai/mcp');
-var sdk = require('@haven_ai/sdk');
 var signer = require('@haven_ai/signer');
+var fs = require('fs');
 var url = require('url');
 
 var _documentCurrentScript = typeof document !== 'undefined' ? document.currentScript : null;
@@ -48,6 +49,10 @@ function createConnectApiClient(baseUrl, fetchImpl = fetch) {
         }
       })
     }),
+    getConnectorStatus: (setupId, apiKey) => request(fetchImpl, `${root}/agent-connection-setups/${encodeURIComponent(setupId)}/connector-status`, {
+      method: "GET",
+      headers: { Authorization: `Bearer ${apiKey}` }
+    }),
     updateInstallStatus: async (setupId, apiKey, input) => {
       await request(fetchImpl, `${root}/agent-connection-setups/${encodeURIComponent(setupId)}/install-status`, {
         method: "POST",
@@ -74,6 +79,14 @@ function createConnectApiClient(baseUrl, fetchImpl = fetch) {
     }
   };
 }
+var ConnectRequestError = class extends Error {
+  constructor(message, status) {
+    super(message);
+    this.status = status;
+    this.name = "ConnectRequestError";
+  }
+  status;
+};
 async function request(fetchImpl, url, init) {
   const response = await fetchImpl(url, {
     ...init,
@@ -86,7 +99,7 @@ async function request(fetchImpl, url, init) {
   const body = text ? JSON.parse(text) : null;
   if (!response.ok) {
     const message = body?.error ?? body?.message ?? `${response.status} ${response.statusText}`;
-    throw new Error(`Haven setup request failed: ${message}`);
+    throw new ConnectRequestError(`Haven setup request failed: ${message}`, response.status);
   }
   return body;
 }
@@ -239,9 +252,9 @@ var MCP_RUNTIME_MANIFEST = {
   mcpPackage: "@haven_ai/mcp",
   mcpVersion: mcp.MCP_VERSION,
   sdkPackage: "@haven_ai/sdk",
-  sdkVersion: "0.1.23-alpha.0",
+  sdkVersion: "0.1.23-alpha.2",
   signerPackage: "@haven_ai/signer",
-  signerVersion: "0.1.23-alpha.0",
+  signerVersion: "0.1.23-alpha.2",
   // Sourced from the SDK, never a literal (#1161). This field read '20.0.0'
   // while every package's `engines` said `>=24` and the docs said `>=24.0.0`,
   // so the guard that was supposed to enforce the floor waved Node v23 through
@@ -1940,7 +1953,7 @@ function localRuntimePrepareErrorCode(err) {
 }
 
 // src/runtime.ts
-var CONNECTOR_VERSION = "0.1.23-alpha.0";
+var CONNECTOR_VERSION = "0.1.23-alpha.2";
 var CONNECT_OUTCOME_SCHEMA_VERSION = 1;
 async function runConnect(options, deps = {}) {
   assertSupportedNodeVersion(deps.nodeVersion, MCP_RUNTIME_MANIFEST.minimumNodeVersion);
@@ -2054,7 +2067,6 @@ async function runConnect(options, deps = {}) {
   } else {
     log("Haven setup on this machine is complete.");
   }
-  printNextSteps(runtimeInstall, log);
   try {
     await api.updateInstallStatus(registration.setup_id, localApiKey, {
       runtime: runtimeInstall.runtime,
@@ -2077,6 +2089,10 @@ async function runConnect(options, deps = {}) {
   } catch (err) {
     log(`Could not report install status to Haven: ${err instanceof Error ? err.message : String(err)}`);
   }
+  if (options.waitForApproval !== false && !runtimeInstall.errorCode) {
+    await waitForBudgetApproval(api, registration.setup_id, localApiKey, log, options.approvalWait);
+  }
+  printNextSteps(runtimeInstall, log);
   return {
     setupId: registration.setup_id,
     agentId: registration.agent_id,
@@ -2192,6 +2208,61 @@ function printRuntimeInstall(result, log) {
     log("Local Haven signer still needs runtime setup.");
   }
 }
+function formatAtomicAmount(atomic, decimals) {
+  const s = atomic.toString().padStart(decimals + 1, "0");
+  const intPart = s.slice(0, s.length - decimals) || "0";
+  const fracPart = s.slice(s.length - decimals).replace(/0+$/, "");
+  return fracPart ? `${intPart}.${fracPart}` : intPart;
+}
+function describeResetPeriod(resetPeriodMin) {
+  if (resetPeriodMin === 1440) return "per day";
+  if (resetPeriodMin === 60) return "per hour";
+  if (resetPeriodMin === 0) return "with no automatic reset";
+  return `per ${resetPeriodMin} minutes`;
+}
+function describeApprovedBudget(budget) {
+  const token = sdk.resolveTokenFromAddress(budget.token_address);
+  const amount = token ? `${formatAtomicAmount(BigInt(budget.amount), token.decimals)} ${budget.token_symbol}` : `${budget.amount} ${budget.token_symbol} (atomic units)`;
+  return `${amount} ${describeResetPeriod(budget.reset_period_min)}`;
+}
+async function waitForBudgetApproval(api, setupId, apiKey, log, options = {}) {
+  const intervalMs = options.intervalMs ?? 5e3;
+  const timeoutMs = options.timeoutMs ?? 18e4;
+  const sleep = options.sleep ?? ((ms) => new Promise((resolve8) => setTimeout(resolve8, ms)));
+  const maxPolls = Math.max(1, Math.floor(timeoutMs / intervalMs));
+  const remindEvery = Math.max(1, Math.floor(3e4 / intervalMs));
+  log("Registered with Haven \u2014 waiting for you to approve the budget in the dashboard\u2026");
+  for (let i = 0; i < maxPolls; i++) {
+    await sleep(intervalMs);
+    let status;
+    try {
+      status = await api.getConnectorStatus(setupId, apiKey);
+    } catch (err) {
+      if (err instanceof ConnectRequestError && (err.status === 401 || err.status === 404)) {
+        log("This setup ended in Haven \u2014 start a fresh connection from the dashboard when ready.");
+        return "ended";
+      }
+      continue;
+    }
+    if (status.status === "active") {
+      log(
+        status.approved_budget ? `Budget approved \u{1F389} \u2014 I can now spend up to ${describeApprovedBudget(status.approved_budget)} from your Haven wallet.` : "Budget approved \u{1F389} \u2014 the agent can now spend within its Haven rules."
+      );
+      return "approved";
+    }
+    if (status.status === "cancelled" || status.status === "expired" || status.status === "failed") {
+      log(`This setup ended in Haven (${status.status}) \u2014 start a fresh connection from the dashboard when ready.`);
+      return "ended";
+    }
+    if ((i + 1) % remindEvery === 0) {
+      log("Still waiting for budget approval in Haven\u2026");
+    }
+  }
+  log(
+    "Budget approval is still pending in Haven. Approve it in the dashboard whenever you are ready \u2014 the agent tools unlock the moment you do. Verify later with the read-only haven_get_agent tool."
+  );
+  return "pending";
+}
 function completionHandoffLines(result) {
   if (result.errorCode === "manual_runtime_setup_required") {
     return [
@@ -2326,11 +2397,14 @@ async function runCli(argv, io = {
     return 0;
   }
   try {
-    const result = await runConnect(parsed.options, {
-      log: (message) => (parsed.json ? io.stderr : io.stdout)(`${message}
+    const result = await runConnect(
+      { ...parsed.options, waitForApproval: !parsed.json },
+      {
+        log: (message) => (parsed.json ? io.stderr : io.stdout)(`${message}
 `),
-      redactPaths: parsed.json
-    });
+        redactPaths: parsed.json
+      }
+    );
     if (parsed.json) io.stdout(`${JSON.stringify(result.outcome)}
 `);
     return 0;
@@ -2349,7 +2423,15 @@ async function main() {
   const exitCode = await runCli(process.argv.slice(2));
   if (exitCode !== 0) process.exitCode = exitCode;
 }
-if (process.argv[1] && url.pathToFileURL(process.argv[1]).href === (typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('index.cjs', document.baseURI).href))) void main();
+function isCliEntrypoint(argvPath = process.argv[1], moduleUrl = (typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('index.cjs', document.baseURI).href))) {
+  if (!argvPath) return false;
+  try {
+    return fs.realpathSync(argvPath) === fs.realpathSync(url.fileURLToPath(moduleUrl));
+  } catch {
+    return url.pathToFileURL(argvPath).href === moduleUrl;
+  }
+}
+if (isCliEntrypoint()) void main();
 
 exports.CONNECTOR_VERSION = CONNECTOR_VERSION;
 exports.CONNECT_OUTCOME_SCHEMA_VERSION = CONNECT_OUTCOME_SCHEMA_VERSION;