Small improvements

.gitignore

@@ -1,2 +1,3 @@
 .DS_Store
 downloads
+.venv

mkbsd.js

@@ -1,62 +1,67 @@
 // 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 readline = require('readline');
 
-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}`);
-	}
+async function main(replaceExisting) {
+    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, replaceExisting);
+                console.info(`🖼️ Saved image to ${filePath}`);
+                fileIndex++;
+                await delay(250);
+            }
+        }
+    } catch (error) {
+        console.error(`Error: ${error.message}`);
+    }
 }
 
-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);
+async function downloadImage(url, filePath, replaceExisting) {
+    if (!replaceExisting && fs.existsSync(filePath)) {
+        console.info(`⚠️ File ${filePath} already exists. Skipping download.`);
+        return;
+    }
+    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 +70,32 @@ 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);
+function promptUser() {
+    return new Promise((resolve) => {
+        const rl = readline.createInterface({
+            input: process.stdin,
+            output: process.stdout
+        });
+        rl.question("Files already exist in the 'downloads' directory.\n1. Skip existing files and download new ones.\n2. Replace existing files and download all.\nEnter your choice (1 or 2): ", (answer) => {
+            rl.close();
+            resolve(answer === '2');
+        });
+    });
+}
+
+(async () => {
+    asciiArt();
+    await new Promise(resolve => setTimeout(resolve, 5000));
+
+    const downloadDir = path.join(__dirname, 'downloads');
+    let replaceExisting = true;
+    if (fs.existsSync(downloadDir) && fs.readdirSync(downloadDir).length > 0) {
+        replaceExisting = await promptUser();
+    }
+
+    await main(replaceExisting);
 })();

mkbsd.py

@@ -5,13 +5,18 @@
 import aiohttp
 import asyncio
 from urllib.parse import urlparse
+
 url = 'https://storage.googleapis.com/panels-api/data/20240916/media-1a-i-p~s'
 
 async def delay(ms):
     await asyncio.sleep(ms / 1000)
 
-async def download_image(session, image_url, file_path):
+async def download_image(session, image_url, file_path, replace_existing):
     try:
+        if not replace_existing and os.path.exists(file_path):
+            print(f"⚠️ File {file_path} already exists. Skipping download.")
+            return
+
         async with session.get(image_url) as response:
             if response.status != 200:
                 raise Exception(f"Failed to download image: {response.status}")
@@ -21,7 +26,7 @@ async def download_image(session, image_url, file_path):
     except Exception as e:
         print(f"Error downloading image: {str(e)}")
 
-async def main():
+async def main(replace_existing):
     try:
         async with aiohttp.ClientSession() as session:
             async with session.get(url) as response:
@@ -48,7 +53,7 @@ async def main():
                         filename = f"{file_index}{ext}"
                         file_path = os.path.join(download_dir, filename)
 
-                        await download_image(session, image_url, file_path)
+                        await download_image(session, image_url, file_path, replace_existing)
                         print(f"🖼️ Saved image to {file_path}")
 
                         file_index += 1
@@ -70,7 +75,24 @@ def ascii_art():
     print("")
     print("🤑 Starting downloads from your favorite sellout grifter's wallpaper app...")
 
+def prompt_user():
+    while True:
+        print("Files already exist in the 'downloads' directory.")
+        print("1. Skip existing files and download new ones.")
+        print("2. Replace existing files and download all.")
+        choice = input("Enter your choice (1 or 2): ")
+        if choice in ['1', '2']:
+            return choice == '2'
+        print("Invalid choice. Please enter 1 or 2.")
+
 if __name__ == "__main__":
     ascii_art()
     time.sleep(5)
-    asyncio.run(main())
+
+    download_dir = os.path.join(os.getcwd(), 'downloads')
+    if os.path.exists(download_dir) and os.listdir(download_dir):
+        replace_existing = prompt_user()
+    else:
+        replace_existing = True
+
+    asyncio.run(main(replace_existing))