@codearcade/subtitle-generator 1.0.2 → 1.0.4

package/cli/index.js

@@ -1,4 +1,4 @@
-#!/usr/bin/env node
+#!/usr/bin/env node
 
 const args = process.argv.slice(2);
 

package/init/index.js

@@ -3,6 +3,9 @@ import path from "path";
 import fs from "fs";
 import os from "os";
 import { execSync } from "child_process";
+import { Readable } from "stream";
+import { pipeline } from "stream/promises";
+import unzipper from "unzipper";
 
 const init = async () => {
   const { modelSize } = await inquirer.prompt([
@@ -38,36 +41,51 @@ const init = async () => {
     }
 
     console.log(`Downloading ${label}...`);
-    const res = await fetch(url);
+    const res = await fetch(url, {
+      redirect: "follow",
+      headers: {
+        "User-Agent": "node",
+      },
+    });
+
+    const contentType = res.headers.get("content-type");
+
+    if (!contentType || !contentType.includes("application")) {
+      throw new Error(
+        `Invalid response. Expected binary file but got: ${contentType}`,
+      );
+    }
 
-    if (!res.ok) throw new Error(`Failed to download: ${res.statusText}`);
+    if (!res.ok) {
+      throw new Error(`Failed to download: ${res.statusText}`);
+    }
 
     const total = Number(res.headers.get("content-length")) || 0;
     let downloaded = 0;
 
-    const file = fs.createWriteStream(dest);
-
-    await res.body.pipeTo(
-      new WritableStream({
-        write(chunk) {
-          downloaded += chunk.length;
-          if (total) {
-            const percent = ((downloaded / total) * 100).toFixed(2);
-            process.stdout.write(`Downloading ${label}... ${percent}%\r`);
-          } else {
-            process.stdout.write(`Downloaded ${downloaded} bytes\r`);
-          }
-          file.write(chunk);
-        },
-        close() {
-          file.end();
-          console.log(`\n${label} download complete.`);
-        },
-        abort(err) {
-          file.destroy(err);
-        },
-      }),
-    );
+    const fileStream = fs.createWriteStream(dest);
+
+    const nodeStream = Readable.fromWeb(res.body);
+
+    nodeStream.on("data", (chunk) => {
+      downloaded += chunk.length;
+      if (total) {
+        const percent = ((downloaded / total) * 100).toFixed(2);
+        process.stdout.write(`Downloading ${label}... ${percent}%\r`);
+      }
+    });
+
+    await pipeline(nodeStream, fileStream);
+
+    const stats = fs.statSync(dest);
+
+    if (stats.size < 100000) {
+      throw new Error(
+        `Downloaded file too small (${stats.size} bytes). Likely HTML instead of ZIP.`,
+      );
+    }
+
+    console.log(`\n${label} download complete.`);
   }
 
   // 1. Download the AI Model
@@ -89,11 +107,11 @@ const init = async () => {
   }
 
   if (isWindows) {
-    // Windows Setup
     const binaryDest = path.join(whisperDir, "whisper-cli.exe");
+
     if (!fs.existsSync(binaryDest)) {
       console.log("Detecting Windows architecture...");
-      const arch = os.arch(); // Usually 'x64', 'ia32', or 'arm64'
+      const arch = os.arch();
 
       let zipName;
 
@@ -102,15 +120,10 @@ const init = async () => {
       } else if (arch === "ia32") {
         zipName = "whisper-bin-Win32.zip";
       } else if (arch === "arm64") {
-        // Windows ARM can emulate x64 executables
-        console.log(
-          "ARM64 architecture detected. Using x64 binary via Windows emulation...",
-        );
+        console.log("ARM64 detected → using x64 binary via emulation...");
         zipName = "whisper-bin-x64.zip";
       } else {
-        throw new Error(
-          `Unsupported Windows architecture: ${arch}. Whisper.cpp does not provide pre-compiled binaries for this system.`,
-        );
+        throw new Error(`Unsupported architecture: ${arch}`);
       }
 
       const zipUrl = `https://github.com/ggml-org/whisper.cpp/releases/download/v1.8.4/${zipName}`;
@@ -118,72 +131,53 @@ const init = async () => {
 
       await downloadFile(zipUrl, zipPath, `Whisper Windows Binary (${arch})`);
 
-      console.log("Extracting binary...");
-
-      let extractionSuccess = false;
-      let maxRetries = 5;
-      let retryDelay = 1000; // 1 second
-
-      // 1. The Retry Loop for safe extraction
-      for (let attempt = 1; attempt <= maxRetries; attempt++) {
-        try {
-          execSync(
-            `powershell -command "Expand-Archive -Force -LiteralPath '${zipPath}' -DestinationPath '${whisperDir}'"`,
-            { stdio: "inherit" },
-          );
-          extractionSuccess = true;
-          break; // It worked! Break out of the loop.
-        } catch (error) {
-          if (attempt === maxRetries) {
-            console.error(
-              `\nExtraction failed after ${maxRetries} attempts. File might be permanently locked or corrupted.`,
-            );
-            throw error;
-          }
-          console.log(
-            `\nFile locked by Windows (likely Antivirus). Retrying in 1 second... (Attempt ${attempt} of ${maxRetries})`,
-          );
-          await new Promise((resolve) => setTimeout(resolve, retryDelay));
-        }
+      // ✅ sanity check
+      const stats = fs.statSync(zipPath);
+      if (stats.size < 1_000_000) {
+        throw new Error(`Downloaded ZIP too small (${stats.size} bytes)`);
       }
 
-      // 2. The Cleanup Phase (Only runs if extraction worked)
-      if (extractionSuccess) {
-        try {
-          // Move everything out of the "Release" folder
-          const possibleReleaseDir = path.join(whisperDir, "Release");
+      console.log("Extracting binary (unzipper)...");
+
+      await new Promise((resolve, reject) => {
+        fs.createReadStream(zipPath)
+          .pipe(unzipper.Parse())
+          .on("entry", (entry) => {
+            const filePath = path.join(whisperDir, entry.path);
 
-          if (fs.existsSync(possibleReleaseDir)) {
-            const files = fs.readdirSync(possibleReleaseDir);
+            if (entry.type === "Directory") {
+              fs.mkdirSync(filePath, { recursive: true });
+              entry.autodrain();
+            } else {
+              fs.mkdirSync(path.dirname(filePath), { recursive: true });
 
-            for (const file of files) {
-              fs.renameSync(
-                path.join(possibleReleaseDir, file),
-                path.join(whisperDir, file),
-              );
+              entry.pipe(fs.createWriteStream(filePath));
             }
-            // Delete the now-empty Release folder
-            fs.rmdirSync(possibleReleaseDir);
-          }
-
-          // Rename extracted main.exe to match our wrapper expectations
-          const extractedMain = path.join(whisperDir, "main.exe");
-          if (fs.existsSync(extractedMain)) {
-            fs.renameSync(extractedMain, binaryDest);
-          } else {
-            console.error("Warning: Could not find main.exe after extraction.");
-          }
-
-          // Clean up the zip file
-          if (fs.existsSync(zipPath)) {
-            fs.unlinkSync(zipPath);
-          }
-
-          console.log(`Whisper binary setup complete for Windows (${arch}).`);
-        } catch (error) {
-          console.error("Error during binary cleanup/renaming:", error.message);
+          })
+          .on("close", resolve)
+          .on("error", reject);
+      });
+
+      // 🔍 Handle "Release" folder
+      const releaseDir = path.join(whisperDir, "Release");
+
+      if (fs.existsSync(releaseDir)) {
+        const files = fs.readdirSync(releaseDir);
+
+        for (const file of files) {
+          fs.renameSync(
+            path.join(releaseDir, file),
+            path.join(whisperDir, file),
+          );
         }
+
+        fs.rmSync(releaseDir, { recursive: true, force: true });
       }
+
+      // ✅ cleanup zip
+      fs.unlinkSync(zipPath);
+
+      console.log(`Whisper binary setup complete for Windows (${arch}).`);
     } else {
       console.log("Whisper binary already exists for Windows.");
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@codearcade/subtitle-generator",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "main": "index.js",
   "type": "module",
   "module": "./dist/index.js",
@@ -58,6 +58,7 @@
     "@types/bun": "^1.3.11"
   },
   "dependencies": {
-    "inquirer": "^13.3.2"
+    "inquirer": "^13.3.2",
+    "unzipper": "^0.12.3"
   }
 }