electron/lib/common/reset-search-paths.ts

53 lines
1.7 KiB
TypeScript
Raw Normal View History

2020-03-20 20:28:31 +00:00
import * as path from 'path';
2020-03-20 20:28:31 +00:00
const Module = require('module');
2016-01-12 02:40:23 +00:00
2016-01-14 18:35:29 +00:00
// Clear Node's global search paths.
2020-03-20 20:28:31 +00:00
Module.globalPaths.length = 0;
2016-01-12 02:40:23 +00:00
2016-01-14 18:35:29 +00:00
// Prevent Node from adding paths outside this app to search paths.
2020-03-20 20:28:31 +00:00
const resourcesPathWithTrailingSlash = process.resourcesPath + path.sep;
const originalNodeModulePaths = Module._nodeModulePaths;
Module._nodeModulePaths = function (from: string) {
2020-03-20 20:28:31 +00:00
const paths: string[] = originalNodeModulePaths(from);
const fromPath = path.resolve(from) + path.sep;
2016-06-23 22:56:29 +00:00
// If "from" is outside the app then we do nothing.
if (fromPath.startsWith(resourcesPathWithTrailingSlash)) {
return paths.filter(function (candidate) {
2020-03-20 20:28:31 +00:00
return candidate.startsWith(resourcesPathWithTrailingSlash);
});
2016-06-23 22:56:29 +00:00
} else {
2020-03-20 20:28:31 +00:00
return paths;
2016-01-12 02:40:23 +00:00
}
2020-03-20 20:28:31 +00:00
};
2016-05-23 17:56:49 +00:00
2019-06-02 20:03:03 +00:00
// Make a fake Electron module that we will insert into the module cache
const makeElectronModule = (name: string) => {
const electronModule = new Module('electron', null);
electronModule.id = 'electron';
electronModule.loaded = true;
electronModule.filename = name;
Object.defineProperty(electronModule, 'exports', {
get: () => require('electron')
});
Module._cache[name] = electronModule;
};
2019-06-02 20:03:03 +00:00
makeElectronModule('electron');
makeElectronModule('electron/common');
if (process.type === 'browser') {
makeElectronModule('electron/main');
}
if (process.type === 'renderer') {
makeElectronModule('electron/renderer');
}
2020-03-20 20:28:31 +00:00
const originalResolveFilename = Module._resolveFilename;
Module._resolveFilename = function (request: string, parent: NodeModule, isMain: boolean) {
if (request === 'electron' || request.startsWith('electron/')) {
2020-03-20 20:28:31 +00:00
return 'electron';
2016-05-23 17:56:49 +00:00
} else {
2020-03-20 20:28:31 +00:00
return originalResolveFilename(request, parent, isMain);
2016-05-23 17:56:49 +00:00
}
2020-03-20 20:28:31 +00:00
};