CyberChef/src/core/operations/BlurImage.mjs

103 lines
2.6 KiB
JavaScript
Raw Normal View History

/**
* @author j433866 [j433866@gmail.com]
* @copyright Crown Copyright 2019
* @license Apache-2.0
*/
import Operation from "../Operation";
import OperationError from "../errors/OperationError";
2019-03-09 08:23:11 +01:00
import { isImage } from "../lib/FileType";
import { toBase64 } from "../lib/Base64";
import jimp from "jimp";
/**
* Blur Image operation
*/
class BlurImage extends Operation {
/**
* BlurImage constructor
*/
constructor() {
super();
this.name = "Blur Image";
this.module = "Image";
this.description = "Applies a blur effect to the image.<br><br>Gaussian blur is much slower than fast blur, but produces better results.";
2019-03-04 14:48:13 +01:00
this.infoURL = "https://wikipedia.org/wiki/Gaussian_blur";
this.inputType = "byteArray";
this.outputType = "byteArray";
this.presentType = "html";
this.args = [
{
2019-03-09 08:23:11 +01:00
name: "Amount",
type: "number",
value: 5,
min: 1
},
{
2019-03-09 08:23:11 +01:00
name: "Type",
type: "option",
value: ["Fast", "Gaussian"]
}
];
}
/**
* @param {byteArray} input
* @param {Object[]} args
* @returns {byteArray}
*/
async run(input, args) {
const [blurAmount, blurType] = args;
2019-03-09 08:23:11 +01:00
if (!isImage(input)) {
throw new OperationError("Invalid file type.");
}
2019-03-09 08:23:11 +01:00
let image;
try {
image = await jimp.read(Buffer.from(input));
} catch (err) {
throw new OperationError(`Error loading image. (${err})`);
}
try {
switch (blurType){
case "Fast":
image.blur(blurAmount);
break;
case "Gaussian":
if (ENVIRONMENT_IS_WORKER())
self.sendStatusMessage("Gaussian blurring image. This may take a while...");
image.gaussian(blurAmount);
break;
2019-03-07 12:19:04 +01:00
}
2019-03-09 08:23:11 +01:00
const imageBuffer = await image.getBufferAsync(jimp.AUTO);
return [...imageBuffer];
} catch (err) {
throw new OperationError(`Error blurring image. (${err})`);
}
}
/**
* Displays the blurred image using HTML for web apps
2019-03-09 08:23:11 +01:00
*
* @param {byteArray} data
* @returns {html}
*/
present(data) {
if (!data.length) return "";
2019-03-09 08:23:11 +01:00
const type = isImage(data);
if (!type) {
throw new OperationError("Invalid file type.");
}
2019-03-09 08:23:11 +01:00
return `<img src="data:${type};base64,${toBase64(data)}">`;
}
}
export default BlurImage;