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.
36 lines
912 B
36 lines
912 B
6 years ago
|
const util = require('util');
|
||
|
|
const fs = require('fs');
|
||
|
|
const axios = require('axios');
|
||
|
|
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('chrome-extension://') || image.startsWith('file://')) {
|
||
|
|
const { data: _data } = await axios.get(image, { responseType: 'arraybuffer' });
|
||
|
|
data = _data;
|
||
|
|
} 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);
|
||
|
|
};
|