trajectories-sh 1.2.2 → 1.3.1

package/dist/cli.js

@@ -101,6 +101,26 @@ program.command("upload <directory>").description("Upload a trajectory job direc
     process.exit(1);
   }
 });
+program.command("download <url>").description("Download a trajectory job from trajectories.sh").option("-o, --output <dir>", "Output directory (default: slug name)").action(async (url, opts) => {
+  const chalk = (await import("chalk")).default;
+  console.log(chalk.bold(`
+  Downloading trajectory...
+`));
+  try {
+    const { downloadJob, parseJobId } = await import("./download-YWW2BV6K.js");
+    const outputDir = opts.output ?? parseJobId(url).replace(/^org:/, "").replace(/:/g, "-");
+    const result = await downloadJob(url, outputDir);
+    console.log(chalk.green.bold(`  \u2713 Downloaded ${result.files} files`));
+    console.log(`    Size: ${(result.totalSize / 1024 / 1024).toFixed(1)} MB`);
+    console.log(`    Output: ${outputDir}
+`);
+  } catch (e) {
+    console.error(chalk.red(`
+  \u2717 ${e.message}
+`));
+    process.exit(1);
+  }
+});
 program.command("whoami").description("Show current authentication status").action(async () => {
   const chalk = (await import("chalk")).default;
   const { status } = await import("./auth-UD3KTSHK.js");

package/dist/download-YWW2BV6K.js

@@ -0,0 +1,111 @@
+import {
+  getApiUrl,
+  getAuthHeader
+} from "./chunk-P6J5X7LP.js";
+
+// src/download.ts
+import { mkdirSync, writeFileSync } from "fs";
+import { join, dirname } from "path";
+var UUID_RE = /[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}/i;
+function parseJobId(input) {
+  const trimmed = input.trim();
+  if (UUID_RE.test(trimmed) && !trimmed.includes("/")) {
+    return trimmed.match(UUID_RE)[0];
+  }
+  const trajMatch = trimmed.match(/\/trajectories\/([0-9a-f-]{36})/i);
+  if (trajMatch) return trajMatch[1];
+  const trajViewerMatch = trimmed.match(/\/traj\/([0-9a-f-]{36})\/([^/]+)/i);
+  if (trajViewerMatch) {
+    return `org:${trajViewerMatch[1]}:${trajViewerMatch[2]}`;
+  }
+  const anyUuid = trimmed.match(UUID_RE);
+  if (anyUuid) return anyUuid[0];
+  throw new Error(`Could not find a trajectory job ID in: ${input}`);
+}
+async function downloadJob(input, outputDir) {
+  const apiUrl = getApiUrl();
+  const auth = await getAuthHeader();
+  const headers = {};
+  if (auth) headers["Authorization"] = auth;
+  let jobId = parseJobId(input);
+  if (jobId.startsWith("org:")) {
+    const [, orgId, slug2] = jobId.split(":");
+    const resp = await fetch(`${apiUrl}/api/org/trajectories/public/${orgId}?slug=${slug2}`, { headers });
+    if (!resp.ok) {
+      const configResp = await fetch(`${apiUrl}/api/viewer/${orgId}/${slug2}/api/jobs`, { headers });
+      if (!configResp.ok) throw new Error(`Could not resolve job: ${orgId}/${slug2}`);
+      const configData = await configResp.json();
+      const items = configData.items || [];
+      if (items.length > 0 && items[0].id) {
+        jobId = items[0].id;
+      } else {
+        throw new Error(`Could not resolve job ID for ${orgId}/${slug2}`);
+      }
+    } else {
+      const data = await resp.json();
+      jobId = data.id;
+    }
+  }
+  const listResp = await fetch(`${apiUrl}/api/cli/download/${jobId}/files`, {
+    headers,
+    signal: AbortSignal.timeout(3e4)
+  });
+  if (!listResp.ok) {
+    const text = await listResp.text();
+    throw new Error(`Failed to list files: ${text}`);
+  }
+  const { slug, files } = await listResp.json();
+  console.log(`  Job: ${slug} (${jobId})`);
+  console.log(`  ${files.length} files to download`);
+  mkdirSync(outputDir, { recursive: true });
+  let downloaded = 0;
+  let totalSize = 0;
+  const concurrency = 8;
+  const pending = [];
+  for (const file of files) {
+    const task = (async () => {
+      for (let attempt = 0; attempt < 3; attempt++) {
+        try {
+          const fileResp = await fetch(
+            `${apiUrl}/api/cli/download/${jobId}/file?path=${encodeURIComponent(file.path)}`,
+            { headers, signal: AbortSignal.timeout(6e4) }
+          );
+          if (!fileResp.ok) throw new Error(`HTTP ${fileResp.status}`);
+          const data = Buffer.from(await fileResp.arrayBuffer());
+          const outPath = join(outputDir, file.path);
+          mkdirSync(dirname(outPath), { recursive: true });
+          writeFileSync(outPath, data);
+          downloaded++;
+          totalSize += data.length;
+          if (downloaded % 10 === 0 || downloaded === files.length) {
+            process.stdout.write(`  \u2193 ${downloaded}/${files.length} downloaded\r`);
+          }
+          return;
+        } catch {
+          if (attempt < 2) await new Promise((r) => setTimeout(r, 1e3 * (attempt + 1)));
+        }
+      }
+      console.log(`  \u2717 ${file.path}`);
+    })();
+    pending.push(task);
+    if (pending.length >= concurrency) {
+      await Promise.race(pending);
+      const next = [];
+      for (const p of pending) {
+        let settled = false;
+        p.then(() => settled = true, () => settled = true);
+        if (!settled) next.push(p);
+      }
+      if (next.length === pending.length) await Promise.race(pending);
+      pending.length = 0;
+      pending.push(...next);
+    }
+  }
+  await Promise.all(pending);
+  console.log("");
+  return { files: downloaded, totalSize };
+}
+export {
+  downloadJob,
+  parseJobId
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "trajectories-sh",
-  "version": "1.2.2",
+  "version": "1.3.1",
   "description": "CLI for uploading trajectory jobs to trajectories.sh",
   "type": "module",
   "bin": {