mirror of
https://github.com/gchq/CyberChef.git
synced 2024-09-19 18:11:29 +02:00
146 lines
4.0 KiB
JavaScript
146 lines
4.0 KiB
JavaScript
/**
|
|
* @author j433866 [j433866@gmail.com]
|
|
* @copyright Crown Copyright 2019
|
|
* @license Apache-2.0
|
|
*/
|
|
|
|
import Operation from "../Operation.mjs";
|
|
import OperationError from "../errors/OperationError.mjs";
|
|
import { isImage } from "../lib/FileType.mjs";
|
|
import { toBase64 } from "../lib/Base64.mjs";
|
|
import { isWorkerEnvironment } from "../Utils.mjs";
|
|
import jimp from "jimp";
|
|
|
|
/**
|
|
* Resize Image operation
|
|
*/
|
|
class ResizeImage extends Operation {
|
|
|
|
/**
|
|
* ResizeImage constructor
|
|
*/
|
|
constructor() {
|
|
super();
|
|
|
|
this.name = "Resize Image";
|
|
this.module = "Image";
|
|
this.description = "Resizes an image to the specified width and height values.";
|
|
this.infoURL = "https://wikipedia.org/wiki/Image_scaling";
|
|
this.inputType = "ArrayBuffer";
|
|
this.outputType = "ArrayBuffer";
|
|
this.presentType = "html";
|
|
this.args = [
|
|
{
|
|
name: "Width",
|
|
type: "number",
|
|
value: 100,
|
|
min: 1
|
|
},
|
|
{
|
|
name: "Height",
|
|
type: "number",
|
|
value: 100,
|
|
min: 1
|
|
},
|
|
{
|
|
name: "Unit type",
|
|
type: "option",
|
|
value: ["Pixels", "Percent"]
|
|
},
|
|
{
|
|
name: "Maintain aspect ratio",
|
|
type: "boolean",
|
|
value: false
|
|
},
|
|
{
|
|
name: "Resizing algorithm",
|
|
type: "option",
|
|
value: [
|
|
"Nearest Neighbour",
|
|
"Bilinear",
|
|
"Bicubic",
|
|
"Hermite",
|
|
"Bezier"
|
|
],
|
|
defaultIndex: 1
|
|
}
|
|
];
|
|
}
|
|
|
|
/**
|
|
* @param {ArrayBuffer} input
|
|
* @param {Object[]} args
|
|
* @returns {byteArray}
|
|
*/
|
|
async run(input, args) {
|
|
let width = args[0],
|
|
height = args[1];
|
|
const unit = args[2],
|
|
aspect = args[3],
|
|
resizeAlg = args[4];
|
|
|
|
const resizeMap = {
|
|
"Nearest Neighbour": jimp.RESIZE_NEAREST_NEIGHBOR,
|
|
"Bilinear": jimp.RESIZE_BILINEAR,
|
|
"Bicubic": jimp.RESIZE_BICUBIC,
|
|
"Hermite": jimp.RESIZE_HERMITE,
|
|
"Bezier": jimp.RESIZE_BEZIER
|
|
};
|
|
|
|
if (!isImage(input)) {
|
|
throw new OperationError("Invalid file type.");
|
|
}
|
|
|
|
let image;
|
|
try {
|
|
image = await jimp.read(input);
|
|
} catch (err) {
|
|
throw new OperationError(`Error loading image. (${err})`);
|
|
}
|
|
try {
|
|
if (unit === "Percent") {
|
|
width = image.getWidth() * (width / 100);
|
|
height = image.getHeight() * (height / 100);
|
|
}
|
|
|
|
if (isWorkerEnvironment())
|
|
self.sendStatusMessage("Resizing image...");
|
|
if (aspect) {
|
|
image.scaleToFit(width, height, resizeMap[resizeAlg]);
|
|
} else {
|
|
image.resize(width, height, resizeMap[resizeAlg]);
|
|
}
|
|
|
|
let imageBuffer;
|
|
if (image.getMIME() === "image/gif") {
|
|
imageBuffer = await image.getBufferAsync(jimp.MIME_PNG);
|
|
} else {
|
|
imageBuffer = await image.getBufferAsync(jimp.AUTO);
|
|
}
|
|
return imageBuffer.buffer;
|
|
} catch (err) {
|
|
throw new OperationError(`Error resizing image. (${err})`);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Displays the resized image using HTML for web apps
|
|
* @param {ArrayBuffer} data
|
|
* @returns {html}
|
|
*/
|
|
present(data) {
|
|
if (!data.byteLength) return "";
|
|
const dataArray = new Uint8Array(data);
|
|
|
|
const type = isImage(dataArray);
|
|
if (!type) {
|
|
throw new OperationError("Invalid file type.");
|
|
}
|
|
|
|
return `<img src="data:${type};base64,${toBase64(dataArray)}">`;
|
|
}
|
|
|
|
}
|
|
|
|
export default ResizeImage;
|