mirror of
https://github.com/gchq/CyberChef.git
synced 2024-11-16 08:58:30 +01:00
Add rotate image operation
This commit is contained in:
parent
8148c1a8a8
commit
91f4681a3c
1 changed files with 96 additions and 0 deletions
96
src/core/operations/RotateImage.mjs
Normal file
96
src/core/operations/RotateImage.mjs
Normal file
|
@ -0,0 +1,96 @@
|
||||||
|
/**
|
||||||
|
* @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 { toBase64 } from "../lib/Base64";
|
||||||
|
import jimp from "jimp";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Rotate Image operation
|
||||||
|
*/
|
||||||
|
class RotateImage extends Operation {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* RotateImage constructor
|
||||||
|
*/
|
||||||
|
constructor() {
|
||||||
|
super();
|
||||||
|
|
||||||
|
this.name = "Rotate Image";
|
||||||
|
this.module = "Image";
|
||||||
|
this.description = "Rotates an image by the specified number of degrees.";
|
||||||
|
this.infoURL = "";
|
||||||
|
this.inputType = "byteArray";
|
||||||
|
this.outputType = "byteArray";
|
||||||
|
this.presentType = "html";
|
||||||
|
this.args = [
|
||||||
|
{
|
||||||
|
"name": "Rotation amount (degrees)",
|
||||||
|
"type": "number",
|
||||||
|
"value": 90
|
||||||
|
}
|
||||||
|
];
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @param {byteArray} input
|
||||||
|
* @param {Object[]} args
|
||||||
|
* @returns {byteArray}
|
||||||
|
*/
|
||||||
|
run(input, args) {
|
||||||
|
const [degrees] = args;
|
||||||
|
const type = Magic.magicFileType(input);
|
||||||
|
|
||||||
|
if (type && type.mime.indexOf("image") === 0){
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
jimp.read(Buffer.from(input))
|
||||||
|
.then(image => {
|
||||||
|
image
|
||||||
|
.rotate(degrees / 100)
|
||||||
|
.getBuffer(jimp.AUTO, (error, result) => {
|
||||||
|
if (error){
|
||||||
|
reject(new OperationError("Error getting the new image buffer"));
|
||||||
|
} else {
|
||||||
|
resolve([...result]);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
})
|
||||||
|
.catch(err => {
|
||||||
|
reject(new OperationError("Error reading the input image."));
|
||||||
|
});
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
throw new OperationError("Invalid file type.");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Displays the rotated image using HTML for web apps
|
||||||
|
*
|
||||||
|
* @param {byteArray} data
|
||||||
|
* @returns {html}
|
||||||
|
*/
|
||||||
|
present(data) {
|
||||||
|
if (!data.length) return "";
|
||||||
|
|
||||||
|
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 + "'>";
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
export default RotateImage;
|
Loading…
Reference in a new issue