From 9034ea124f9be52b57292d3ba78d00522d4c63dd Mon Sep 17 00:00:00 2001
From: Vinay Viswanathan <v.vinay@wustl.edu>
Date: Wed, 25 Sep 2024 23:52:18 -0700
Subject: [PATCH] Added js progress bar | logs save to file

---
 .gitignore |   1 +
 mkbsd.js   | 147 +++++++++++++++++++++++++++++++++++------------------
 2 files changed, 98 insertions(+), 50 deletions(-)

diff --git a/.gitignore b/.gitignore
index de2b2c6..4ee20d7 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,3 @@
 .DS_Store
 downloads
+logs
diff --git a/mkbsd.js b/mkbsd.js
index d02593a..8ddbb07 100644
--- a/mkbsd.js
+++ b/mkbsd.js
@@ -3,60 +3,105 @@
 
 const fs = require(`fs`);
 const path = require(`path`);
+const cliProgress = require(`cli-progress`);
+const colors = require("ansi-colors");
+
+async function getTotalPics(data) {
+  let totalPics = 0;
+  for (const key in data) {
+    const subproperty = data[key];
+    if (subproperty && subproperty.dhd) {
+      totalPics++;
+    }
+  }
+  return totalPics;
+}
 
 async function main() {
-	const url = 'https://storage.googleapis.com/panels-api/data/20240916/media-1a-i-p~s';
-	const delay = (ms) => {
-		return new Promise(resolve => setTimeout(resolve, ms));
-	}
-	try {
-		const response = await fetch(url);
-		if (!response.ok) {
-			throw new Error(`⛔ Failed to fetch JSON file: ${response.statusText}`);
-		}
-		const jsonData = await response.json();
-		const data = jsonData.data;
-		if (!data) {
-			throw new Error('⛔ JSON does not have a "data" property at its root.');
-		}
-		const downloadDir = path.join(__dirname, 'downloads');
-		if (!fs.existsSync(downloadDir)) {
-			fs.mkdirSync(downloadDir);
-			console.info(`📁 Created directory: ${downloadDir}`);
-		}
-		let fileIndex = 1;
-		for (const key in data) {
-			const subproperty = data[key];
-			if (subproperty && subproperty.dhd) {
-				const imageUrl = subproperty.dhd;
-				console.info(`🔍 Found image URL!`);
-				await delay(100);
-				const ext = path.extname(new URL(imageUrl).pathname) || '.jpg';
-				const filename = `${fileIndex}${ext}`;
-				const filePath = path.join(downloadDir, filename);
-				await downloadImage(imageUrl, filePath);
-				console.info(`🖼️ Saved image to ${filePath}`);
-				fileIndex++;
-				await delay(250);
-			}
-		}
-	} catch (error) {
-		console.error(`Error: ${error.message}`);
-	}
+  const url =
+    "https://storage.googleapis.com/panels-api/data/20240916/media-1a-i-p~s";
+  const delay = (ms) => {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+  };
+  try {
+    const response = await fetch(url);
+    if (!response.ok) {
+      throw new Error(`⛔ Failed to fetch JSON file: ${response.statusText}`);
+    }
+    const jsonData = await response.json();
+    const data = jsonData.data;
+    if (!data) {
+      throw new Error('⛔ JSON does not have a "data" property at its root.');
+    }
+    const downloadDir = path.join(__dirname, "downloads");
+    const logsDir = path.join(__dirname, "logs");
+    if (!fs.existsSync(downloadDir)) {
+      fs.mkdirSync(downloadDir);
+      console.info(`📁 Created directory: ${downloadDir}`);
+    }
+    if (!fs.existsSync(logsDir)) {
+      fs.mkdirSync(logsDir);
+      console.info(`📁 Created logs directory: ${logsDir}`);
+    }
+    const logsFilePath = path.join(logsDir, `logs.txt`);
+    let fileIndex = 1;
+    let totalPictures = await getTotalPics(data);
+    const bar = new cliProgress.SingleBar(
+      {
+        format:
+          "Picture Progress Download |" +
+          colors.cyan("{bar}") +
+          "| {percentage}% || {value}/{total} Pictures",
+        barCompleteChar: "\u2588",
+        barIncompleteChar: "\u2591",
+        hideCursor: true,
+      },
+      cliProgress.Presets.shades_classic
+    );
+    bar.start(totalPictures, 0, {
+      speed: "N/A",
+    });
+    for (const key in data) {
+      const subproperty = data[key];
+      if (subproperty && subproperty.dhd) {
+        const imageUrl = subproperty.dhd;
+        writeLogs(`🔍 Found image URL!`, logsFilePath);
+        await delay(100);
+        const ext = path.extname(new URL(imageUrl).pathname) || ".jpg";
+        const filename = `${fileIndex}${ext}`;
+        const filePath = path.join(downloadDir, filename);
+        await downloadImage(imageUrl, filePath);
+        writeLogs(`🖼️ Saved image to ${filePath}`, logsFilePath);
+        fileIndex++;
+        bar.increment();
+        bar.update();
+        await delay(250);
+      }
+    }
+    bar.stop();
+    console.info(`✅ All done! Saved images to ${downloadDir}`);
+  } catch (error) {
+    console.error(`Error: ${error.message}`);
+  }
+}
+
+async function writeLogs(log, logsFilePath) {
+  const buffer = Buffer.from(log + "\n");
+  await fs.promises.appendFile(logsFilePath, buffer);
 }
 
 async function downloadImage(url, filePath) {
-	const response = await fetch(url);
-	if (!response.ok) {
-		throw new Error(`Failed to download image: ${response.statusText}`);
-	}
-	const arrayBuffer = await response.arrayBuffer();
-	const buffer = Buffer.from(arrayBuffer);
-	await fs.promises.writeFile(filePath, buffer);
+  const response = await fetch(url);
+  if (!response.ok) {
+    throw new Error(`Failed to download image: ${response.statusText}`);
+  }
+  const arrayBuffer = await response.arrayBuffer();
+  const buffer = Buffer.from(arrayBuffer);
+  await fs.promises.writeFile(filePath, buffer);
 }
 
 function asciiArt() {
-	console.info(`
+  console.info(`
  /$$      /$$ /$$   /$$ /$$$$$$$   /$$$$$$  /$$$$$$$
 | $$$    /$$$| $$  /$$/| $$__  $$ /$$__  $$| $$__  $$
 | $$$$  /$$$$| $$ /$$/ | $$  \\ $$| $$  \\__/| $$  \\ $$
@@ -65,11 +110,13 @@ function asciiArt() {
 | $$\\  $ | $$| $$\\  $$ | $$  \\ $$ /$$  \\ $$| $$  | $$
 | $$ \\/  | $$| $$ \\  $$| $$$$$$$/|  $$$$$$/| $$$$$$$/
 |__/     |__/|__/  \\__/|_______/  \\______/ |_______/`);
-	console.info(``);
-	console.info(`🤑 Starting downloads from your favorite sellout grifter's wallpaper app...`);
+  console.info(``);
+  console.info(
+    `🤑 Starting downloads from your favorite sellout grifter's wallpaper app...`
+  );
 }
 
 (() => {
-	asciiArt();
-	setTimeout(main, 5000);
+  asciiArt();
+  setTimeout(main, 5000);
 })();