electron/lib/common/init.ts

71 lines
2.4 KiB
TypeScript
Raw Normal View History

2020-03-20 20:28:31 +00:00
import * as util from 'util';
2020-03-20 20:28:31 +00:00
import { electronBindingSetup } from '@electron/internal/common/electron-binding-setup';
2016-01-12 02:40:23 +00:00
2020-03-20 20:28:31 +00:00
const timers = require('timers');
2020-03-20 20:28:31 +00:00
process.electronBinding = electronBindingSetup(process._linkedBinding, process.type);
type AnyFn = (...args: any[]) => any
2016-01-12 02:40:23 +00:00
2016-01-14 18:44:21 +00:00
// setImmediate and process.nextTick makes use of uv_check and uv_prepare to
// run the callbacks, however since we only run uv loop on requests, the
// callbacks wouldn't be called until something else activated the uv loop,
// which would delay the callbacks for arbitrary long time. So we should
// initiatively activate the uv loop once setImmediate and process.nextTick is
// called.
const wrapWithActivateUvLoop = function <T extends AnyFn> (func: T): T {
return wrap(func, function (func) {
return function (this: any, ...args: any[]) {
2020-03-20 20:28:31 +00:00
process.activateUvLoop();
return func.apply(this, args);
};
}) as T;
};
/**
* Casts to any below for func are due to Typescript not supporting symbols
* in index signatures
*
* Refs: https://github.com/Microsoft/TypeScript/issues/1863
*/
function wrap <T extends AnyFn> (func: T, wrapper: (fn: AnyFn) => T) {
2020-03-20 20:28:31 +00:00
const wrapped = wrapper(func);
if ((func as any)[util.promisify.custom]) {
2020-03-20 20:28:31 +00:00
(wrapped as any)[util.promisify.custom] = wrapper((func as any)[util.promisify.custom]);
2016-03-25 19:57:17 +00:00
}
2020-03-20 20:28:31 +00:00
return wrapped;
2016-03-25 19:57:17 +00:00
}
2016-01-12 02:40:23 +00:00
2020-03-20 20:28:31 +00:00
process.nextTick = wrapWithActivateUvLoop(process.nextTick);
2016-01-12 02:40:23 +00:00
2020-03-20 20:28:31 +00:00
global.setImmediate = timers.setImmediate = wrapWithActivateUvLoop(timers.setImmediate);
global.clearImmediate = timers.clearImmediate;
2016-01-12 02:40:23 +00:00
// setTimeout needs to update the polling timeout of the event loop, when
// called under Chromium's event loop the node's event loop won't get a chance
// to update the timeout, so we have to force the node's event loop to
// recalculate the timeout in browser process.
2020-03-20 20:28:31 +00:00
timers.setTimeout = wrapWithActivateUvLoop(timers.setTimeout);
timers.setInterval = wrapWithActivateUvLoop(timers.setInterval);
// Only override the global setTimeout/setInterval impls in the browser process
2016-01-12 02:40:23 +00:00
if (process.type === 'browser') {
2020-03-20 20:28:31 +00:00
global.setTimeout = timers.setTimeout;
global.setInterval = timers.setInterval;
2016-01-12 02:40:23 +00:00
}
if (process.platform === 'win32') {
// Always returns EOF for stdin stream.
2020-03-20 20:28:31 +00:00
const { Readable } = require('stream');
const stdin = new Readable();
stdin.push(null);
Object.defineProperty(process, 'stdin', {
configurable: false,
enumerable: true,
get () {
2020-03-20 20:28:31 +00:00
return stdin;
}
2020-03-20 20:28:31 +00:00
});
}