qmai-cli-public 0.1.3 → 0.1.4

package/npm/install.js

@@ -2,9 +2,11 @@
 
 const fs = require("node:fs");
 const fsp = require("node:fs/promises");
+const http = require("node:http");
 const https = require("node:https");
 const path = require("node:path");
 const { spawnSync } = require("node:child_process");
+const { URL } = require("node:url");
 
 const packageRoot = path.resolve(__dirname, "..");
 const distDir = path.join(packageRoot, "npm", "dist");
@@ -21,6 +23,10 @@ const archMap = {
   arm64: "arm64",
 };
 
+const DOWNLOAD_TIMEOUT_MS = parseInt(process.env.QMAI_CLI_DOWNLOAD_TIMEOUT_MS || "30000", 10);
+const DOWNLOAD_RETRIES = parseInt(process.env.QMAI_CLI_DOWNLOAD_RETRIES || "3", 10);
+const MAX_REDIRECTS = parseInt(process.env.QMAI_CLI_DOWNLOAD_MAX_REDIRECTS || "5", 10);
+
 function fail(message) {
   console.error(message);
   process.exit(1);
@@ -59,22 +65,70 @@ function getAssetInfo() {
   return { version, platform, arch, baseName, archiveName, binaryName };
 }
 
-function getDownloadUrl(archiveName) {
-  const customUrl = process.env.QMAI_CLI_DOWNLOAD_URL;
-  if (customUrl) {
-    return customUrl;
+function parseList(value) {
+  return (value || "")
+    .split(",")
+    .map((item) => item.trim())
+    .filter(Boolean);
+}
+
+function getDownloadCandidates(archiveName) {
+  const candidates = [];
+  const seen = new Set();
+
+  const add = (url) => {
+    if (!url || seen.has(url)) {
+      return;
+    }
+    seen.add(url);
+    candidates.push(url);
+  };
+
+  add(process.env.QMAI_CLI_DOWNLOAD_URL);
+  parseList(process.env.QMAI_CLI_FALLBACK_DOWNLOAD_URLS).forEach(add);
+
+  const baseUrls = [
+    process.env.QMAI_CLI_RELEASE_BASE_URL || "https://github.com/feixiao629/qmai-cli-public/releases/download",
+    ...parseList(process.env.QMAI_CLI_FALLBACK_RELEASE_BASE_URLS),
+  ];
+  for (const baseUrl of baseUrls) {
+    const normalized = baseUrl.replace(/\/+$/, "");
+    add(`${normalized}/v${packageJson.version}/${archiveName}`);
+  }
+
+  return candidates;
+}
+
+function getHttpClient(url) {
+  const protocol = new URL(url).protocol;
+  if (protocol === "https:") {
+    return https;
+  }
+  if (protocol === "http:") {
+    return http;
   }
+  throw new Error(`unsupported protocol: ${protocol}`);
+}
 
-  const baseUrl =
-    process.env.QMAI_CLI_RELEASE_BASE_URL ||
-    "https://github.com/feixiao629/qmai-cli-public/releases/download";
-  return `${baseUrl}/v${packageJson.version}/${archiveName}`;
+async function removeIfExists(filePath) {
+  await fsp.rm(filePath, { force: true });
 }
 
-async function downloadFile(url, outputPath) {
+async function downloadFileOnce(url, outputPath, redirectCount = 0) {
+  if (redirectCount > MAX_REDIRECTS) {
+    throw new Error(`redirect limit exceeded (${MAX_REDIRECTS})`);
+  }
+
   await new Promise((resolve, reject) => {
     const file = fs.createWriteStream(outputPath);
-    const request = https.get(
+    const client = getHttpClient(url);
+
+    const cleanup = (callback) => {
+      file.destroy();
+      removeIfExists(outputPath).then(() => callback()).catch(callback);
+    };
+
+    const request = client.get(
       url,
       {
         headers: {
@@ -83,42 +137,85 @@ async function downloadFile(url, outputPath) {
         },
       },
       (response) => {
-        if (
-          response.statusCode &&
-          response.statusCode >= 300 &&
-          response.statusCode < 400 &&
-          response.headers.location
-        ) {
-          file.close();
-          fs.unlinkSync(outputPath);
-          downloadFile(response.headers.location, outputPath).then(resolve).catch(reject);
+        const { statusCode = 0, statusMessage = "", headers } = response;
+
+        if (statusCode >= 300 && statusCode < 400 && headers.location) {
+          const nextUrl = new URL(headers.location, url).toString();
+          response.resume();
+          file.close(() => {
+            removeIfExists(outputPath)
+              .then(() => downloadFileOnce(nextUrl, outputPath, redirectCount + 1))
+              .then(resolve)
+              .catch(reject);
+          });
           return;
         }
 
-        if (response.statusCode !== 200) {
-          file.close();
-          fs.unlinkSync(outputPath);
-          reject(
-            new Error(`download failed: ${response.statusCode} ${response.statusMessage || ""}`.trim())
+        if (statusCode !== 200) {
+          response.resume();
+          cleanup(() =>
+            reject(new Error(`download failed: ${statusCode} ${statusMessage}`.trim()))
           );
           return;
         }
 
         response.pipe(file);
         file.on("finish", () => file.close(resolve));
+        response.on("error", (error) => {
+          cleanup(() => reject(error));
+        });
       }
     );
 
+    request.setTimeout(DOWNLOAD_TIMEOUT_MS, () => {
+      request.destroy(new Error(`download timeout after ${DOWNLOAD_TIMEOUT_MS}ms`));
+    });
+
     request.on("error", (error) => {
-      file.close();
-      if (fs.existsSync(outputPath)) {
-        fs.unlinkSync(outputPath);
-      }
-      reject(error);
+      cleanup(() => reject(error));
+    });
+
+    file.on("error", (error) => {
+      request.destroy(error);
     });
   });
 }
 
+async function downloadFile(urls, outputPath) {
+  const attempts = [];
+
+  for (const url of urls) {
+    for (let attempt = 1; attempt <= DOWNLOAD_RETRIES; attempt += 1) {
+      try {
+        console.log(`下载 qmai 二进制（${attempt}/${DOWNLOAD_RETRIES}）：${url}`);
+        await downloadFileOnce(url, outputPath);
+        return;
+      } catch (error) {
+        attempts.push(`${url} [${attempt}/${DOWNLOAD_RETRIES}]: ${error.message}`);
+        console.warn(`下载失败：${error.message}`);
+        await removeIfExists(outputPath);
+      }
+    }
+  }
+
+  throw new Error(
+    [
+      "无法下载 qmai 二进制。",
+      `已尝试 ${urls.length} 个下载源，每个最多重试 ${DOWNLOAD_RETRIES} 次。`,
+      `超时设置：${DOWNLOAD_TIMEOUT_MS}ms。`,
+      "可设置以下环境变量重试：",
+      "- QMAI_CLI_DOWNLOAD_TIMEOUT_MS=60000",
+      "- QMAI_CLI_DOWNLOAD_RETRIES=5",
+      "- QMAI_CLI_RELEASE_BASE_URL=https://<your-mirror>/releases/download",
+      "- QMAI_CLI_FALLBACK_RELEASE_BASE_URLS=https://mirror-a/releases/download,https://mirror-b/releases/download",
+      "- QMAI_CLI_DOWNLOAD_URL=https://.../qmai_<version>_<os>_<arch>.tar.gz",
+      "- QMAI_CLI_LOCAL_ASSET_DIR=/path/to/pre-downloaded-assets",
+      "失败明细：",
+      ...attempts.map((item) => `  - ${item}`),
+    ].join("\n")
+  );
+}
+
 async function extractArchive(archivePath, targetDir, platform) {
   if (platform === "windows") {
     runCommand("powershell.exe", [
@@ -153,9 +250,8 @@ async function main() {
   if (localAssetDir) {
     await copyLocalAsset(localAssetDir, archiveName, archivePath);
   } else {
-    const url = getDownloadUrl(archiveName);
-    console.log(`下载 qmai 二进制：${url}`);
-    await downloadFile(url, archivePath);
+    const urls = getDownloadCandidates(archiveName);
+    await downloadFile(urls, archivePath);
   }
 
   await extractArchive(archivePath, distDir, platform);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "qmai-cli-public",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "qmai-cli npm installer package",
   "license": "MIT",
   "repository": {