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

View file

@ -1,8 +1,9 @@
// 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';
@ -34,25 +35,57 @@ async function main() {
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);
if (fs.existsSync(filePath)) {
console.info(`⏩ Skipping ${filePath} as it already exists.`);
} else {
await downloadImage(imageUrl, filePath); await downloadImage(imageUrl, filePath);
console.info(`🖼️ Saved image to ${filePath}`); console.info(`🖼️ Saved image to ${filePath}`);
fileIndex++; fileIndex++;
await delay(250); await delay(250);
} }
} }
}
} catch (error) { } catch (error) {
console.error(`Error: ${error.message}`); 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}`));
return;
} }
const arrayBuffer = await response.arrayBuffer(); const total = Number(response.headers['content-length']);
const buffer = Buffer.from(arrayBuffer); let received = 0;
await fs.promises.writeFile(filePath, buffer);
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() {