You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
280 lines
8.0 KiB
280 lines
8.0 KiB
"use strict"; |
|
|
|
Object.defineProperty(exports, "__esModule", { |
|
value: true |
|
}); |
|
exports.default = void 0; |
|
|
|
var _assert = require("assert"); |
|
|
|
var _t = require("@babel/types"); |
|
|
|
var _importBuilder = require("./import-builder"); |
|
|
|
var _isModule = require("./is-module"); |
|
|
|
const { |
|
numericLiteral, |
|
sequenceExpression |
|
} = _t; |
|
|
|
class ImportInjector { |
|
constructor(path, importedSource, opts) { |
|
this._defaultOpts = { |
|
importedSource: null, |
|
importedType: "commonjs", |
|
importedInterop: "babel", |
|
importingInterop: "babel", |
|
ensureLiveReference: false, |
|
ensureNoContext: false, |
|
importPosition: "before" |
|
}; |
|
const programPath = path.find(p => p.isProgram()); |
|
this._programPath = programPath; |
|
this._programScope = programPath.scope; |
|
this._hub = programPath.hub; |
|
this._defaultOpts = this._applyDefaults(importedSource, opts, true); |
|
} |
|
|
|
addDefault(importedSourceIn, opts) { |
|
return this.addNamed("default", importedSourceIn, opts); |
|
} |
|
|
|
addNamed(importName, importedSourceIn, opts) { |
|
_assert(typeof importName === "string"); |
|
|
|
return this._generateImport(this._applyDefaults(importedSourceIn, opts), importName); |
|
} |
|
|
|
addNamespace(importedSourceIn, opts) { |
|
return this._generateImport(this._applyDefaults(importedSourceIn, opts), null); |
|
} |
|
|
|
addSideEffect(importedSourceIn, opts) { |
|
return this._generateImport(this._applyDefaults(importedSourceIn, opts), void 0); |
|
} |
|
|
|
_applyDefaults(importedSource, opts, isInit = false) { |
|
let newOpts; |
|
|
|
if (typeof importedSource === "string") { |
|
newOpts = Object.assign({}, this._defaultOpts, { |
|
importedSource |
|
}, opts); |
|
} else { |
|
_assert(!opts, "Unexpected secondary arguments."); |
|
|
|
newOpts = Object.assign({}, this._defaultOpts, importedSource); |
|
} |
|
|
|
if (!isInit && opts) { |
|
if (opts.nameHint !== undefined) newOpts.nameHint = opts.nameHint; |
|
if (opts.blockHoist !== undefined) newOpts.blockHoist = opts.blockHoist; |
|
} |
|
|
|
return newOpts; |
|
} |
|
|
|
_generateImport(opts, importName) { |
|
const isDefault = importName === "default"; |
|
const isNamed = !!importName && !isDefault; |
|
const isNamespace = importName === null; |
|
const { |
|
importedSource, |
|
importedType, |
|
importedInterop, |
|
importingInterop, |
|
ensureLiveReference, |
|
ensureNoContext, |
|
nameHint, |
|
importPosition, |
|
blockHoist |
|
} = opts; |
|
let name = nameHint || importName; |
|
const isMod = (0, _isModule.default)(this._programPath); |
|
const isModuleForNode = isMod && importingInterop === "node"; |
|
const isModuleForBabel = isMod && importingInterop === "babel"; |
|
|
|
if (importPosition === "after" && !isMod) { |
|
throw new Error(`"importPosition": "after" is only supported in modules`); |
|
} |
|
|
|
const builder = new _importBuilder.default(importedSource, this._programScope, this._hub); |
|
|
|
if (importedType === "es6") { |
|
if (!isModuleForNode && !isModuleForBabel) { |
|
throw new Error("Cannot import an ES6 module from CommonJS"); |
|
} |
|
|
|
builder.import(); |
|
|
|
if (isNamespace) { |
|
builder.namespace(nameHint || importedSource); |
|
} else if (isDefault || isNamed) { |
|
builder.named(name, importName); |
|
} |
|
} else if (importedType !== "commonjs") { |
|
throw new Error(`Unexpected interopType "${importedType}"`); |
|
} else if (importedInterop === "babel") { |
|
if (isModuleForNode) { |
|
name = name !== "default" ? name : importedSource; |
|
const es6Default = `${importedSource}$es6Default`; |
|
builder.import(); |
|
|
|
if (isNamespace) { |
|
builder.default(es6Default).var(name || importedSource).wildcardInterop(); |
|
} else if (isDefault) { |
|
if (ensureLiveReference) { |
|
builder.default(es6Default).var(name || importedSource).defaultInterop().read("default"); |
|
} else { |
|
builder.default(es6Default).var(name).defaultInterop().prop(importName); |
|
} |
|
} else if (isNamed) { |
|
builder.default(es6Default).read(importName); |
|
} |
|
} else if (isModuleForBabel) { |
|
builder.import(); |
|
|
|
if (isNamespace) { |
|
builder.namespace(name || importedSource); |
|
} else if (isDefault || isNamed) { |
|
builder.named(name, importName); |
|
} |
|
} else { |
|
builder.require(); |
|
|
|
if (isNamespace) { |
|
builder.var(name || importedSource).wildcardInterop(); |
|
} else if ((isDefault || isNamed) && ensureLiveReference) { |
|
if (isDefault) { |
|
name = name !== "default" ? name : importedSource; |
|
builder.var(name).read(importName); |
|
builder.defaultInterop(); |
|
} else { |
|
builder.var(importedSource).read(importName); |
|
} |
|
} else if (isDefault) { |
|
builder.var(name).defaultInterop().prop(importName); |
|
} else if (isNamed) { |
|
builder.var(name).prop(importName); |
|
} |
|
} |
|
} else if (importedInterop === "compiled") { |
|
if (isModuleForNode) { |
|
builder.import(); |
|
|
|
if (isNamespace) { |
|
builder.default(name || importedSource); |
|
} else if (isDefault || isNamed) { |
|
builder.default(importedSource).read(name); |
|
} |
|
} else if (isModuleForBabel) { |
|
builder.import(); |
|
|
|
if (isNamespace) { |
|
builder.namespace(name || importedSource); |
|
} else if (isDefault || isNamed) { |
|
builder.named(name, importName); |
|
} |
|
} else { |
|
builder.require(); |
|
|
|
if (isNamespace) { |
|
builder.var(name || importedSource); |
|
} else if (isDefault || isNamed) { |
|
if (ensureLiveReference) { |
|
builder.var(importedSource).read(name); |
|
} else { |
|
builder.prop(importName).var(name); |
|
} |
|
} |
|
} |
|
} else if (importedInterop === "uncompiled") { |
|
if (isDefault && ensureLiveReference) { |
|
throw new Error("No live reference for commonjs default"); |
|
} |
|
|
|
if (isModuleForNode) { |
|
builder.import(); |
|
|
|
if (isNamespace) { |
|
builder.default(name || importedSource); |
|
} else if (isDefault) { |
|
builder.default(name); |
|
} else if (isNamed) { |
|
builder.default(importedSource).read(name); |
|
} |
|
} else if (isModuleForBabel) { |
|
builder.import(); |
|
|
|
if (isNamespace) { |
|
builder.default(name || importedSource); |
|
} else if (isDefault) { |
|
builder.default(name); |
|
} else if (isNamed) { |
|
builder.named(name, importName); |
|
} |
|
} else { |
|
builder.require(); |
|
|
|
if (isNamespace) { |
|
builder.var(name || importedSource); |
|
} else if (isDefault) { |
|
builder.var(name); |
|
} else if (isNamed) { |
|
if (ensureLiveReference) { |
|
builder.var(importedSource).read(name); |
|
} else { |
|
builder.var(name).prop(importName); |
|
} |
|
} |
|
} |
|
} else { |
|
throw new Error(`Unknown importedInterop "${importedInterop}".`); |
|
} |
|
|
|
const { |
|
statements, |
|
resultName |
|
} = builder.done(); |
|
|
|
this._insertStatements(statements, importPosition, blockHoist); |
|
|
|
if ((isDefault || isNamed) && ensureNoContext && resultName.type !== "Identifier") { |
|
return sequenceExpression([numericLiteral(0), resultName]); |
|
} |
|
|
|
return resultName; |
|
} |
|
|
|
_insertStatements(statements, importPosition = "before", blockHoist = 3) { |
|
const body = this._programPath.get("body"); |
|
|
|
if (importPosition === "after") { |
|
for (let i = body.length - 1; i >= 0; i--) { |
|
if (body[i].isImportDeclaration()) { |
|
body[i].insertAfter(statements); |
|
return; |
|
} |
|
} |
|
} else { |
|
statements.forEach(node => { |
|
node._blockHoist = blockHoist; |
|
}); |
|
const targetPath = body.find(p => { |
|
const val = p.node._blockHoist; |
|
return Number.isFinite(val) && val < 4; |
|
}); |
|
|
|
if (targetPath) { |
|
targetPath.insertBefore(statements); |
|
return; |
|
} |
|
} |
|
|
|
this._programPath.unshiftContainer("body", statements); |
|
} |
|
|
|
} |
|
|
|
exports.default = ImportInjector; |