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))