@thinkingai/ae-cli 6.0.40 → 6.0.41

package/dist/{chunk-VKD5WQKN.js → chunk-VLWOLBGZ.js}

@@ -1,7 +1,7 @@
 import {
   getActiveHost,
   getSandboxRuntimeRoot
-} from "./chunk-GJJA4CQZ.js";
+} from "./chunk-3FY3RJ26.js";
 
 // src/core/te-agent-credentials.ts
 import { readFileSync } from "fs";
@@ -94,6 +94,10 @@ function getClaudeConfigDir() {
 }
 
 // src/core/te-agent-client.ts
+import { open, unlink } from "fs/promises";
+import path from "path";
+import { Readable } from "stream";
+import { pipeline } from "stream/promises";
 var TeAgentApiError = class extends Error {
   constructor(message, status, code, body) {
     super(message);
@@ -108,6 +112,7 @@ var TeAgentApiError = class extends Error {
 };
 var DEFAULT_TIMEOUT_MS = 3e4;
 var UPLOAD_TIMEOUT_MS = 12e4;
+var DOWNLOAD_TIMEOUT_MS = 12e4;
 async function fetchWithTimeout(url, init, timeoutMs = DEFAULT_TIMEOUT_MS) {
   const controller = new AbortController();
   const timer = setTimeout(() => controller.abort(), timeoutMs);
@@ -131,7 +136,7 @@ function teClaudeBaseFromActiveHost(hostOverride) {
   const bp = process.env.TE_CLAUDE_BASE_PATH || "/agent";
   return base.endsWith(bp) ? base : base + bp;
 }
-async function signRequest(method, path, rawBody, hostOverride) {
+async function signRequest(method, path2, rawBody, hostOverride) {
   const sandboxCred = tryLoadTeAgentSandboxCredentials();
   if (!hostOverride && sandboxCred && sandboxCred.sandboxId && sandboxCred.sandboxSecretKey) {
     const headers = {
@@ -142,13 +147,13 @@ async function signRequest(method, path, rawBody, hostOverride) {
       headers["Content-Type"] = "application/json";
     }
     return {
-      url: `${sandboxCred.url.replace(/\/$/, "")}${path}`,
+      url: `${sandboxCred.url.replace(/\/$/, "")}${path2}`,
       headers,
       rawBody
     };
   }
-  const { getToken } = await import("./auth-NN55553T.js");
-  const { getCliToken } = await import("./cli-token-GL5MS5FK.js");
+  const { getToken } = await import("./auth-GBMV6TEJ.js");
+  const { getCliToken } = await import("./cli-token-4UPER74P.js");
   const baseUrl = hostOverride ? teClaudeBaseFromActiveHost(hostOverride) : sandboxCred?.url || teClaudeBaseFromActiveHost();
   if (!baseUrl) {
     throw new TeAgentCredentialsError(
@@ -166,7 +171,7 @@ async function signRequest(method, path, rawBody, hostOverride) {
   }
   if (accessToken) {
     return {
-      url: `${baseUrl.replace(/\/$/, "")}${path}`,
+      url: `${baseUrl.replace(/\/$/, "")}${path2}`,
       headers: {
         "Authorization": `bearer ${accessToken}`,
         ...contentTypeHeader
@@ -182,7 +187,7 @@ async function signRequest(method, path, rawBody, hostOverride) {
   }
   if (cliToken) {
     return {
-      url: `${baseUrl.replace(/\/$/, "")}${path}`,
+      url: `${baseUrl.replace(/\/$/, "")}${path2}`,
       headers: {
         "cli-token": cliToken,
         ...contentTypeHeader
@@ -220,9 +225,9 @@ async function parseResponse(response, defaultErrorPrefix) {
   }
   return parsed;
 }
-async function postToMainApp(path, body, hostOverride) {
+async function postToMainApp(path2, body, hostOverride) {
   const rawBody = JSON.stringify(body);
-  const signed = await signRequest("POST", path, rawBody, hostOverride);
+  const signed = await signRequest("POST", path2, rawBody, hostOverride);
   const response = await fetchWithTimeout(signed.url, {
     method: "POST",
     headers: signed.headers,
@@ -230,8 +235,8 @@ async function postToMainApp(path, body, hostOverride) {
   });
   return parseResponse(response, "Main app returned");
 }
-async function getFromMainApp(path, hostOverride) {
-  const signed = await signRequest("GET", path, "", hostOverride);
+async function getFromMainApp(path2, hostOverride) {
+  const signed = await signRequest("GET", path2, "", hostOverride);
   const response = await fetchWithTimeout(signed.url, {
     method: "GET",
     headers: signed.headers
@@ -264,17 +269,17 @@ async function postSandboxSyncPull(args) {
     mode: "merge"
   });
 }
-async function deleteFromMainApp(path, hostOverride) {
-  const signed = await signRequest("DELETE", path, "", hostOverride);
+async function deleteFromMainApp(path2, hostOverride) {
+  const signed = await signRequest("DELETE", path2, "", hostOverride);
   const response = await fetchWithTimeout(signed.url, {
     method: "DELETE",
     headers: signed.headers
   });
   return parseResponse(response, "Main app returned");
 }
-async function patchToMainApp(path, body, hostOverride) {
+async function patchToMainApp(path2, body, hostOverride) {
   const rawBody = JSON.stringify(body);
-  const signed = await signRequest("PATCH", path, rawBody, hostOverride);
+  const signed = await signRequest("PATCH", path2, rawBody, hostOverride);
   const response = await fetchWithTimeout(signed.url, {
     method: "PATCH",
     headers: signed.headers,
@@ -282,9 +287,9 @@ async function patchToMainApp(path, body, hostOverride) {
   });
   return parseResponse(response, "Main app returned");
 }
-async function putToMainApp(path, body, hostOverride) {
+async function putToMainApp(path2, body, hostOverride) {
   const rawBody = JSON.stringify(body);
-  const signed = await signRequest("PUT", path, rawBody, hostOverride);
+  const signed = await signRequest("PUT", path2, rawBody, hostOverride);
   const response = await fetchWithTimeout(signed.url, {
     method: "PUT",
     headers: signed.headers,
@@ -292,8 +297,8 @@ async function putToMainApp(path, body, hostOverride) {
   });
   return parseResponse(response, "Main app returned");
 }
-async function getBufferFromMainApp(path, hostOverride) {
-  const signed = await signRequest("GET", path, "", hostOverride);
+async function getBufferFromMainApp(path2, hostOverride) {
+  const signed = await signRequest("GET", path2, "", hostOverride);
   const response = await fetchWithTimeout(signed.url, {
     method: "GET",
     headers: signed.headers
@@ -325,9 +330,121 @@ async function getBufferFromMainApp(path, hostOverride) {
   const cd = response.headers.get("Content-Disposition") || "";
   const match = cd.match(/filename\*=UTF-8''([^;]+)/);
   const fileName = match ? decodeURIComponent(match[1]) : null;
-  return { buffer, fileName };
+  const contentType = response.headers.get("Content-Type");
+  return { buffer, fileName, contentType };
 }
-async function uploadToMainApp(path, formData, hostOverride) {
+function responseFileName(response) {
+  const disposition = response.headers.get("Content-Disposition") ?? "";
+  const encoded = disposition.match(/filename\*=UTF-8''([^;]+)/i)?.[1];
+  if (encoded) {
+    try {
+      return decodeURIComponent(encoded.trim());
+    } catch {
+      return encoded.trim();
+    }
+  }
+  return disposition.match(/filename="([^"]+)"/i)?.[1] ?? disposition.match(/filename=([^;]+)/i)?.[1]?.trim() ?? null;
+}
+async function* responseBodyChunks(body) {
+  const reader = body.getReader();
+  try {
+    while (true) {
+      const chunk = await reader.read();
+      if (chunk.done) return;
+      yield chunk.value;
+    }
+  } finally {
+    reader.releaseLock();
+  }
+}
+async function downloadFromMainApp(requestPath, outputPath, hostOverride) {
+  const absolutePath = path.resolve(outputPath);
+  let fileHandle;
+  let created = false;
+  const controller = new AbortController();
+  const timer = setTimeout(() => controller.abort(), DOWNLOAD_TIMEOUT_MS);
+  try {
+    try {
+      fileHandle = await open(absolutePath, "wx");
+      created = true;
+    } catch (error) {
+      if (error.code === "EEXIST") {
+        throw new TeAgentApiError(
+          `Output file already exists: ${absolutePath}`,
+          0,
+          "OUTPUT_EXISTS"
+        );
+      }
+      throw new TeAgentApiError(
+        `Unable to create output file: ${absolutePath}`,
+        0,
+        "OUTPUT_ERROR"
+      );
+    }
+    const signed = await signRequest("GET", requestPath, "", hostOverride);
+    let response;
+    try {
+      response = await fetch(signed.url, {
+        method: "GET",
+        headers: signed.headers,
+        signal: controller.signal
+      });
+    } catch (error) {
+      if (error?.name === "AbortError") {
+        throw new TeAgentApiError(
+          `Request timed out (${DOWNLOAD_TIMEOUT_MS}ms): ${signed.url}`,
+          0,
+          "TIMEOUT"
+        );
+      }
+      throw new TeAgentApiError(
+        `Network request failed: ${error instanceof Error ? error.message : String(error)}`,
+        0,
+        "NETWORK_ERROR"
+      );
+    }
+    if (!response.ok) {
+      await parseResponse(response, "Main app returned");
+    }
+    if (!response.body) {
+      throw new TeAgentApiError("Main app returned an empty download stream", 0, "EMPTY_BODY");
+    }
+    const output = fileHandle.createWriteStream();
+    try {
+      await pipeline(
+        Readable.from(responseBodyChunks(response.body)),
+        output
+      );
+    } catch (error) {
+      if (error?.name === "AbortError") {
+        throw new TeAgentApiError(
+          `Request timed out (${DOWNLOAD_TIMEOUT_MS}ms): ${signed.url}`,
+          0,
+          "TIMEOUT"
+        );
+      }
+      throw new TeAgentApiError(
+        `Download failed: ${error instanceof Error ? error.message : String(error)}`,
+        0,
+        "DOWNLOAD_ERROR"
+      );
+    }
+    return {
+      path: absolutePath,
+      bytes: output.bytesWritten,
+      fileName: responseFileName(response),
+      contentType: response.headers.get("Content-Type")
+    };
+  } catch (error) {
+    if (fileHandle) await fileHandle.close().catch(() => void 0);
+    if (created) await unlink(absolutePath).catch(() => void 0);
+    throw error;
+  } finally {
+    clearTimeout(timer);
+    controller.abort();
+  }
+}
+async function uploadToMainApp(path2, formData, hostOverride) {
   const sandboxCred = tryLoadTeAgentSandboxCredentials();
   let baseUrl;
   let authHeaders;
@@ -338,8 +455,8 @@ async function uploadToMainApp(path, formData, hostOverride) {
       "X-Sandbox-Secret-Key": sandboxCred.sandboxSecretKey
     };
   } else {
-    const { getToken } = await import("./auth-NN55553T.js");
-    const { getCliToken } = await import("./cli-token-GL5MS5FK.js");
+    const { getToken } = await import("./auth-GBMV6TEJ.js");
+    const { getCliToken } = await import("./cli-token-4UPER74P.js");
     baseUrl = (hostOverride ? teClaudeBaseFromActiveHost(hostOverride) : sandboxCred?.url || teClaudeBaseFromActiveHost())?.replace(/\/$/, "") || "";
     if (!baseUrl) {
       throw new TeAgentCredentialsError(
@@ -373,7 +490,7 @@ async function uploadToMainApp(path, formData, hostOverride) {
     }
   }
   const response = await fetchWithTimeout(
-    `${baseUrl}${path}`,
+    `${baseUrl}${path2}`,
     {
       method: "POST",
       headers: authHeaders,
@@ -399,5 +516,6 @@ export {
   patchToMainApp,
   putToMainApp,
   getBufferFromMainApp,
+  downloadFromMainApp,
   uploadToMainApp
 };