76 lines
2.5 KiB
JavaScript
76 lines
2.5 KiB
JavaScript
import { createRequire } from "module";
|
|
import { isMainThread, parentPort, workerData } from "worker_threads";
|
|
import { request } from "undici";
|
|
import path from "path";
|
|
import { fileURLToPath } from "url";
|
|
|
|
const nodeRequire = createRequire(import.meta.url);
|
|
|
|
const relPath = `../build/${process.env.DEBUG && process.env.DEBUG === "true" ? "Debug" : "Release"}/image.node`;
|
|
const img = nodeRequire(relPath);
|
|
|
|
const enumMap = {
|
|
"forget": 0,
|
|
"northwest": 1,
|
|
"north": 2,
|
|
"northeast": 3,
|
|
"west": 4,
|
|
"center": 5,
|
|
"east": 6,
|
|
"southwest": 7,
|
|
"south": 8,
|
|
"southeast": 9
|
|
};
|
|
|
|
export default function run(object) {
|
|
return new Promise((resolve, reject) => {
|
|
// If the image has a path, it must also have a type
|
|
let promise = Promise.resolve();
|
|
if (object.path) {
|
|
if (object.params.type !== "image/gif" && object.onlyGIF) resolve({
|
|
buffer: Buffer.alloc(0),
|
|
fileExtension: "nogif"
|
|
});
|
|
promise = request(object.path).then(res => res.body.arrayBuffer()).then(buf => Buffer.from(buf));
|
|
}
|
|
// Convert from a MIME type (e.g. "image/png") to something the image processor understands (e.g. "png").
|
|
// Don't set `type` directly on the object we are passed as it will be read afterwards.
|
|
// If no image type is given (say, the command generates its own image), make it a PNG.
|
|
const fileExtension = object.params.type ? object.params.type.split("/")[1] : "png";
|
|
promise.then(buf => {
|
|
if (buf) object.params.data = buf;
|
|
const objectWithFixedType = Object.assign({}, object.params, { type: fileExtension });
|
|
if (objectWithFixedType.gravity) {
|
|
if (isNaN(objectWithFixedType.gravity)) {
|
|
objectWithFixedType.gravity = enumMap[objectWithFixedType.gravity];
|
|
}
|
|
}
|
|
objectWithFixedType.basePath = path.join(path.dirname(fileURLToPath(import.meta.url)), "../");
|
|
try {
|
|
const result = img.image(object.cmd, objectWithFixedType);
|
|
const returnObject = {
|
|
buffer: result.data,
|
|
fileExtension: result.type,
|
|
width: result.width,
|
|
height: result.height
|
|
};
|
|
resolve(returnObject);
|
|
} catch (e) {
|
|
reject(e);
|
|
}
|
|
});
|
|
});
|
|
}
|
|
|
|
if (!isMainThread) {
|
|
run(workerData)
|
|
.then(returnObject => {
|
|
parentPort.postMessage(returnObject);
|
|
process.exit();
|
|
})
|
|
.catch(err => {
|
|
// turn promise rejection into normal error
|
|
throw err;
|
|
});
|
|
}
|