310 lines
9.8 KiB
JavaScript
310 lines
9.8 KiB
JavaScript
'use strict';
|
|
|
|
Object.defineProperty(exports, '__esModule', { value: true });
|
|
|
|
var fs = require('node:fs');
|
|
var node_module = require('node:module');
|
|
var path = require('node:path');
|
|
var node_url = require('node:url');
|
|
var node_worker_threads = require('node:worker_threads');
|
|
var utils = require('@pkgr/utils');
|
|
|
|
function _interopDefaultLegacy (e) { return e && typeof e === 'object' && 'default' in e ? e : { 'default': e }; }
|
|
|
|
var fs__default = /*#__PURE__*/_interopDefaultLegacy(fs);
|
|
var path__default = /*#__PURE__*/_interopDefaultLegacy(path);
|
|
|
|
var __async = (__this, __arguments, generator) => {
|
|
return new Promise((resolve, reject) => {
|
|
var fulfilled = (value) => {
|
|
try {
|
|
step(generator.next(value));
|
|
} catch (e) {
|
|
reject(e);
|
|
}
|
|
};
|
|
var rejected = (value) => {
|
|
try {
|
|
step(generator.throw(value));
|
|
} catch (e) {
|
|
reject(e);
|
|
}
|
|
};
|
|
var step = (x) => x.done ? resolve(x.value) : Promise.resolve(x.value).then(fulfilled, rejected);
|
|
step((generator = generator.apply(__this, __arguments)).next());
|
|
});
|
|
};
|
|
const import_meta = {};
|
|
const TsRunner = {
|
|
TsNode: "ts-node",
|
|
EsbuildRegister: "esbuild-register",
|
|
EsbuildRunner: "esbuild-runner",
|
|
SWC: "swc",
|
|
TSX: "tsx"
|
|
};
|
|
const {
|
|
SYNCKIT_BUFFER_SIZE,
|
|
SYNCKIT_TIMEOUT,
|
|
SYNCKIT_EXEC_ARGV,
|
|
SYNCKIT_TS_RUNNER,
|
|
NODE_OPTIONS
|
|
} = process.env;
|
|
const DEFAULT_BUFFER_SIZE = SYNCKIT_BUFFER_SIZE ? +SYNCKIT_BUFFER_SIZE : void 0;
|
|
const DEFAULT_TIMEOUT = SYNCKIT_TIMEOUT ? +SYNCKIT_TIMEOUT : void 0;
|
|
const DEFAULT_WORKER_BUFFER_SIZE = DEFAULT_BUFFER_SIZE || 1024;
|
|
const DEFAULT_EXEC_ARGV = (SYNCKIT_EXEC_ARGV == null ? void 0 : SYNCKIT_EXEC_ARGV.split(",")) || [];
|
|
const DEFAULT_TS_RUNNER = SYNCKIT_TS_RUNNER || TsRunner.TsNode;
|
|
const MTS_SUPPORTED_NODE_VERSION = 16;
|
|
const syncFnCache = /* @__PURE__ */ new Map();
|
|
function extractProperties(object) {
|
|
if (object && typeof object === "object") {
|
|
const properties = {};
|
|
for (const key in object) {
|
|
properties[key] = object[key];
|
|
}
|
|
return properties;
|
|
}
|
|
}
|
|
function createSyncFn(workerPath, bufferSizeOrOptions, timeout) {
|
|
if (!path__default["default"].isAbsolute(workerPath)) {
|
|
throw new Error("`workerPath` must be absolute");
|
|
}
|
|
const cachedSyncFn = syncFnCache.get(workerPath);
|
|
if (cachedSyncFn) {
|
|
return cachedSyncFn;
|
|
}
|
|
const syncFn = startWorkerThread(
|
|
workerPath,
|
|
typeof bufferSizeOrOptions === "number" ? { bufferSize: bufferSizeOrOptions, timeout } : bufferSizeOrOptions
|
|
);
|
|
syncFnCache.set(workerPath, syncFn);
|
|
return syncFn;
|
|
}
|
|
const cjsRequire = typeof require === "undefined" ? node_module.createRequire(import_meta.url) : require;
|
|
const dataUrl = (code) => new URL(`data:text/javascript,${encodeURIComponent(code)}`);
|
|
const isFile = (path2) => {
|
|
try {
|
|
return fs__default["default"].statSync(path2).isFile();
|
|
} catch (e) {
|
|
return false;
|
|
}
|
|
};
|
|
const setupTsRunner = (workerPath, { execArgv, tsRunner }) => {
|
|
let ext = path__default["default"].extname(workerPath);
|
|
if (!/[/\\]node_modules[/\\]/.test(workerPath) && (!ext || /^\.[cm]?js$/.test(ext))) {
|
|
const workPathWithoutExt = ext ? workerPath.slice(0, -ext.length) : workerPath;
|
|
let extensions;
|
|
switch (ext) {
|
|
case ".cjs":
|
|
extensions = [".cts", ".cjs"];
|
|
break;
|
|
case ".mjs":
|
|
extensions = [".mts", ".mjs"];
|
|
break;
|
|
default:
|
|
extensions = [".ts", ".js"];
|
|
break;
|
|
}
|
|
const found = utils.tryExtensions(workPathWithoutExt, extensions);
|
|
let differentExt;
|
|
if (found && (!ext || (differentExt = found !== workPathWithoutExt))) {
|
|
workerPath = found;
|
|
if (differentExt) {
|
|
ext = path__default["default"].extname(workerPath);
|
|
}
|
|
}
|
|
}
|
|
const isTs = /\.[cm]?ts$/.test(workerPath);
|
|
let tsUseEsm = workerPath.endsWith(".mts");
|
|
if (isTs) {
|
|
if (!tsUseEsm) {
|
|
const pkg = utils.findUp(workerPath);
|
|
if (pkg) {
|
|
tsUseEsm = cjsRequire(pkg).type === "module";
|
|
}
|
|
}
|
|
switch (tsRunner) {
|
|
case TsRunner.TsNode: {
|
|
if (tsUseEsm) {
|
|
if (!execArgv.includes("--loader")) {
|
|
execArgv = ["--loader", `${TsRunner.TsNode}/esm`, ...execArgv];
|
|
}
|
|
} else if (!execArgv.includes("-r")) {
|
|
execArgv = ["-r", `${TsRunner.TsNode}/register`, ...execArgv];
|
|
}
|
|
break;
|
|
}
|
|
case TsRunner.EsbuildRegister: {
|
|
if (!execArgv.includes("-r")) {
|
|
execArgv = ["-r", TsRunner.EsbuildRegister, ...execArgv];
|
|
}
|
|
break;
|
|
}
|
|
case TsRunner.EsbuildRunner: {
|
|
if (!execArgv.includes("-r")) {
|
|
execArgv = ["-r", `${TsRunner.EsbuildRunner}/register`, ...execArgv];
|
|
}
|
|
break;
|
|
}
|
|
case TsRunner.SWC: {
|
|
if (!execArgv.includes("-r")) {
|
|
execArgv = ["-r", `@${TsRunner.SWC}-node/register`, ...execArgv];
|
|
}
|
|
break;
|
|
}
|
|
case TsRunner.TSX: {
|
|
if (!execArgv.includes("--loader")) {
|
|
execArgv = ["--loader", TsRunner.TSX, ...execArgv];
|
|
}
|
|
break;
|
|
}
|
|
default: {
|
|
throw new Error(`Unknown ts runner: ${String(tsRunner)}`);
|
|
}
|
|
}
|
|
}
|
|
if (process.versions.pnp) {
|
|
const nodeOptions = NODE_OPTIONS == null ? void 0 : NODE_OPTIONS.split(/\s+/);
|
|
let pnpApiPath;
|
|
try {
|
|
pnpApiPath = cjsRequire.resolve("pnpapi");
|
|
} catch (e) {
|
|
}
|
|
if (pnpApiPath && !(nodeOptions == null ? void 0 : nodeOptions.some(
|
|
(option, index) => ["-r", "--require"].includes(option) && pnpApiPath === cjsRequire.resolve(nodeOptions[index + 1])
|
|
)) && !execArgv.includes(pnpApiPath)) {
|
|
execArgv = ["-r", pnpApiPath, ...execArgv];
|
|
const pnpLoaderPath = path__default["default"].resolve(pnpApiPath, "../.pnp.loader.mjs");
|
|
if (isFile(pnpLoaderPath)) {
|
|
const experimentalLoader = node_url.pathToFileURL(pnpLoaderPath).toString();
|
|
execArgv = ["--experimental-loader", experimentalLoader, ...execArgv];
|
|
}
|
|
}
|
|
}
|
|
return {
|
|
ext,
|
|
isTs,
|
|
tsUseEsm,
|
|
workerPath,
|
|
execArgv
|
|
};
|
|
};
|
|
function startWorkerThread(workerPath, {
|
|
bufferSize = DEFAULT_WORKER_BUFFER_SIZE,
|
|
timeout = DEFAULT_TIMEOUT,
|
|
execArgv = DEFAULT_EXEC_ARGV,
|
|
tsRunner = DEFAULT_TS_RUNNER
|
|
} = {}) {
|
|
const { port1: mainPort, port2: workerPort } = new node_worker_threads.MessageChannel();
|
|
const {
|
|
isTs,
|
|
ext,
|
|
tsUseEsm,
|
|
workerPath: finalWorkerPath,
|
|
execArgv: finalExecArgv
|
|
} = setupTsRunner(workerPath, { execArgv, tsRunner });
|
|
const workerPathUrl = node_url.pathToFileURL(finalWorkerPath);
|
|
if (/\.[cm]ts$/.test(finalWorkerPath)) {
|
|
const isTsxSupported = !tsUseEsm || Number.parseFloat(process.versions.node) >= MTS_SUPPORTED_NODE_VERSION;
|
|
if ([
|
|
TsRunner.EsbuildRegister,
|
|
TsRunner.EsbuildRunner,
|
|
TsRunner.SWC,
|
|
...isTsxSupported ? [] : [TsRunner.TSX]
|
|
].includes(tsRunner)) {
|
|
throw new Error(
|
|
`${tsRunner} is not supported for ${ext} files yet` + (isTsxSupported ? ", you can try [tsx](https://github.com/esbuild-kit/tsx) instead" : "")
|
|
);
|
|
}
|
|
}
|
|
const useEval = isTs && !tsUseEsm;
|
|
const worker = new node_worker_threads.Worker(
|
|
tsUseEsm && tsRunner === TsRunner.TsNode ? dataUrl(`import '${String(workerPathUrl)}'`) : useEval ? `require('${finalWorkerPath.replace(/\\/g, "\\\\")}')` : workerPathUrl,
|
|
{
|
|
eval: useEval,
|
|
workerData: { workerPort },
|
|
transferList: [workerPort],
|
|
execArgv: finalExecArgv
|
|
}
|
|
);
|
|
let nextID = 0;
|
|
const syncFn = (...args) => {
|
|
const id = nextID++;
|
|
const sharedBuffer = new SharedArrayBuffer(bufferSize);
|
|
const sharedBufferView = new Int32Array(sharedBuffer);
|
|
const msg = { sharedBuffer, id, args };
|
|
worker.postMessage(msg);
|
|
const status = Atomics.wait(sharedBufferView, 0, 0, timeout);
|
|
if (!["ok", "not-equal"].includes(status)) {
|
|
throw new Error("Internal error: Atomics.wait() failed: " + status);
|
|
}
|
|
const {
|
|
id: id2,
|
|
result,
|
|
error,
|
|
properties
|
|
} = node_worker_threads.receiveMessageOnPort(mainPort).message;
|
|
if (id !== id2) {
|
|
throw new Error(`Internal error: Expected id ${id} but got id ${id2}`);
|
|
}
|
|
if (error) {
|
|
throw Object.assign(error, properties);
|
|
}
|
|
return result;
|
|
};
|
|
worker.unref();
|
|
return syncFn;
|
|
}
|
|
function runAsWorker(fn) {
|
|
if (!node_worker_threads.workerData) {
|
|
return;
|
|
}
|
|
const { workerPort } = node_worker_threads.workerData;
|
|
try {
|
|
node_worker_threads.parentPort.on(
|
|
"message",
|
|
({ sharedBuffer, id, args }) => {
|
|
;
|
|
(() => __async(this, null, function* () {
|
|
const sharedBufferView = new Int32Array(sharedBuffer);
|
|
let msg;
|
|
try {
|
|
msg = { id, result: yield fn(...args) };
|
|
} catch (error) {
|
|
msg = { id, error, properties: extractProperties(error) };
|
|
}
|
|
workerPort.postMessage(msg);
|
|
Atomics.add(sharedBufferView, 0, 1);
|
|
Atomics.notify(sharedBufferView, 0);
|
|
}))();
|
|
}
|
|
);
|
|
} catch (error) {
|
|
node_worker_threads.parentPort.on(
|
|
"message",
|
|
({ sharedBuffer, id }) => {
|
|
const sharedBufferView = new Int32Array(sharedBuffer);
|
|
workerPort.postMessage({
|
|
id,
|
|
error,
|
|
properties: extractProperties(error)
|
|
});
|
|
Atomics.add(sharedBufferView, 0, 1);
|
|
Atomics.notify(sharedBufferView, 0);
|
|
}
|
|
);
|
|
}
|
|
}
|
|
|
|
exports.DEFAULT_BUFFER_SIZE = DEFAULT_BUFFER_SIZE;
|
|
exports.DEFAULT_EXEC_ARGV = DEFAULT_EXEC_ARGV;
|
|
exports.DEFAULT_TIMEOUT = DEFAULT_TIMEOUT;
|
|
exports.DEFAULT_TS_RUNNER = DEFAULT_TS_RUNNER;
|
|
exports.DEFAULT_WORKER_BUFFER_SIZE = DEFAULT_WORKER_BUFFER_SIZE;
|
|
exports.MTS_SUPPORTED_NODE_VERSION = MTS_SUPPORTED_NODE_VERSION;
|
|
exports.TsRunner = TsRunner;
|
|
exports.createSyncFn = createSyncFn;
|
|
exports.extractProperties = extractProperties;
|
|
exports.isFile = isFile;
|
|
exports.runAsWorker = runAsWorker;
|