use relative tools/protect.js filenames

This commit is contained in:
Sebastian McKenzie 2015-06-25 23:20:18 +01:00
parent 2a3142273d
commit 4476e8311c

View File

@ -1,3 +1,5 @@
var path = require("path");
module.exports = function (babel) {
var t = babel.types;
@ -7,8 +9,13 @@ module.exports = function (babel) {
if (file.opts.filename.indexOf("tools/protect") >= 0) return;
if (file.opts.filename.indexOf("templates") >= 0) return;
var from = "/" + path.dirname(file.opts.filename.replace(/^src/, "lib"));
var to = "/lib/babel/tools";
var protectLoc = "./" + path.relative(from, to) + "/protect.js";
this.unshiftContainer("body", [
t.expressionStatement(t.callExpression(file.addImport("babel-core/lib/babel/tools/protect"), [t.identifier("module")]))
t.expressionStatement(t.callExpression(file.addImport(protectLoc), [t.identifier("module")]))
]);
}
}