feat: improve install progress and native unzip

src/cache.ts

@@ -1,5 +1,11 @@
 import { ensureDirSync } from "https://deno.land/[email protected]/fs/ensure_dir.ts";
 import { resolve } from "https://deno.land/[email protected]/path/mod.ts";
+import { ensureDir } from "https://deno.land/[email protected]/fs/ensure_dir.ts";
+import { dirname } from "https://deno.land/[email protected]/path/dirname.ts";
+import { join } from "https://deno.land/[email protected]/path/join.ts";
+import { ZipReader } from "https://deno.land/x/[email protected]/index.js";
+import ProgressBar from "https://deno.land/x/[email protected]/mod.ts";
+import { exists } from "https://deno.land/[email protected]/fs/exists.ts";
 
 export const SUPPORTED_VERSIONS = {
   chrome: "118.0.5943.0",
@@ -41,28 +47,55 @@ function getCache(): Record<string, string> {
   }
 }
 
+/**
+ * Clean cache
+ */
+export async function cleanCache() {
+  if (await exists(BASE_PATH)) {
+    await Deno.remove(BASE_PATH, { recursive: true });
+  }
+}
+
+async function isQuietInstall() {
+  const { state } = await Deno.permissions.query({
+    name: "env",
+    variable: "ASTRAL_QUIET_INSTALL",
+  });
+  if (state === "granted") {
+    return `${Deno.env.get("ASTRAL_QUIET_INSTALL")}` === "1";
+  }
+}
+
 async function decompressArchive(source: string, destination: string) {
-  const unzipCommand = new Deno.Command(
-    Deno.build.os === "windows" ? "PowerShell" : "unzip",
-    {
-      args: Deno.build.os === "windows"
-        ? [
-          "Expand-Archive",
-          "-Path",
-          `"${source}"`,
-          "-DestinationPath",
-          `"${destination}"`,
-          "-Force",
-        ]
-        : [
-          "-o",
-          source,
-          "-d",
-          destination,
-        ],
-    },
-  );
-  await unzipCommand.output();
+  const archive = await Deno.open(source);
+  const zip = new ZipReader(archive);
+  const entries = await zip.getEntries();
+  const bar = new ProgressBar({
+    title: `Inflating ${destination}`,
+    total: entries.length,
+    clear: true,
+    display: ":title :bar :percent",
+  });
+  let progress = 0;
+  for (const entry of entries) {
+    if ((!entry.directory) && (entry.getData)) {
+      const path = join(destination, entry.filename);
+      await ensureDir(dirname(path));
+      const file = await Deno.open(path, {
+        create: true,
+        truncate: true,
+        write: true,
+        mode: 0o755,
+      });
+      await entry.getData(file, { checkSignature: true, useWebWorkers: false });
+    }
+    progress++;
+    bar.render(progress);
+  }
+  await zip.close();
+  if (!await isQuietInstall()) {
+    console.log(`Browser saved to ${destination}`);
+  }
 }
 
 /**
@@ -98,17 +131,38 @@ export async function getBinary(
       );
     })[0];
 
-    console.log(
-      "Downloading browser... (this may take a while depending on your internet connection)",
-    );
     const req = await fetch(download.url);
     if (!req.body) {
       throw new Error(
         "Download failed, please check your internet connection and try again",
       );
     }
-    await Deno.writeFile(resolve(BASE_PATH, `raw_${VERSION}.zip`), req.body);
-    console.log(`Download complete (${browser} version ${VERSION})`);
+    const reader = req.body.getReader();
+    const archive = await Deno.open(resolve(BASE_PATH, `raw_${VERSION}.zip`), {
+      write: true,
+      truncate: true,
+      create: true,
+    });
+    const bar = new ProgressBar({
+      title: `Downloading ${browser} ${VERSION}`,
+      total: Number(req.headers.get("Content-Length") ?? 0),
+      clear: true,
+      display: ":title :bar :percent",
+    });
+    let downloaded = 0;
+    do {
+      const { done, value } = await reader.read();
+      if (done) {
+        break;
+      }
+      await Deno.write(archive.rid, value);
+      downloaded += value.length;
+      bar.render(downloaded);
+    } while (true);
+    Deno.close(archive.rid);
+    if (!await isQuietInstall()) {
+      console.log(`Download complete (${browser} version ${VERSION})`);
+    }
     await decompressArchive(
       resolve(BASE_PATH, `raw_${VERSION}.zip`),
       resolve(BASE_PATH, VERSION),

tests/_get_binary_test.ts

@@ -0,0 +1,30 @@
+import { assertMatch } from "https://deno.land/[email protected]/assert/assert_match.ts";
+import { cleanCache, getBinary, launch } from "../mod.ts";
+import { assert } from "https://deno.land/[email protected]/assert/assert.ts";
+
+Deno.test("Test download", async () => {
+  // Download browser
+  await cleanCache();
+  const path = await getBinary("chrome");
+
+  // Ensure browser is executable
+  // Note: it seems that on Windows the --version flag does not exists and spawn a
+  //   browser instance instead. The next test ensure that everything is working
+  //   properly anyways
+  if (Deno.build.os !== "windows") {
+    const command = new Deno.Command(path, {
+      args: [
+        "--version",
+      ],
+    });
+    const { success, stdout } = await command.output();
+    assert(success);
+    assertMatch(new TextDecoder().decode(stdout), /Google Chrome/i);
+  }
+
+  // Ensure browser is capable of loading pages
+  const browser = await launch();
+  const page = await browser.newPage("http://example.com");
+  await page.waitForSelector("h1");
+  await browser.close();
+});