You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
928 B
35 lines
928 B
const util = require('util'); |
|
const fs = require('fs'); |
|
const fetch = require('node-fetch'); |
|
const isURL = require('is-url'); |
|
|
|
const readFile = util.promisify(fs.readFile); |
|
|
|
/** |
|
* loadImage |
|
* |
|
* @name loadImage |
|
* @function load image from different source |
|
* @access public |
|
*/ |
|
module.exports = async (image) => { |
|
let data = image; |
|
if (typeof image === 'undefined') { |
|
return image; |
|
} |
|
|
|
if (typeof image === 'string') { |
|
if (isURL(image) || image.startsWith('moz-extension://') || image.startsWith('chrome-extension://') || image.startsWith('file://')) { |
|
const resp = await fetch(image); |
|
data = await resp.arrayBuffer(); |
|
} else if (/data:image\/([a-zA-Z]*);base64,([^"]*)/.test(image)) { |
|
data = Buffer.from(image.split(',')[1], 'base64'); |
|
} else { |
|
data = await readFile(image); |
|
} |
|
} else if (Buffer.isBuffer(image)) { |
|
data = image; |
|
} |
|
|
|
return new Uint8Array(data); |
|
};
|
|
|