@wavespeed/cli 0.4.26 → 0.4.27

package/dist/commands/history.js

@@ -2,6 +2,7 @@ import chalk from 'chalk';
 import ora from 'ora';
 import { fetchHistory, PREDICTION_STATUSES } from '../lib/api.js';
 import { toIso } from '../lib/time.js';
+import { describeOutput } from '../lib/output.js';
 function statusBadge(status) {
     const map = {
         completed: chalk.bgGreen.black,
@@ -40,7 +41,7 @@ export function beyondWindow(since, until) {
 function firstOutput(item) {
     const out = item.outputs?.[0];
     if (typeof out === 'string')
-        return out;
+        return describeOutput(out);
     return '';
 }
 export function registerHistory(program) {

package/dist/lib/output.js

@@ -11,6 +11,73 @@ function extFromUrl(url, fallback) {
         return m[1].toLowerCase();
     return fallback;
 }
+// With enable_base64_output=true each URL output comes back as naked base64,
+// but an output over 64 MiB or one the API failed to fetch stays a URL, and
+// text/JSON outputs are unchanged — so the kind is decided per output.
+const BASE64_RE = /^[A-Za-z0-9+/]+={0,2}$/;
+const DATA_URI_RE = /^data:([^;,]*)(?:;[^,]*)?;base64,(.*)$/s;
+export function isHttpUrl(value) {
+    return /^https?:\/\//i.test(value);
+}
+/** Naked base64 as returned for enable_base64_output (short strings are treated as text). */
+export function isBase64Output(value) {
+    return value.length >= 64 && value.length % 4 === 0 && BASE64_RE.test(value);
+}
+const MAGIC = [
+    [[0x89, 0x50, 0x4e, 0x47], 'png'],
+    [[0xff, 0xd8, 0xff], 'jpg'],
+    [[0x47, 0x49, 0x46, 0x38], 'gif'],
+    [[0x1a, 0x45, 0xdf, 0xa3], 'webm'],
+    [[0x49, 0x44, 0x33], 'mp3'],
+    [[0x66, 0x4c, 0x61, 0x43], 'flac'],
+    [[0x4f, 0x67, 0x67, 0x53], 'ogg'],
+    [[0x67, 0x6c, 0x54, 0x46], 'glb'],
+    [[0x50, 0x4b, 0x03, 0x04], 'zip'],
+    [[0x25, 0x50, 0x44, 0x46], 'pdf'],
+];
+export function extFromBytes(buf, fallback) {
+    for (const [sig, ext] of MAGIC) {
+        if (sig.every((b, i) => buf[i] === b))
+            return ext;
+    }
+    const at = (o, s) => buf.subarray(o, o + s.length).toString('latin1') === s;
+    if (at(0, 'RIFF') && at(8, 'WEBP'))
+        return 'webp';
+    if (at(0, 'RIFF') && at(8, 'WAVE'))
+        return 'wav';
+    if (at(4, 'ftyp'))
+        return at(8, 'M4A') ? 'm4a' : at(8, 'qt') ? 'mov' : 'mp4';
+    if (buf.length >= 2 && buf[0] === 0xff && (buf[1] & 0xe0) === 0xe0)
+        return 'mp3';
+    return fallback;
+}
+const EXT_FROM_MIME = {
+    'image/jpeg': 'jpg', 'image/png': 'png', 'image/webp': 'webp', 'image/gif': 'gif',
+    'video/mp4': 'mp4', 'video/webm': 'webm', 'audio/mpeg': 'mp3', 'audio/wav': 'wav',
+};
+/** Bytes of an inline (base64 or data URI) output, or null when it is not inline. */
+function inlineBytes(value, fallbackExt) {
+    const m = value.match(DATA_URI_RE);
+    if (m) {
+        const buf = Buffer.from(m[2], 'base64');
+        return { buf, ext: EXT_FROM_MIME[m[1].toLowerCase()] ?? extFromBytes(buf, fallbackExt) };
+    }
+    if (!isBase64Output(value))
+        return null;
+    const buf = Buffer.from(value, 'base64');
+    return { buf, ext: extFromBytes(buf, fallbackExt) };
+}
+/** Short label for an output: URLs as-is, inline data summarised instead of dumped. */
+export function describeOutput(value) {
+    if (isHttpUrl(value))
+        return value;
+    const m = value.match(DATA_URI_RE);
+    const b64 = m ? m[2] : isBase64Output(value) ? value : null;
+    if (b64 === null)
+        return value;
+    const bytes = Math.floor((b64.length * 3) / 4);
+    return `(base64 data, ${bytes >= 1024 * 1024 ? (bytes / 1024 / 1024).toFixed(1) + ' MB' : Math.ceil(bytes / 1024) + ' KB'})`;
+}
 function slug(input) {
     return input
         .toLowerCase()
@@ -31,7 +98,8 @@ export async function downloadOutputs(urls, opts = {}) {
     const saved = [];
     for (let i = 0; i < urls.length; i++) {
         const url = urls[i];
-        const ext = extFromUrl(url, opts.ext || 'bin');
+        const inline = isHttpUrl(url) ? null : inlineBytes(url, opts.ext || 'bin');
+        const ext = inline ? inline.ext : extFromUrl(url, opts.ext || 'bin');
         let filepath;
         if (opts.template) {
             const rendered = renderTemplate(opts.template, {
@@ -53,6 +121,14 @@ export async function downloadOutputs(urls, opts = {}) {
             filepath = path.join(baseDir, `${ts}-${tag}${suffix}.${ext}`);
         }
         fs.mkdirSync(path.dirname(filepath), { recursive: true });
+        if (inline) {
+            fs.writeFileSync(filepath, inline.buf);
+            saved.push(filepath);
+            continue;
+        }
+        if (!isHttpUrl(url)) {
+            throw new Error(`Output ${i + 1} is neither a URL nor base64 data, so it cannot be downloaded: ${url.slice(0, 80)}`);
+        }
         const res = await fetch(url);
         if (!res.ok || !res.body) {
             throw new Error(`Download failed for ${url}: ${res.status} ${res.statusText}`);
@@ -72,6 +148,6 @@ export function printOutputs(saved, remote) {
     }
     console.log(chalk.gray(saved.length ? 'Source URLs:' : 'URLs:'));
     for (const u of remote)
-        console.log('  ' + chalk.cyan(u));
+        console.log('  ' + chalk.cyan(describeOutput(u)));
     console.log();
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wavespeed/cli",
-  "version": "0.4.26",
+  "version": "0.4.27",
   "description": "WaveSpeed CLI — one command to run any WaveSpeed AI model from your terminal.",
   "publishConfig": {
     "access": "public"