Added js progress bar | logs save to file

This commit is contained in:
Vinay Viswanathan 2024-09-25 23:52:18 -07:00
parent 82e50c64f0
commit 9034ea124f
2 changed files with 98 additions and 50 deletions

1
.gitignore vendored
View File

@ -1,2 +1,3 @@
.DS_Store .DS_Store
downloads downloads
logs

147
mkbsd.js
View File

@ -3,60 +3,105 @@
const fs = require(`fs`); const fs = require(`fs`);
const path = require(`path`); const path = require(`path`);
const cliProgress = require(`cli-progress`);
const colors = require("ansi-colors");
async function getTotalPics(data) {
let totalPics = 0;
for (const key in data) {
const subproperty = data[key];
if (subproperty && subproperty.dhd) {
totalPics++;
}
}
return totalPics;
}
async function main() { async function main() {
const url = 'https://storage.googleapis.com/panels-api/data/20240916/media-1a-i-p~s'; const url =
const delay = (ms) => { "https://storage.googleapis.com/panels-api/data/20240916/media-1a-i-p~s";
return new Promise(resolve => setTimeout(resolve, ms)); const delay = (ms) => {
} return new Promise((resolve) => setTimeout(resolve, ms));
try { };
const response = await fetch(url); try {
if (!response.ok) { const response = await fetch(url);
throw new Error(`⛔ Failed to fetch JSON file: ${response.statusText}`); if (!response.ok) {
} throw new Error(`⛔ Failed to fetch JSON file: ${response.statusText}`);
const jsonData = await response.json(); }
const data = jsonData.data; const jsonData = await response.json();
if (!data) { const data = jsonData.data;
throw new Error('⛔ JSON does not have a "data" property at its root.'); 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)) { const downloadDir = path.join(__dirname, "downloads");
fs.mkdirSync(downloadDir); const logsDir = path.join(__dirname, "logs");
console.info(`📁 Created directory: ${downloadDir}`); if (!fs.existsSync(downloadDir)) {
} fs.mkdirSync(downloadDir);
let fileIndex = 1; console.info(`📁 Created directory: ${downloadDir}`);
for (const key in data) { }
const subproperty = data[key]; if (!fs.existsSync(logsDir)) {
if (subproperty && subproperty.dhd) { fs.mkdirSync(logsDir);
const imageUrl = subproperty.dhd; console.info(`📁 Created logs directory: ${logsDir}`);
console.info(`🔍 Found image URL!`); }
await delay(100); const logsFilePath = path.join(logsDir, `logs.txt`);
const ext = path.extname(new URL(imageUrl).pathname) || '.jpg'; let fileIndex = 1;
const filename = `${fileIndex}${ext}`; let totalPictures = await getTotalPics(data);
const filePath = path.join(downloadDir, filename); const bar = new cliProgress.SingleBar(
await downloadImage(imageUrl, filePath); {
console.info(`🖼️ Saved image to ${filePath}`); format:
fileIndex++; "Picture Progress Download |" +
await delay(250); colors.cyan("{bar}") +
} "| {percentage}% || {value}/{total} Pictures",
} barCompleteChar: "\u2588",
} catch (error) { barIncompleteChar: "\u2591",
console.error(`Error: ${error.message}`); hideCursor: true,
} },
cliProgress.Presets.shades_classic
);
bar.start(totalPictures, 0, {
speed: "N/A",
});
for (const key in data) {
const subproperty = data[key];
if (subproperty && subproperty.dhd) {
const imageUrl = subproperty.dhd;
writeLogs(`🔍 Found image URL!`, logsFilePath);
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);
writeLogs(`🖼️ Saved image to ${filePath}`, logsFilePath);
fileIndex++;
bar.increment();
bar.update();
await delay(250);
}
}
bar.stop();
console.info(`✅ All done! Saved images to ${downloadDir}`);
} catch (error) {
console.error(`Error: ${error.message}`);
}
}
async function writeLogs(log, logsFilePath) {
const buffer = Buffer.from(log + "\n");
await fs.promises.appendFile(logsFilePath, buffer);
} }
async function downloadImage(url, filePath) { async function downloadImage(url, filePath) {
const response = await fetch(url); const response = await fetch(url);
if (!response.ok) { if (!response.ok) {
throw new Error(`Failed to download image: ${response.statusText}`); throw new Error(`Failed to download image: ${response.statusText}`);
} }
const arrayBuffer = await response.arrayBuffer(); const arrayBuffer = await response.arrayBuffer();
const buffer = Buffer.from(arrayBuffer); const buffer = Buffer.from(arrayBuffer);
await fs.promises.writeFile(filePath, buffer); await fs.promises.writeFile(filePath, buffer);
} }
function asciiArt() { function asciiArt() {
console.info(` console.info(`
/$$ /$$ /$$ /$$ /$$$$$$$ /$$$$$$ /$$$$$$$ /$$ /$$ /$$ /$$ /$$$$$$$ /$$$$$$ /$$$$$$$
| $$$ /$$$| $$ /$$/| $$__ $$ /$$__ $$| $$__ $$ | $$$ /$$$| $$ /$$/| $$__ $$ /$$__ $$| $$__ $$
| $$$$ /$$$$| $$ /$$/ | $$ \\ $$| $$ \\__/| $$ \\ $$ | $$$$ /$$$$| $$ /$$/ | $$ \\ $$| $$ \\__/| $$ \\ $$
@ -65,11 +110,13 @@ 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);
})(); })();