77 lines
1.8 KiB
JavaScript
77 lines
1.8 KiB
JavaScript
|
const path = require(`path`);
|
||
|
|
|
||
|
|
let pnpApi = null;
|
||
|
|
try {
|
||
|
|
pnpApi = require(`pnpapi`);
|
||
|
|
} catch {
|
||
|
|
}
|
||
|
|
|
||
|
|
function getVirtualLookupFn(pnpApi) {
|
||
|
|
const reverseMap = new Map();
|
||
|
|
|
||
|
|
for (const locator of pnpApi.getAllLocators()) {
|
||
|
|
const pkg = pnpApi.getPackageInformation(locator);
|
||
|
|
console.assert(pkg, `The package information should be available`);
|
||
|
|
|
||
|
|
const resolvedLocation = pnpApi.resolveVirtual(pkg.packageLocation);
|
||
|
|
if (resolvedLocation === null)
|
||
|
|
continue;
|
||
|
|
|
||
|
|
const aliases = reverseMap.get(resolvedLocation) || [resolvedLocation];
|
||
|
|
reverseMap.set(resolvedLocation, aliases);
|
||
|
|
aliases.push(pkg.packageLocation);
|
||
|
|
}
|
||
|
|
|
||
|
|
const keys = [...reverseMap.keys()].sort((a, b) => {
|
||
|
|
return b.length - a.length;
|
||
|
|
});
|
||
|
|
|
||
|
|
return p => {
|
||
|
|
const prefix = keys.find(candidate => p.startsWith(candidate));
|
||
|
|
if (typeof prefix === `undefined`)
|
||
|
|
return [p];
|
||
|
|
|
||
|
|
const sub = p.substr(prefix.length);
|
||
|
|
return reverseMap.get(prefix).map(alias => {
|
||
|
|
return alias + sub;
|
||
|
|
});
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
if (pnpApi != null) {
|
||
|
|
const resolveVirtualAliases = getVirtualLookupFn(pnpApi);
|
||
|
|
|
||
|
|
module.exports = class FsePnp {
|
||
|
|
constructor(p) {
|
||
|
|
this.normalizedPath = path.resolve(p);
|
||
|
|
this.resolvedPath = pnpApi.resolveVirtual(this.normalizedPath) || this.normalizedPath;
|
||
|
|
}
|
||
|
|
|
||
|
|
transpose(p) {
|
||
|
|
if (this.resolvedPath !== null) {
|
||
|
|
return this.normalizedPath + p.substr(this.resolvedPath.length);
|
||
|
|
} else {
|
||
|
|
return p;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
wrap(fn) {
|
||
|
|
return (path, ...args) => {
|
||
|
|
for (const entry of resolveVirtualAliases(path)) {
|
||
|
|
fn(this.transpose(entry), ...args);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}
|
||
|
|
};
|
||
|
|
} else {
|
||
|
|
module.exports = class FsePnp {
|
||
|
|
constructor(p) {
|
||
|
|
this.resolvedPath = p;
|
||
|
|
}
|
||
|
|
|
||
|
|
wrap(fn) {
|
||
|
|
return fn;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}
|