@oxygen-agent/cli 1.1011.326 → 1.1011.440

package/dist/http-client.js

@@ -1,5 +1,6 @@
 import { OXYGEN_VERSION, OxygenError, isCliResult, isVersionGreater, parseSemver, readEnvelopeCompatibility, vercelProtectionBypassHeaders, withRetryAfterDetails, } from "@oxygen/shared";
 import { randomUUID } from "node:crypto";
+import { open, unlink } from "node:fs/promises";
 import { recordMutatingRequestAnswered, recoverFromCliFloorRejection } from "./auto-update.js";
 import { defaultApiUrl, loadCredentialsWithEndpoint } from "./credentials.js";
 import { assertResolvedApiUrl, cliEndpointDetails, describeCliEndpoint, isProdApiUrl, resolveCliUpdateGuidance, } from "./runtime.js";
@@ -86,7 +87,7 @@ path, options = {}) {
     const requestInit = {
         method: options.method ?? "GET",
         headers,
-        ...(options.binaryMaxBytes === undefined ? {} : { redirect: "error" }),
+        ...(options.binaryMaxBytes === undefined && options.binaryOutputPath === undefined ? {} : { redirect: "error" }),
         ...(options.body ? { body: JSON.stringify(options.body) } : {}),
         ...(options.formData ? { body: options.formData } : {}),
     };
@@ -97,6 +98,9 @@ path, options = {}) {
         requestInit.signal = timeout.signal;
     try {
         response = await fetchImpl(`${apiUrl}${path}`, requestInit);
+        if (response.ok && options.binaryOutputPath !== undefined) {
+            return await streamBinaryResponse(response, options.binaryOutputPath);
+        }
         if (response.ok && options.binaryMaxBytes !== undefined) {
             return await readBoundedBinaryResponse(response, options.binaryMaxBytes);
         }
@@ -195,6 +199,42 @@ async function readBoundedBinaryResponse(response, maxBytes) {
     }
     return bytes;
 }
+async function streamBinaryResponse(response, output) {
+    let file;
+    try {
+        file = await open(output, "wx");
+    }
+    catch (error) {
+        await response.body?.cancel();
+        if (error && typeof error === "object" && "code" in error && error.code === "EEXIST") {
+            throw new OxygenError("output_exists", "The output file already exists. Choose another --output path.", { exitCode: 2 });
+        }
+        throw error;
+    }
+    const reader = response.body?.getReader();
+    let bytes = 0;
+    try {
+        if (!reader)
+            throw new OxygenError("invalid_response", "The download had no body.");
+        for (;;) {
+            const chunk = await reader.read();
+            if (chunk.done)
+                break;
+            await file.writeFile(chunk.value);
+            bytes += chunk.value.byteLength;
+        }
+        await file.close();
+        return { path: output, bytes };
+    }
+    catch (error) {
+        await file.close().catch(() => { });
+        await unlink(output).catch(() => { });
+        throw error;
+    }
+    finally {
+        await reader?.cancel().catch(() => { });
+    }
+}
 export async function ensureFreshCliForApiUrl(apiUrl, options = {}) {
     const checkKey = `${apiUrl}|${OXYGEN_VERSION}`;
     if (cliCompatibilityCheckKeys.has(checkKey))