parent
2070e1a96b
commit
ae20a951be
|
@ -7,7 +7,8 @@
|
|||
import Operation from "../Operation";
|
||||
import OperationError from "../errors/OperationError";
|
||||
import qr from "qr-image";
|
||||
|
||||
import { toBase64 } from "../lib/Base64";
|
||||
import Magic from "../lib/Magic";
|
||||
|
||||
/**
|
||||
* Generate QR Code operation
|
||||
|
@ -21,12 +22,29 @@ class GenerateQRCode extends Operation {
|
|||
super();
|
||||
|
||||
this.name = "Generate QR Code";
|
||||
this.module = "Default";
|
||||
this.module = "QRCode";
|
||||
this.description = "Generates a QR code from text.";
|
||||
this.infoURL = "https://en.wikipedia.org/wiki/QR_code";
|
||||
this.inputType = "string";
|
||||
this.outputType = "byteArray";
|
||||
this.args = [];
|
||||
this.presentType = "html";
|
||||
this.args = [
|
||||
{
|
||||
"name": "Image Format",
|
||||
"type": "option",
|
||||
"value": ["SVG", "PNG"]
|
||||
},
|
||||
{
|
||||
"name": "Size of QR module",
|
||||
"type": "number",
|
||||
"value": 5
|
||||
},
|
||||
{
|
||||
"name": "Margin",
|
||||
"type": "number",
|
||||
"value": 2
|
||||
}
|
||||
];
|
||||
}
|
||||
|
||||
/**
|
||||
|
@ -35,11 +53,50 @@ class GenerateQRCode extends Operation {
|
|||
* @returns {File}
|
||||
*/
|
||||
run(input, args) {
|
||||
const qrImage = new Buffer(qr.imageSync(input, { type : "png" }));
|
||||
// Create new QR image from the input data, and convert it to a buffer
|
||||
const [format, size, margin] = args;
|
||||
const qrImage = qr.imageSync(input, { type: format, size: size, margin: margin });
|
||||
if (qrImage == null) {
|
||||
return [input];
|
||||
throw new OperationError("Error generating QR code.");
|
||||
}
|
||||
if (format === "SVG") {
|
||||
return [...Buffer.from(qrImage)];
|
||||
} else if (format === "PNG") {
|
||||
// Return the QR image buffer as a byte array
|
||||
return [...qrImage];
|
||||
} else {
|
||||
throw new OperationError("Error generating QR code.");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Displays the QR image using HTML for web apps
|
||||
*
|
||||
* @param {byteArray} data
|
||||
* @returns {html}
|
||||
*/
|
||||
present(data, args) {
|
||||
if (!data.length) return "";
|
||||
|
||||
const [format] = args;
|
||||
if (format === "SVG") {
|
||||
let outputData = "";
|
||||
for (let i = 0; i < data.length; i++){
|
||||
outputData += String.fromCharCode(parseInt(data[i]));
|
||||
}
|
||||
return outputData;
|
||||
} else if (format === "PNG") {
|
||||
let dataURI = "data:";
|
||||
const type = Magic.magicFileType(data);
|
||||
if (type && type.mime.indexOf("image") === 0){
|
||||
dataURI += type.mime + ";";
|
||||
} else {
|
||||
throw new OperationError("Invalid file type");
|
||||
}
|
||||
dataURI += "base64," + toBase64(data);
|
||||
|
||||
return "<img src='" + dataURI + "'>";
|
||||
}
|
||||
return [...qrImage];
|
||||
}
|
||||
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue