Updated an progress bar for the downloading image in js because why not? and also made a skip condition if the image already exists.

mkbsd.js

@@ -1,62 +1,95 @@
 // Copyright 2024 Nadim Kobeissi
 // Licensed under the WTFPL License
 
-const fs = require(`fs`);
-const path = require(`path`);
+const fs = require('fs');
+const path = require('path');
+const https = require('https');
 
 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');
+        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);
+
+                if (fs.existsSync(filePath)) {
+                    console.info(`⏩ Skipping ${filePath} as it already exists.`);
+                } else {
+                    await downloadImage(imageUrl, filePath);
+                    console.info(`🖼️ Saved image to ${filePath}`);
+                    fileIndex++;
+                    await delay(250);
+                }
+            }
+        }
+    } catch (error) {
+        console.error(`Error: ${error.message}`);
+    }
+}
+
+function downloadImage(url, filePath) {
+    return new Promise((resolve, reject) => {
+        https.get(url, response => {
+            if (response.statusCode !== 200) {
+                reject(new Error(`Failed to download image: ${response.statusMessage}`));
+                return;
+            }
+            const total = Number(response.headers['content-length']);
+            let received = 0;
+
+            const fileStream = fs.createWriteStream(filePath);
+            response.pipe(fileStream);
+
+            response.on('data', chunk => {
+                received += chunk.length;
+                const progress = (received / total) * 100;
+                const progressBar = generateProgressBar(progress);
+                process.stdout.write(`Downloading ${filePath}: ${progressBar} ${progress.toFixed(2)}%\r`);
+            });
+
+            fileStream.on('finish', () => {
+                process.stdout.write('\n'); // move to the next line after download completes
+                resolve();
+            });
+
+            fileStream.on('error', reject);
+        }).on('error', reject);
+    });
 }
 
-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);
+function generateProgressBar(progress) {
+    const totalBars = 20; // Length of the progress bar
+    const filledBars = Math.round((progress / 100) * totalBars);
+    const emptyBars = totalBars - filledBars;
+    return `[${'='.repeat(filledBars)}${' '.repeat(emptyBars)}]`;
 }
 
 function asciiArt() {
-	console.info(`
+    console.info(`
  /$$      /$$ /$$   /$$ /$$$$$$$   /$$$$$$  /$$$$$$$
 | $$$    /$$$| $$  /$$/| $$__  $$ /$$__  $$| $$__  $$
 | $$$$  /$$$$| $$ /$$/ | $$  \\ $$| $$  \\__/| $$  \\ $$
@@ -65,11 +98,11 @@ 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);
 })();