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
			| 
								 
											3 years ago
										 
									 | 
							
								"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;
							 |