Updated an progress bar for the downloading image in js because why not?

This commit is contained in:
arush75 2024-09-30 18:41:18 +05:30
parent 82e50c64f0
commit 680fdd7b92

139
mkbsd.js
View File

@ -1,62 +1,95 @@
// Copyright 2024 Nadim Kobeissi // Copyright 2024 Nadim Kobeissi
// Licensed under the WTFPL License // Licensed under the WTFPL License
const fs = require(`fs`); const fs = require('fs');
const path = require(`path`); const path = require('path');
const https = require('https');
async function main() { async function main() {
const url = 'https://storage.googleapis.com/panels-api/data/20240916/media-1a-i-p~s'; const url = 'https://storage.googleapis.com/panels-api/data/20240916/media-1a-i-p~s';
const delay = (ms) => { const delay = (ms) => {
return new Promise(resolve => setTimeout(resolve, ms)); return new Promise(resolve => setTimeout(resolve, ms));
} }
try { try {
const response = await fetch(url); const response = await fetch(url);
if (!response.ok) { if (!response.ok) {
throw new Error(`⛔ Failed to fetch JSON file: ${response.statusText}`); throw new Error(`⛔ Failed to fetch JSON file: ${response.statusText}`);
} }
const jsonData = await response.json(); const jsonData = await response.json();
const data = jsonData.data; const data = jsonData.data;
if (!data) { if (!data) {
throw new Error('⛔ JSON does not have a "data" property at its root.'); throw new Error('⛔ JSON does not have a "data" property at its root.');
} }
const downloadDir = path.join(__dirname, 'downloads'); const downloadDir = path.join(__dirname, 'downloads');
if (!fs.existsSync(downloadDir)) { if (!fs.existsSync(downloadDir)) {
fs.mkdirSync(downloadDir); fs.mkdirSync(downloadDir);
console.info(`📁 Created directory: ${downloadDir}`); console.info(`📁 Created directory: ${downloadDir}`);
} }
let fileIndex = 1; let fileIndex = 1;
for (const key in data) { for (const key in data) {
const subproperty = data[key]; const subproperty = data[key];
if (subproperty && subproperty.dhd) { if (subproperty && subproperty.dhd) {
const imageUrl = subproperty.dhd; const imageUrl = subproperty.dhd;
console.info(`🔍 Found image URL!`); console.info(`🔍 Found image URL!`);
await delay(100); await delay(100);
const ext = path.extname(new URL(imageUrl).pathname) || '.jpg'; const ext = path.extname(new URL(imageUrl).pathname) || '.jpg';
const filename = `${fileIndex}${ext}`; const filename = `${fileIndex}${ext}`;
const filePath = path.join(downloadDir, filename); const filePath = path.join(downloadDir, filename);
await downloadImage(imageUrl, filePath);
console.info(`🖼️ Saved image to ${filePath}`); if (fs.existsSync(filePath)) {
fileIndex++; console.info(`⏩ Skipping ${filePath} as it already exists.`);
await delay(250); } else {
} await downloadImage(imageUrl, filePath);
} console.info(`🖼️ Saved image to ${filePath}`);
} catch (error) { fileIndex++;
console.error(`Error: ${error.message}`); await delay(250);
} }
}
}
} catch (error) {
console.error(`Error: ${error.message}`);
}
} }
async function downloadImage(url, filePath) { function downloadImage(url, filePath) {
const response = await fetch(url); return new Promise((resolve, reject) => {
if (!response.ok) { https.get(url, response => {
throw new Error(`Failed to download image: ${response.statusText}`); if (response.statusCode !== 200) {
} reject(new Error(`Failed to download image: ${response.statusMessage}`));
const arrayBuffer = await response.arrayBuffer(); return;
const buffer = Buffer.from(arrayBuffer); }
await fs.promises.writeFile(filePath, buffer); 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);
});
}
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() { function asciiArt() {
console.info(` console.info(`
/$$ /$$ /$$ /$$ /$$$$$$$ /$$$$$$ /$$$$$$$ /$$ /$$ /$$ /$$ /$$$$$$$ /$$$$$$ /$$$$$$$
| $$$ /$$$| $$ /$$/| $$__ $$ /$$__ $$| $$__ $$ | $$$ /$$$| $$ /$$/| $$__ $$ /$$__ $$| $$__ $$
| $$$$ /$$$$| $$ /$$/ | $$ \\ $$| $$ \\__/| $$ \\ $$ | $$$$ /$$$$| $$ /$$/ | $$ \\ $$| $$ \\__/| $$ \\ $$
@ -65,11 +98,11 @@ function asciiArt() {
| $$\\ $ | $$| $$\\ $$ | $$ \\ $$ /$$ \\ $$| $$ | $$ | $$\\ $ | $$| $$\\ $$ | $$ \\ $$ /$$ \\ $$| $$ | $$
| $$ \\/ | $$| $$ \\ $$| $$$$$$$/| $$$$$$/| $$$$$$$/ | $$ \\/ | $$| $$ \\ $$| $$$$$$$/| $$$$$$/| $$$$$$$/
|__/ |__/|__/ \\__/|_______/ \\______/ |_______/`); |__/ |__/|__/ \\__/|_______/ \\______/ |_______/`);
console.info(``); console.info(``);
console.info(`🤑 Starting downloads from your favorite sellout grifter's wallpaper app...`); console.info(`🤑 Starting downloads from your favorite sellout grifter's wallpaper app...`);
} }
(() => { (() => {
asciiArt(); asciiArt();
setTimeout(main, 5000); setTimeout(main, 5000);
})(); })();