Add new ParseQRCode operation

master
j433866 2018-12-20 14:45:23 +00:00
parent 1953d9a4c8
commit e5b2b84073
2 changed files with 83 additions and 0 deletions

View File

@ -352,6 +352,7 @@
"Generate TOTP",
"Generate HOTP",
"Generate QR Code",
"Parse QR Code",
"Haversine distance",
"Render Image",
"Remove EXIF",

View File

@ -0,0 +1,82 @@
/**
* @author j433866 [j433866@gmail.com]
* @copyright Crown Copyright 2018
* @license Apache-2.0
*/
import Operation from "../Operation";
import OperationError from "../errors/OperationError";
import Magic from "../lib/Magic";
import jsqr from "jsqr";
import jimp from "jimp";
/**
* Parse QR Code operation
*/
class ParseQRCode extends Operation {
/**
* ParseQRCode constructor
*/
constructor() {
super();
this.name = "Parse QR Code";
this.module = "Image";
this.description = "Reads an image file and attempts to detect and read a QR code from the image.";
this.infoURL = "https://wikipedia.org/wiki/QR_code";
this.inputType = "byteArray";
this.outputType = "string";
this.args = [];
}
/**
* @param {byteArray} input
* @param {Object[]} args
* @returns {string}
*/
async run(input, args) {
const type = Magic.magicFileType(input);
// Make sure that the input is an image
if (type && type.mime.indexOf("image") === 0){
return new Promise((resolve, reject) => {
// Read the input
jimp.read(Buffer.from(input))
.then(image => {
image.rgba(false); // Disable RGBA (uses just RGB)
// Get the buffer of the new image and read it in Jimp
// Don't actually need the new image buffer, just need
// Jimp to refresh the current object
image.getBuffer(image.getMIME(), (err, buffer) => {
jimp.read(buffer)
.then(newImage =>{
// If the image has been read correctly, try to find a QR code
if (image.bitmap != null){
const qrData = jsqr(image.bitmap.data, image.getWidth(), image.getHeight());
if (qrData != null) {
resolve(qrData.data);
} else {
log.error(image.bitmap);
reject(new OperationError("Error parsing QR code from image."));
}
} else {
reject(new OperationError("Error reading the image data."));
}
});
});
})
.catch(err => {
reject(new OperationError("Error opening the image. Are you sure this is an image file?"));
});
});
} else {
throw new OperationError("Invalid file type.");
}
}
}
export default ParseQRCode;