mirror of
https://github.com/nadimkobeissi/mkbsd.git
synced 2024-12-22 18:55:33 +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
|
// 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() {
|
||||||
|
|
Loading…
Reference in a new issue