diff --git a/index.js b/index.js index e0b78fb..9c06b83 100644 --- a/index.js +++ b/index.js @@ -89,27 +89,39 @@ async function run(url, processor) { console.log('Done collecting URLS', galleryUrls.length); const payloads = await Promise.all(galleryUrls.map(image => new Promise(async (resolve, reject) => { - superagent.get(image.url).responseType('blob').then(function (response) { - if (response.statusCode == 200) { - console.log('Resolving', image.url) - return resolve({ - response: { - content: { - mimeType: response.headers["content-type"], - encoding: 'base64', - text: response.body.toString('base64') + if(image.url) { + superagent.get(image.url).responseType('blob').then(function (response) { + if (response.statusCode == 200) { + console.log('Resolving', image.url) + return resolve({ + response: { + content: { + mimeType: response.headers["content-type"], + encoding: 'base64', + text: response.body.toString('base64') + } } - } - }); - } else { - console.log("Invalid status code", response.statusCode, 'for', image.url); - resolve({}) + }); + } else { + console.log("Invalid status code", response.statusCode, 'for', image.url); + resolve({}) - } - }).catch(error => { - console.error(error) - resolve({}) - }); + } + }).catch(error => { + console.error(error) + resolve({}) + }); + } else if (image.base64){ + return resolve({ + response: { + content: { + mimeType: response.headers["content-type"], + encoding: 'base64', + text: image.base64 + } + } + }); + } }))) console.log('URLS done downloading') await browser.close();