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; }); }