mirror of
https://github.com/nadimkobeissi/mkbsd.git
synced 2024-12-22 14:26:48 +00:00
Updated an progress bar for the downloading image in js because why not?
This commit is contained in:
parent
82e50c64f0
commit
680fdd7b92
51
mkbsd.js
51
mkbsd.js
@ -1,8 +1,9 @@
|
||||
// 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';
|
||||
@ -34,25 +35,57 @@ async function main() {
|
||||
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}`);
|
||||
}
|
||||
}
|
||||
|
||||
async function downloadImage(url, filePath) {
|
||||
const response = await fetch(url);
|
||||
if (!response.ok) {
|
||||
throw new Error(`Failed to download image: ${response.statusText}`);
|
||||
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 arrayBuffer = await response.arrayBuffer();
|
||||
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() {
|
||||
|
Loading…
Reference in New Issue
Block a user