100 lines
2.9 KiB
JavaScript
100 lines
2.9 KiB
JavaScript
"use strict";
|
|
Object.defineProperty(exports, "__esModule", {
|
|
value: true
|
|
});
|
|
Object.defineProperty(exports, "default", {
|
|
enumerable: true,
|
|
get: function() {
|
|
return getModuleDependencies;
|
|
}
|
|
});
|
|
const _fs = /*#__PURE__*/ _interop_require_default(require("fs"));
|
|
const _path = /*#__PURE__*/ _interop_require_default(require("path"));
|
|
function _interop_require_default(obj) {
|
|
return obj && obj.__esModule ? obj : {
|
|
default: obj
|
|
};
|
|
}
|
|
let jsExtensions = [
|
|
".js",
|
|
".cjs",
|
|
".mjs"
|
|
];
|
|
// Given the current file `a.ts`, we want to make sure that when importing `b` that we resolve
|
|
// `b.ts` before `b.js`
|
|
//
|
|
// E.g.:
|
|
//
|
|
// a.ts
|
|
// b // .ts
|
|
// c // .ts
|
|
// a.js
|
|
// b // .js or .ts
|
|
let jsResolutionOrder = [
|
|
"",
|
|
".js",
|
|
".cjs",
|
|
".mjs",
|
|
".ts",
|
|
".cts",
|
|
".mts",
|
|
".jsx",
|
|
".tsx"
|
|
];
|
|
let tsResolutionOrder = [
|
|
"",
|
|
".ts",
|
|
".cts",
|
|
".mts",
|
|
".tsx",
|
|
".js",
|
|
".cjs",
|
|
".mjs",
|
|
".jsx"
|
|
];
|
|
function resolveWithExtension(file, extensions) {
|
|
// Try to find `./a.ts`, `./a.ts`, ... from `./a`
|
|
for (let ext of extensions){
|
|
let full = `${file}${ext}`;
|
|
if (_fs.default.existsSync(full) && _fs.default.statSync(full).isFile()) {
|
|
return full;
|
|
}
|
|
}
|
|
// Try to find `./a/index.js` from `./a`
|
|
for (let ext of extensions){
|
|
let full = `${file}/index${ext}`;
|
|
if (_fs.default.existsSync(full)) {
|
|
return full;
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
function* _getModuleDependencies(filename, base, seen, ext = _path.default.extname(filename)) {
|
|
// Try to find the file
|
|
let absoluteFile = resolveWithExtension(_path.default.resolve(base, filename), jsExtensions.includes(ext) ? jsResolutionOrder : tsResolutionOrder);
|
|
if (absoluteFile === null) return; // File doesn't exist
|
|
// Prevent infinite loops when there are circular dependencies
|
|
if (seen.has(absoluteFile)) return; // Already seen
|
|
seen.add(absoluteFile);
|
|
// Mark the file as a dependency
|
|
yield absoluteFile;
|
|
// Resolve new base for new imports/requires
|
|
base = _path.default.dirname(absoluteFile);
|
|
ext = _path.default.extname(absoluteFile);
|
|
let contents = _fs.default.readFileSync(absoluteFile, "utf-8");
|
|
// Find imports/requires
|
|
for (let match of [
|
|
...contents.matchAll(/import[\s\S]*?['"](.{3,}?)['"]/gi),
|
|
...contents.matchAll(/import[\s\S]*from[\s\S]*?['"](.{3,}?)['"]/gi),
|
|
...contents.matchAll(/require\(['"`](.+)['"`]\)/gi)
|
|
]){
|
|
// Bail out if it's not a relative file
|
|
if (!match[1].startsWith(".")) continue;
|
|
yield* _getModuleDependencies(match[1], base, seen, ext);
|
|
}
|
|
}
|
|
function getModuleDependencies(absoluteFilePath) {
|
|
if (absoluteFilePath === null) return new Set();
|
|
return new Set(_getModuleDependencies(absoluteFilePath, _path.default.dirname(absoluteFilePath), new Set()));
|
|
}
|