mirror of
https://github.com/cloudflare/wrangler-action.git
synced 2024-11-23 02:23:26 +01:00
62 lines
1.9 KiB
JavaScript
62 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 };
|