termarium 0.1.7 → 0.1.8

package/npm/scripts/install.js

@@ -11,6 +11,8 @@ const packageRoot = path.resolve(__dirname, "..", "..");
 const packageJson = require(path.join(packageRoot, "package.json"));
 const repo = "https://github.com/7b7b7b/termarium";
 const version = packageJson.version;
+const downloadRetries = readPositiveIntegerEnv("TERMARIUM_NPM_DOWNLOAD_RETRIES", 3);
+const downloadStallTimeoutMs = readPositiveIntegerEnv("TERMARIUM_NPM_DOWNLOAD_STALL_TIMEOUT_MS", 60_000);
 
 const targets = {
   "darwin-arm64": {
@@ -63,11 +65,11 @@ async function main() {
 
   try {
     console.log(`termarium: downloading ${asset}`);
-    const archive = await download(archiveUrl);
+    const archive = await download(archiveUrl, { label: asset, progress: true });
     fs.writeFileSync(archivePath, archive);
 
     if (process.env.TERMARIUM_NPM_SKIP_CHECKSUM !== "1") {
-      const checksums = (await download(checksumUrl)).toString("utf8");
+      const checksums = (await download(checksumUrl, { label: "SHA256SUMS" })).toString("utf8");
       verifyChecksum(asset, archive, checksums);
     }
 
@@ -129,7 +131,25 @@ function run(command, args) {
   }
 }
 
-function download(url, redirects = 0) {
+async function download(url, options = {}) {
+  let lastError;
+  for (let attempt = 1; attempt <= downloadRetries; attempt += 1) {
+    try {
+      return await downloadOnce(url, options);
+    } catch (error) {
+      lastError = error;
+      if (attempt >= downloadRetries) {
+        break;
+      }
+      console.error(
+        `termarium: download failed (${error.message}); retrying ${attempt}/${downloadRetries - 1}`
+      );
+    }
+  }
+  throw lastError;
+}
+
+function downloadOnce(url, options = {}, redirects = 0) {
   return new Promise((resolve, reject) => {
     const request = https.get(
       url,
@@ -150,7 +170,9 @@ function download(url, redirects = 0) {
             reject(new Error(`too many redirects while downloading ${url}`));
             return;
           }
-          resolve(download(new URL(response.headers.location, url).toString(), redirects + 1));
+          resolve(
+            downloadOnce(new URL(response.headers.location, url).toString(), options, redirects + 1)
+          );
           return;
         }
 
@@ -161,14 +183,133 @@ function download(url, redirects = 0) {
         }
 
         const chunks = [];
-        response.on("data", (chunk) => chunks.push(chunk));
-        response.on("end", () => resolve(Buffer.concat(chunks)));
+        const progress = createDownloadProgress(options, response.headers["content-length"]);
+        response.on("data", (chunk) => {
+          chunks.push(chunk);
+          progress.update(chunk.length);
+        });
+        response.on("end", () => {
+          progress.finish();
+          resolve(Buffer.concat(chunks));
+        });
       }
     );
+    request.setTimeout(downloadStallTimeoutMs, () => {
+      request.destroy(new Error(`download stalled for more than ${downloadStallTimeoutMs / 1000}s`));
+    });
     request.on("error", reject);
   });
 }
 
+function createDownloadProgress(options, contentLength) {
+  if (!options.progress) {
+    return { update() {}, finish() {} };
+  }
+
+  const label = options.label ?? "download";
+  const total = Number(contentLength);
+  const hasTotal = Number.isFinite(total) && total > 0;
+  const startedAt = Date.now();
+  const useSingleLine = process.stderr.isTTY;
+  const intervalMs = useSingleLine ? 100 : 5_000;
+  let received = 0;
+  let lastRenderAt = 0;
+  let finished = false;
+
+  const render = (force = false) => {
+    if (finished) {
+      return;
+    }
+    const now = Date.now();
+    if (!force && now - lastRenderAt < intervalMs) {
+      return;
+    }
+    lastRenderAt = now;
+
+    const elapsedSeconds = Math.max((now - startedAt) / 1000, 0.001);
+    const speed = received / elapsedSeconds;
+    const receivedText = formatBytes(received);
+    let message;
+
+    if (hasTotal) {
+      const percent = Math.min(received / total, 1);
+      const eta = speed > 0 ? (total - received) / speed : 0;
+      message = [
+        `termarium: ${label}`,
+        formatPercent(percent),
+        progressBar(percent),
+        `${receivedText}/${formatBytes(total)}`,
+        `${formatBytes(speed)}/s`,
+        `ETA ${formatDuration(eta)}`,
+      ].join(" ");
+    } else {
+      message = `termarium: ${label} ${receivedText} ${formatBytes(speed)}/s`;
+    }
+
+    if (useSingleLine) {
+      process.stderr.write(`\r${message}`);
+    } else {
+      console.error(message);
+    }
+  };
+
+  return {
+    update(chunkLength) {
+      received += chunkLength;
+      render(false);
+    },
+    finish() {
+      render(true);
+      finished = true;
+      if (useSingleLine) {
+        process.stderr.write("\n");
+      }
+    },
+  };
+}
+
+function progressBar(percent) {
+  const width = 20;
+  const filled = Math.max(0, Math.min(width, Math.round(percent * width)));
+  return `[${"#".repeat(filled)}${"-".repeat(width - filled)}]`;
+}
+
+function formatPercent(percent) {
+  return `${Math.round(percent * 100).toString().padStart(3, " ")}%`;
+}
+
+function formatBytes(bytes) {
+  const units = ["B", "KiB", "MiB", "GiB"];
+  let value = bytes;
+  let unitIndex = 0;
+  while (value >= 1024 && unitIndex < units.length - 1) {
+    value /= 1024;
+    unitIndex += 1;
+  }
+  const precision = value >= 100 || unitIndex === 0 ? 0 : 1;
+  return `${value.toFixed(precision)} ${units[unitIndex]}`;
+}
+
+function formatDuration(seconds) {
+  if (!Number.isFinite(seconds) || seconds < 0) {
+    return "--";
+  }
+  if (seconds < 60) {
+    return `${Math.ceil(seconds)}s`;
+  }
+  const minutes = Math.floor(seconds / 60);
+  const remainingSeconds = Math.ceil(seconds % 60);
+  return `${minutes}m${remainingSeconds.toString().padStart(2, "0")}s`;
+}
+
+function readPositiveIntegerEnv(name, fallback) {
+  const value = Number(process.env[name]);
+  if (Number.isInteger(value) && value > 0) {
+    return value;
+  }
+  return fallback;
+}
+
 function escapePowerShell(value) {
   return value.replace(/'/g, "''");
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "termarium",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "description": "Termarium 穹顶终端: a quiet terminal planetarium.",
   "license": "MIT AND CC-BY-SA-4.0 AND CC-BY-4.0",
   "homepage": "https://github.com/7b7b7b/termarium#readme",