work.suroh.tk/node_modules/app-module-path/lib/index.js

109 lines
2.9 KiB
JavaScript
Raw Normal View History

2019-12-02 12:22:45 +00:00
var Module = require('module').Module;
var nodePath = require('path');
var appModulePaths = [];
var old_nodeModulePaths = Module._nodeModulePaths;
var allowedDirs = {};
function checkIfDirAllowed(from) {
var currentDir = from;
while (currentDir) {
if (allowedDirs[currentDir]) {
return true;
}
var basename = nodePath.basename(currentDir);
if (basename === 'node_modules') {
return false;
}
var parentDir = nodePath.dirname(currentDir);
if (parentDir === currentDir) {
break;
}
currentDir = parentDir;
}
return true;
}
Module._nodeModulePaths = function(from) {
var paths = old_nodeModulePaths.call(this, from);
// Only include the app module path for top-level modules
// that were not installed or that were explicitly allowed
if (checkIfDirAllowed(from)) {
paths = paths.concat(appModulePaths);
}
return paths;
};
function enableForDir(dir) {
allowedDirs[dir] = true;
}
function addPath(path, parent) {
// Anable app-module-path to work under any directories that are explicitly added
enableForDir(path);
function addPathHelper(targetArray) {
path = nodePath.normalize(path);
if (targetArray && targetArray.indexOf(path) === -1) {
targetArray.push(path);
}
}
path = nodePath.normalize(path);
if (appModulePaths.indexOf(path) === -1) {
appModulePaths.push(path);
// Enable the search path for the current top-level module
if (require.main) {
addPathHelper(require.main.paths);
}
parent = parent || module.parent;
// Also modify the paths of the module that was used to load the app-module-paths module
// and all of it's parents
while(parent && parent !== require.main) {
addPathHelper(parent.paths);
parent = parent.parent;
}
}
}
function removePath(path) {
function removePathHelper(targetArray) {
path = nodePath.normalize(path);
if (!targetArray) return;
var index = targetArray.indexOf(path);
if (index === -1) return;
targetArray.splice(index, 1);
}
var parent;
path = nodePath.normalize(path);
var index = appModulePaths.indexOf(path);
if (index > -1) {
appModulePaths.splice(index, 1);
// Enable the search path for the current top-level module
if (require.main) removePathHelper(require.main.paths);
parent = module.parent;
// Also modify the paths of the module that was used to load the app-module-paths module
// and all of it's parents
while(parent && parent !== require.main) {
removePathHelper(parent.paths);
parent = parent.parent;
}
}
}
exports.addPath = addPath;
exports.removePath = removePath;
exports.enableForDir = enableForDir;