125 lines
5.2 KiB
JavaScript
125 lines
5.2 KiB
JavaScript
|
/*
|
||
|
MIT License http://www.opensource.org/licenses/mit-license.php
|
||
|
Author Tobias Koppers @sokra
|
||
|
*/
|
||
|
"use strict";
|
||
|
|
||
|
const HarmonyImportDependency = require("./HarmonyImportDependency");
|
||
|
const HarmonyImportSpecifierDependency = require("./HarmonyImportSpecifierDependency");
|
||
|
const HarmonyAcceptImportDependency = require("./HarmonyAcceptImportDependency");
|
||
|
const HarmonyAcceptDependency = require("./HarmonyAcceptDependency");
|
||
|
const HarmonyModulesHelpers = require("./HarmonyModulesHelpers");
|
||
|
|
||
|
module.exports = class HarmonyImportDependencyParserPlugin {
|
||
|
constructor(moduleOptions) {
|
||
|
this.strictExportPresence = moduleOptions.strictExportPresence;
|
||
|
this.strictThisContextOnImports = moduleOptions.strictThisContextOnImports;
|
||
|
}
|
||
|
|
||
|
apply(parser) {
|
||
|
parser.plugin("import", (statement, source) => {
|
||
|
const dep = new HarmonyImportDependency(source, HarmonyModulesHelpers.getNewModuleVar(parser.state, source), statement.range);
|
||
|
dep.loc = statement.loc;
|
||
|
parser.state.current.addDependency(dep);
|
||
|
parser.state.lastHarmonyImport = dep;
|
||
|
return true;
|
||
|
});
|
||
|
parser.plugin("import specifier", (statement, source, id, name) => {
|
||
|
parser.scope.definitions.length--;
|
||
|
parser.scope.renames[`$${name}`] = "imported var";
|
||
|
if(!parser.state.harmonySpecifier) parser.state.harmonySpecifier = {};
|
||
|
parser.state.harmonySpecifier[`$${name}`] = [parser.state.lastHarmonyImport, HarmonyModulesHelpers.getModuleVar(parser.state, source), id];
|
||
|
return true;
|
||
|
});
|
||
|
parser.plugin("expression imported var", (expr) => {
|
||
|
const name = expr.name;
|
||
|
const settings = parser.state.harmonySpecifier[`$${name}`];
|
||
|
const dep = new HarmonyImportSpecifierDependency(settings[0], settings[1], settings[2], name, expr.range, this.strictExportPresence);
|
||
|
dep.shorthand = parser.scope.inShorthand;
|
||
|
dep.directImport = true;
|
||
|
dep.loc = expr.loc;
|
||
|
parser.state.current.addDependency(dep);
|
||
|
return true;
|
||
|
});
|
||
|
parser.plugin("expression imported var.*", (expr) => {
|
||
|
const name = expr.object.name;
|
||
|
const settings = parser.state.harmonySpecifier[`$${name}`];
|
||
|
if(settings[2] !== null)
|
||
|
return false;
|
||
|
const dep = new HarmonyImportSpecifierDependency(settings[0], settings[1], expr.property.name || expr.property.value, name, expr.range, this.strictExportPresence);
|
||
|
dep.shorthand = parser.scope.inShorthand;
|
||
|
dep.directImport = false;
|
||
|
dep.loc = expr.loc;
|
||
|
parser.state.current.addDependency(dep);
|
||
|
return true;
|
||
|
});
|
||
|
if(this.strictThisContextOnImports) {
|
||
|
// only in case when we strictly follow the spec we need a special case here
|
||
|
parser.plugin("call imported var.*", (expr) => {
|
||
|
if(expr.callee.type !== "MemberExpression") return;
|
||
|
if(expr.callee.object.type !== "Identifier") return;
|
||
|
const name = expr.callee.object.name;
|
||
|
const settings = parser.state.harmonySpecifier[`$${name}`];
|
||
|
if(settings[2] !== null)
|
||
|
return false;
|
||
|
const dep = new HarmonyImportSpecifierDependency(settings[0], settings[1], expr.callee.property.name || expr.callee.property.value, name, expr.callee.range, this.strictExportPresence);
|
||
|
dep.shorthand = parser.scope.inShorthand;
|
||
|
dep.directImport = false;
|
||
|
dep.namespaceObjectAsContext = true;
|
||
|
dep.loc = expr.callee.loc;
|
||
|
parser.state.current.addDependency(dep);
|
||
|
if(expr.arguments)
|
||
|
parser.walkExpressions(expr.arguments);
|
||
|
return true;
|
||
|
});
|
||
|
}
|
||
|
parser.plugin("call imported var", (expr) => {
|
||
|
const args = expr.arguments;
|
||
|
const fullExpr = expr;
|
||
|
expr = expr.callee;
|
||
|
if(expr.type !== "Identifier") return;
|
||
|
const name = expr.name;
|
||
|
const settings = parser.state.harmonySpecifier[`$${name}`];
|
||
|
const dep = new HarmonyImportSpecifierDependency(settings[0], settings[1], settings[2], name, expr.range, this.strictExportPresence);
|
||
|
dep.directImport = true;
|
||
|
dep.callArgs = args;
|
||
|
dep.call = fullExpr;
|
||
|
dep.loc = expr.loc;
|
||
|
parser.state.current.addDependency(dep);
|
||
|
if(args)
|
||
|
parser.walkExpressions(args);
|
||
|
return true;
|
||
|
});
|
||
|
parser.plugin("hot accept callback", (expr, requests) => {
|
||
|
const dependencies = requests
|
||
|
.filter(request => HarmonyModulesHelpers.checkModuleVar(parser.state, request))
|
||
|
.map(request => {
|
||
|
const dep = new HarmonyAcceptImportDependency(request, HarmonyModulesHelpers.getModuleVar(parser.state, request), expr.range);
|
||
|
dep.loc = expr.loc;
|
||
|
parser.state.current.addDependency(dep);
|
||
|
return dep;
|
||
|
});
|
||
|
if(dependencies.length > 0) {
|
||
|
const dep = new HarmonyAcceptDependency(expr.range, dependencies, true);
|
||
|
dep.loc = expr.loc;
|
||
|
parser.state.current.addDependency(dep);
|
||
|
}
|
||
|
});
|
||
|
parser.plugin("hot accept without callback", (expr, requests) => {
|
||
|
const dependencies = requests
|
||
|
.filter(request => HarmonyModulesHelpers.checkModuleVar(parser.state, request))
|
||
|
.map(request => {
|
||
|
const dep = new HarmonyAcceptImportDependency(request, HarmonyModulesHelpers.getModuleVar(parser.state, request), expr.range);
|
||
|
dep.loc = expr.loc;
|
||
|
parser.state.current.addDependency(dep);
|
||
|
return dep;
|
||
|
});
|
||
|
if(dependencies.length > 0) {
|
||
|
const dep = new HarmonyAcceptDependency(expr.range, dependencies, false);
|
||
|
dep.loc = expr.loc;
|
||
|
parser.state.current.addDependency(dep);
|
||
|
}
|
||
|
});
|
||
|
}
|
||
|
};
|