wrangler-action/node_modules/vitest/dist/vendor-rpc.ad5b08c7.js
2023-08-07 15:11:15 -05:00

61 lines
1.9 KiB
JavaScript

import { getSafeTimers } from '@vitest/utils';
import { g as getWorkerState } from './vendor-global.6795f91f.js';
const { get } = Reflect;
function withSafeTimers(fn) {
var _a;
const { setTimeout, clearTimeout, nextTick, setImmediate, clearImmediate } = getSafeTimers();
const currentSetTimeout = globalThis.setTimeout;
const currentClearTimeout = globalThis.clearTimeout;
const currentSetImmediate = globalThis.setImmediate;
const currentClearImmediate = globalThis.clearImmediate;
const currentNextTick = (_a = globalThis.process) == null ? void 0 : _a.nextTick;
try {
globalThis.setTimeout = setTimeout;
globalThis.clearTimeout = clearTimeout;
globalThis.setImmediate = setImmediate;
globalThis.clearImmediate = clearImmediate;
if (globalThis.process)
globalThis.process.nextTick = nextTick;
const result = fn();
return result;
} finally {
globalThis.setTimeout = currentSetTimeout;
globalThis.clearTimeout = currentClearTimeout;
globalThis.setImmediate = currentSetImmediate;
globalThis.clearImmediate = currentClearImmediate;
if (globalThis.process) {
nextTick(() => {
globalThis.process.nextTick = currentNextTick;
});
}
}
}
const promises = /* @__PURE__ */ new Set();
async function rpcDone() {
if (!promises.size)
return;
const awaitable = Array.from(promises);
return Promise.all(awaitable);
}
function rpc() {
const { rpc: rpc2 } = getWorkerState();
return new Proxy(rpc2, {
get(target, p, handler) {
const sendCall = get(target, p, handler);
const safeSendCall = (...args) => withSafeTimers(async () => {
const result = sendCall(...args);
promises.add(result);
try {
return await result;
} finally {
promises.delete(result);
}
});
safeSendCall.asEvent = sendCall.asEvent;
return safeSendCall;
}
});
}
export { rpc as a, rpcDone as r };