const magick = require("../build/Release/image.node"); const { Worker, isMainThread, parentPort, workerData } = require("worker_threads"); const fetch = require("node-fetch"); const { promisify } = require("util"); const AbortController = require("abort-controller"); const net = require("net"); const dgram = require("dgram"); const fileType = require("file-type"); const execPromise = promisify(require("child_process").exec); const servers = require("../servers.json").image; const formats = ["image/jpeg", "image/png", "image/webp", "image/gif"]; const getFormat = (buffer, delimiter) => { for (var i = 0; i < buffer.length ; i++) { if (String.fromCharCode(buffer[i]) === delimiter) { return { buffer: buffer.slice(0, i), dataStart: i }; } } }; exports.check = (cmd) => { return magick[cmd] ? true : false; }; exports.getType = async (image) => { if (!image.startsWith("http")) { const imageType = await fileType.fromFile(image); if (imageType && formats.includes(imageType.mime)) { return imageType.mime; } return undefined; } let type; const controller = new AbortController(); const timeout = setTimeout(() => { controller.abort(); }, 25000); try { const imageRequest = await fetch(image, { signal: controller.signal, highWaterMark: 512 }); const imageBuffer = await imageRequest.buffer(); const imageType = await fileType.fromBuffer(imageBuffer); if (imageType && formats.includes(imageType.mime)) { type = imageType.mime; } } catch (error) { if (error.name === "AbortError") { throw Error("Timed out"); } else { throw error; } } finally { clearTimeout(timeout); } return type; }; exports.run = (object, fromAPI = false) => { return new Promise(async (resolve, reject) => { if (process.env.API === "true" && !fromAPI) { const currentServer = servers[Math.floor(Math.random() * servers.length)]; const socket = dgram.createSocket("udp4"); const data = Buffer.concat([Buffer.from([0x1]), Buffer.from(JSON.stringify(object))]); const timeout = setTimeout(() => { reject("Timed out"); }, 25000); let jobID; socket.on("message", (msg) => { const opcode = msg.readUint8(0); const req = msg.slice(37, msg.length); const uuid = msg.slice(1, 36).toString(); if (opcode === 0x0) { clearTimeout(timeout); jobID = uuid; } else if (opcode === 0x1) { if (jobID === uuid) { const client = net.createConnection(req.toString(), currentServer); const array = []; client.on("data", (rawData) => { array.push(rawData); }); client.once("end", () => { const data = Buffer.concat(array); const format = getFormat(data, "\n"); const payload = { buffer: data.slice(format.dataStart + 1), type: format.buffer.toString().split("/")[1] }; socket.close(); resolve(payload); }); client.on("error", (err) => { throw err; }); } } else if (opcode === 0x2) { if (jobID === uuid) reject(req); } }); socket.send(data, 8080, currentServer, (err) => { if (err) reject(err); }); } else if (isMainThread && !fromAPI) { const worker = new Worker(__filename, { workerData: object }); worker.on("message", (data) => { resolve({ buffer: Buffer.from([...data.buffer]), type: data.type }); }); worker.on("error", reject); } else { let type; if (!fromAPI && object.path) { const newType = (object.type ? object.type : await this.getType(object.path)); type = newType ? newType.split("/")[1] : "png"; if (type !== "gif" && object.onlyGIF) resolve({ buffer: "nogif", type: null }); object.type = type; const delay = (await execPromise(`ffprobe -v 0 -of csv=p=0 -select_streams v:0 -show_entries stream=r_frame_rate ${object.path}`)).stdout.replace("\n", ""); object.delay = (100 / delay.split("/")[0]) * delay.split("/")[1]; } const data = await promisify(magick[object.cmd])(object); const returnObject = fromAPI ? data : { buffer: data, type: type }; if (!isMainThread && !fromAPI) { parentPort.postMessage(returnObject); process.exit(); } else { resolve(returnObject); } } }); }; if (!isMainThread && process.env.API !== "true") this.run(workerData);