v5.6.9

Use path-exists module instead of fs.exists.

.babelrc

@@ -2,5 +2,6 @@
   "stage": 0,
   "loose": ["all"],
   "blacklist": ["es6.tailCall"],
-  "optional": ["optimisation.flow.forOf"]
+  "optional": ["optimisation.flow.forOf"],
+  "plugins": ["./tools/build-plugins/protect"]
 }

.eslintrc

@@ -18,7 +18,8 @@
     "new-cap": 0,
     "no-loop-func": 0,
     "no-unreachable": 0,
-    "no-labels": 0
+    "no-labels": 0,
+    "no-process-exit": 0
   },
   "env": {
     "node": true

CHANGELOG.md

@@ -13,6 +13,12 @@ _Note: Gaps between patch versions are faulty/broken releases._
 
 See [CHANGELOG - 6to5](CHANGELOG-6to5.md) for the pre-4.0.0 version changelog.
 
+## 5.6.8
+
+ * **Bug Fix**
+  * Fix binary expressions colliding with unary expression operators in compact mode.
+  * Fix node properties being set to `null` when using computed properties.
+
 ## 5.6.7
 
  * **Bug Fix**
@@ -29,7 +35,7 @@ See [CHANGELOG - 6to5](CHANGELOG-6to5.md) for the pre-4.0.0 version changelog.
 
 ## 5.6.5
 
- * *Bug Fix**
+ * **Bug Fix**
   * Fix nested functions causing rest parameter optimisation to not properly detect when it should deopt on a reference.
  * **Internal**
   * Update Regenerator `0.8.31`.

CONTRIBUTING.md

@@ -113,8 +113,6 @@ your [`$PATH`](http://unix.stackexchange.com/questions/26047/how-to-correctly-ad
 
 + [js-tokens](http://ghub.io/js-tokens) This is used to get tokens for syntax error highlighting.
 
-+ [leven](http://ghub.io/leven) A levenstein algorithm to determine how close a word is to another. This is used to offer suggestions when using the utility.undeclaredVariableCheck transformer.
-
 + [line-numbers](http://ghub.io/line-numbers) Used to produce the code frames in syntax errors.
 
 + [lodash](http://ghub.io/lodash) Used for various utilities.

Makefile

@@ -24,7 +24,7 @@ clean-core:
 	rm -rf lib
 
 lint:
-	eslint src/babel
+	eslint src/babel packages/babel-cli/bin
 
 build:
 	mkdir -p dist

index.js

@@ -0,0 +1 @@
+module.exports = require("./lib/babel/api/node.js");

package.json

@@ -1,12 +1,11 @@
 {
   "name": "babel-core",
   "description": "A compiler for writing next generation JavaScript",
-  "version": "5.6.7",
+  "version": "5.6.9",
   "author": "Sebastian McKenzie <sebmck@gmail.com>",
   "homepage": "https://babeljs.io/",
   "license": "MIT",
   "repository": "babel/babel",
-  "main": "lib/babel/api/node.js",
   "browser": {
     "./lib/babel/api/register/node.js": "./lib/babel/api/register/browser.js"
   },
@@ -57,11 +56,11 @@
     "home-or-tmp": "^1.0.0",
     "is-integer": "^1.0.4",
     "js-tokens": "1.0.1",
-    "leven": "^1.0.1",
     "line-numbers": "0.2.0",
     "lodash": "^3.6.0",
     "minimatch": "^2.0.3",
     "output-file-sync": "^1.1.0",
+    "path-exists": "^1.0.0",
     "path-is-absolute": "^1.0.0",
     "private": "^0.1.6",
     "regenerator": "0.8.31",
@@ -77,7 +76,7 @@
     "trim-right": "^1.0.0"
   },
   "devDependencies": {
-    "babel": "5.5.7",
+    "babel": "5.6.7",
     "browserify": "^9.0.8",
     "chai": "^2.2.0",
     "eslint": "^0.21.2",

packages/babel-cli/bin/babel-plugin/index.js

@@ -1,9 +1,10 @@
 #!/usr/bin/env node
 
-var readline = require("readline");
-var child    = require("child_process");
-var path     = require("path");
-var fs       = require("fs");
+var pathExists = require("path-exists");
+var readline   = require("readline");
+var child      = require("child_process");
+var path       = require("path");
+var fs         = require("fs");
 
 function spawn(cmd, args, callback) {
   console.log(">", cmd, args);
@@ -34,7 +35,7 @@ function spawnMultiple(cmds) {
 }
 
 function template(name, data) {
-  var source = fs.readFileSync(__dirname + "/templates/" + name, "utf8");
+  var source = fs.readFileSync(path.join(__dirname, "templates", name), "utf8");
   source = source.replace(/[A-Z_]+/g, function (key) {
     return data[key] === undefined ? key : data[key];
   });
@@ -98,7 +99,7 @@ var cmds = {
 
         write("README.md", template("README.md", templateData));
 
-        if (!fs.existsSync("src")) {
+        if (!pathExists.sync("src")) {
           fs.mkdirSync("src");
           write("src/index.js", template("index.js", templateData));
         }

packages/babel-cli/bin/babel/dir.js

@@ -1,4 +1,5 @@
 var outputFileSync = require("output-file-sync");
+var pathExists     = require("path-exists");
 var chokidar       = require("chokidar");
 var slash          = require("slash");
 var path           = require("path");
@@ -6,7 +7,7 @@ var util           = require("./util");
 var fs             = require("fs");
 var _              = require("lodash");
 
-module.exports = function (commander, filenames, opts) {
+module.exports = function (commander, filenames) {
   var write = function (src, relative) {
     // remove extension and then append back on .js
     relative = relative.replace(/\.(\w*?)$/, "") + ".js";
@@ -41,7 +42,7 @@ module.exports = function (commander, filenames, opts) {
   };
 
   var handle = function (filename) {
-    if (!fs.existsSync(filename)) return;
+    if (!pathExists.sync(filename)) return;
 
     var stat = fs.statSync(filename);
 

packages/babel-cli/bin/babel/file.js

@@ -1,4 +1,5 @@
 var convertSourceMap = require("convert-source-map");
+var pathExists       = require("path-exists");
 var sourceMap        = require("source-map");
 var chokidar         = require("chokidar");
 var slash            = require("slash");
@@ -100,7 +101,7 @@ module.exports = function (commander, filenames, opts) {
     results = [];
 
     _.each(filenames, function (filename) {
-      if (!fs.existsSync(filename)) return;
+      if (!pathExists.sync(filename)) return;
 
       var stat = fs.statSync(filename);
       if (stat.isDirectory()) {

packages/babel-cli/bin/babel/index.js

@@ -1,6 +1,7 @@
 #!/usr/bin/env node
 
 var moduleFormatters = require("babel-core/lib/babel/transformation/modules");
+var pathExists       = require("path-exists");
 var commander        = require("commander");
 var transform        = require("babel-core").transform;
 var kebabCase        = require("lodash/string/kebabCase");
@@ -9,7 +10,6 @@ var util             = require("babel-core").util;
 var uniq             = require("lodash/array/uniq");
 var each             = require("lodash/collection/each");
 var keys             = require("lodash/object/keys");
-var fs               = require("fs");
 var glob             = require("glob");
 
 each(options, function (option, key) {
@@ -88,7 +88,7 @@ var filenames = commander.args.reduce(function (globbed, input) {
 filenames = uniq(filenames);
 
 each(filenames, function (filename) {
-  if (!fs.existsSync(filename)) {
+  if (!pathExists.sync(filename)) {
     errors.push(filename + " doesn't exist");
   }
 });

packages/babel-cli/package.json

@@ -1,14 +1,14 @@
 {
   "name": "babel",
   "description": "Turn ES6 code into readable vanilla ES5 with source maps",
-  "version": "5.6.6",
+  "version": "5.6.8",
   "author": "Sebastian McKenzie <sebmck@gmail.com>",
   "homepage": "https://babeljs.io/",
   "license": "MIT",
   "repository": "babel/babel",
   "preferGlobal": true,
   "dependencies": {
-    "babel-core": "^5.6.6",
+    "babel-core": "^5.6.8",
     "chokidar": "^1.0.0",
     "commander": "^2.6.0",
     "convert-source-map": "^1.1.0",
@@ -16,6 +16,7 @@
     "glob": "^5.0.5",
     "lodash": "^3.2.0",
     "output-file-sync": "^1.1.0",
+    "path-exists": "^1.0.0",
     "path-is-absolute": "^1.0.0",
     "source-map": "^0.4.0",
     "slash": "^1.0.0"

packages/babel-runtime/package.json

@@ -1,7 +1,7 @@
 {
   "name": "babel-runtime",
   "description": "babel selfContained runtime",
-  "version": "5.6.6",
+  "version": "5.6.8",
   "license": "MIT",
   "repository": "babel/babel",
   "author": "Sebastian McKenzie <sebmck@gmail.com>",

src/babel/api/register/cache.js

@@ -1,6 +1,7 @@
 import path from "path";
 import fs from "fs";
 import homeOrTmp from "home-or-tmp";
+import pathExists from "path-exists";
 
 const FILENAME = process.env.BABEL_CACHE_PATH || path.join(homeOrTmp, ".babel.json");
 var data = {};
@@ -15,7 +16,7 @@ export function load() {
   process.on("exit", save);
   process.nextTick(save);
 
-  if (!fs.existsSync(FILENAME)) return;
+  if (!pathExists.sync(FILENAME)) return;
 
   try {
     data = JSON.parse(fs.readFileSync(FILENAME));

src/babel/generation/generators/classes.js

@@ -1,5 +1,5 @@
 export function ClassDeclaration(node, print) {
-  print.list(node.decorators);
+  print.list(node.decorators, { separator: "" });
   this.push("class");
 
   if (node.id) {
@@ -43,7 +43,7 @@ export function ClassBody(node, print) {
 
 
 export function ClassProperty(node, print) {
-  print.list(node.decorators);
+  print.list(node.decorators, { separator: "" });
 
   if (node.static) this.push("static ");
   print.plain(node.key);
@@ -58,7 +58,7 @@ export function ClassProperty(node, print) {
 }
 
 export function MethodDefinition(node, print) {
-  print.list(node.decorators);
+  print.list(node.decorators, { separator: "" });
 
   if (node.static) {
     this.push("static ");

src/babel/generation/generators/expressions.js

@@ -140,6 +140,7 @@ export function AssignmentExpression(node, print) {
   print.plain(node.left);
 
   var spaces = node.operator === "in" || node.operator === "instanceof";
+  spaces = true; // todo: https://github.com/babel/babel/issues/1835
   this.space(spaces);
 
   this.push(node.operator);

src/babel/generation/generators/types.js

@@ -33,6 +33,8 @@ export function ObjectExpression(node, print) {
 export { ObjectExpression as ObjectPattern };
 
 export function Property(node, print) {
+  print.list(node.decorators, { separator: "" });
+
   if (node.method || node.kind === "get" || node.kind === "set") {
     this._method(node, print);
   } else {

src/babel/tools/protect.js

@@ -0,0 +1,9 @@
+import path from "path";
+
+var root = path.resolve(__dirname, "../../../");
+
+export default function (module) {
+  if (module.parent && module.parent.filename.indexOf(root) !== 0) {
+    throw new Error("Don't hotlink internal Babel files.");
+  }
+}

src/babel/transformation/file/options/resolve-rc.js

@@ -3,16 +3,15 @@ import { normaliseOptions } from "./index";
 import merge from "../../../helpers/merge";
 import path from "path";
 import fs from "fs";
+import pathExists from "path-exists";
 
 var cache = {};
 var jsons = {};
 
 function exists(filename) {
-  if (!fs.existsSync) return false;
-
   var cached = cache[filename];
   if (cached != null) return cached;
-  return cache[filename] = fs.existsSync(filename);
+  return cache[filename] = pathExists.sync(filename);
 }
 
 export default function (loc, opts = {}) {

src/babel/transformation/templates/helper-define-property.js

@@ -1,8 +1,18 @@
 (function (obj, key, value) {
-  return Object.defineProperty(obj, key, {
-    value: value,
-    enumerable: true,
-    configurable: true,
-    writable: true
-  });
+  // Shortcircuit the slow defineProperty path when possible.
+  // We are trying to avoid issues where setters defined on the
+  // prototype cause side effects under the fast path of simple
+  // assignment. By checking for existence of the property with
+  // the in operator, we can optimize most of this overhead away.
+  if (key in obj) {
+    Object.defineProperty(obj, key, {
+      value: value,
+      enumerable: true,
+      configurable: true,
+      writable: true
+    });
+  } else {
+    obj[key] = value;
+  }
+  return obj;
 });

src/babel/transformation/templates/helper-sliced-to-array.js

@@ -1,7 +1,7 @@
-(function (arr, i) {
-  if (Array.isArray(arr)) {
-    return arr;
-  } else if (Symbol.iterator in Object(arr)) {
+(function () {
+  // Broken out into a separate function to avoid deoptimizations due to the try/catch for the
+  // array iterator case.
+  function sliceIterator(arr, i) {
     // this is an expanded form of `for...of` that properly supports abrupt completions of
     // iterators etc. variable names have been minimised to reduce the size of this massive
     // helper. sometimes spec compliancy is annoying :(
@@ -32,7 +32,15 @@
       }
     }
     return _arr;
-  } else {
-    throw new TypeError("Invalid attempt to destructure non-iterable instance");
   }
-});
+
+  return function (arr, i) {
+    if (Array.isArray(arr)) {
+      return arr;
+    } else if (Symbol.iterator in Object(arr)) {
+      return sliceIterator(arr, i);
+    } else {
+      throw new TypeError("Invalid attempt to destructure non-iterable instance");
+    }
+  };
+})();

src/babel/transformation/transformers/es6/properties.computed.js

@@ -2,8 +2,6 @@ import * as t from "../../../types";
 
 function loose(node, body, objId) {
   for (var prop of (node.properties: Array)) {
-    if (!prop) continue;
-
     body.push(t.expressionStatement(
       t.assignmentExpression(
         "=",
@@ -19,8 +17,6 @@ function spec(node, body, objId, initProps, file) {
   // otherwise use Object.defineProperty
 
   for (let prop of (node.properties: Array)) {
-    if (!prop) continue;
-
     // this wont work with Object.defineProperty
     if (t.isLiteral(t.toComputedKey(prop), { value: "__proto__" })) {
       initProps.push(prop);
@@ -67,17 +63,18 @@ export var visitor = {
       var initProps = [];
       var stopInits = false;
 
-      for (var i = 0; i < node.properties.length; i++) {
-        let prop = node.properties[i];
+      node.properties = node.properties.filter(function (prop) {
         if (prop.computed) {
           stopInits = true;
         }
 
         if (prop.kind !== "init" || !stopInits) {
           initProps.push(prop);
-          node.properties[i] = null;
+          return false;
+        } else {
+          return true;
         }
-      }
+      });
 
       //
 

src/babel/traversal/path/inference/index.js

@@ -1,3 +1,4 @@
+import type NodePath from "./index";
 import * as inferers from "./inferers";
 import * as t from "../../../types";
 
@@ -76,6 +77,8 @@ function _isBaseType(baseName: string, type?, soft?): boolean {
     return t.isAnyTypeAnnotation(type);
   } else if (baseName === "mixed") {
     return t.isMixedTypeAnnotation(type);
+  } else if (baseName === "void") {
+    return t.isVoidTypeAnnotation(type);
   } else {
     if (soft) {
       return false;
@@ -105,6 +108,19 @@ export function couldBeBaseType(name: string): boolean {
   }
 }
 
+/**
+ * Description
+ */
+
+export function baseTypeStrictlyMatches(right: NodePath) {
+  var left = this.getTypeAnnotation();
+  right = right.getTypeAnnotation();
+
+  if (!t.isAnyTypeAnnotation() && t.isFlowBaseAnnotation(left)) {
+    return right.type === left.type;
+  }
+}
+
 /**
  * Description
  */

src/babel/traversal/path/lib/hoister.js

@@ -122,7 +122,7 @@ export default class PathHoister {
     if (parent.isJSXElement() && this.path.container === parent.node.children) {
       // turning the `span` in `<div><span /></div>` to an expression so we need to wrap it with
       // an expression container
-      uid = t.jSXExpressionContainer(uid);
+      uid = t.JSXExpressionContainer(uid);
     }
 
     this.path.replaceWith(uid);

src/babel/types/flow.js

@@ -112,5 +112,9 @@ export function createTypeAnnotationBasedOnTypeof(type) {
     return t.genericTypeAnnotation(t.identifier("Function"));
   } else if (type === "object") {
     return t.genericTypeAnnotation(t.identifier("Object"));
+  } else if (type === "symbol") {
+    return t.genericTypeAnnotation(t.identifier("Symbol"));
+  } else {
+    throw new Error("Invalid typeof value");
   }
 }

src/babel/types/index.js

@@ -108,7 +108,7 @@ each(t.VISITOR_KEYS, function (keys, type) {
 });
 
 each(t.BUILDER_KEYS, function (keys, type) {
-  t[type[0].toLowerCase() + type.slice(1)] = function () {
+  var builder = function () {
     var node = {};
     node.start = null;
     node.type = type;
@@ -123,6 +123,9 @@ each(t.BUILDER_KEYS, function (keys, type) {
 
     return node;
   };
+
+  t[type] = builder;
+  t[type[0].toLowerCase() + type.slice(1)] = builder;
 });
 
 /*

src/babel/util.js

@@ -18,6 +18,7 @@ import has from "lodash/object/has";
 import fs from "fs";
 import * as t from "./types";
 import slash from "slash";
+import pathExists from "path-exists";
 
 export { inherits, inspect } from "util";
 
@@ -182,7 +183,7 @@ function loadTemplates() {
   var templates = {};
 
   var templatesLoc = path.join(__dirname, "transformation/templates");
-  if (!fs.existsSync(templatesLoc)) {
+  if (!pathExists.sync(templatesLoc)) {
     throw new ReferenceError(messages.get("missingTemplatesDirectory"));
   }
 

test/core/_helper.js

@@ -1,8 +1,9 @@
-var esvalid = require("esvalid");
-var util    = require("../../lib/babel/util");
-var path    = require("path");
-var fs      = require("fs");
-var _       = require("lodash");
+var pathExists = require("path-exists");
+var esvalid    = require("esvalid");
+var util       = require("../../lib/babel/util");
+var path       = require("path");
+var fs         = require("fs");
+var _          = require("lodash");
 
 var humanize = function (val, noext) {
   if (noext) val = path.basename(val, path.extname(val));
@@ -10,7 +11,7 @@ var humanize = function (val, noext) {
 };
 
 var readFile = exports.readFile = function (filename) {
-  if (fs.existsSync(filename)) {
+  if (pathExists.sync(filename)) {
     var file = fs.readFileSync(filename, "utf8").trim();
     file = file.replace(/\r\n/g, "\n");
     return file;
@@ -31,7 +32,7 @@ exports.esvalid = function (ast, code, loc) {
 };
 
 exports.assertVendor = function (name) {
-  if (!fs.existsSync(__dirname + "/../../vendor/" + name)) {
+  if (!pathExists.sync(__dirname + "/../../vendor/" + name)) {
     console.error("No vendor/" + name + " - run `make bootstrap`");
     process.exit(1);
   }
@@ -130,13 +131,13 @@ exports.get = function (entryName, entryLoc) {
       suite.tests.push(test);
 
       var sourceMappingsLoc = taskDir + "/source-mappings.json";
-      if (fs.existsSync(sourceMappingsLoc)) {
+      if (pathExists.sync(sourceMappingsLoc)) {
         test.options.sourceMap = true;
         test.sourceMappings = require(sourceMappingsLoc);
       }
 
       var sourceMap = taskDir + "/source-map.json";
-      if (fs.existsSync(sourceMap)) {
+      if (pathExists.sync(sourceMap)) {
         test.options.sourceMap = true;
         test.sourceMap = require(sourceMap);
       }

test/core/api.js

@@ -219,7 +219,7 @@ suite("api", function () {
 
     function getModuleNameTest(moduleFormat, expected) {
       var result = transform("foo('bar');", {
-        filename: "foo/bar/index",
+        filename: "/foo/bar/index",
         modules: moduleFormat,
         moduleIds: true,
         getModuleId: function (name) {
@@ -232,7 +232,7 @@ suite("api", function () {
 
     test("amd", function () {
       var expected = [
-        "define('foo/bar', ['exports'], function (exports) {",
+        "define('/foo/bar', ['exports'], function (exports) {",
         "  'use strict';",
         "",
         "  foo('bar');",
@@ -246,7 +246,7 @@ suite("api", function () {
       var expected = [
         "(function (global, factory) {",
         "  if (typeof define === 'function' && define.amd) {",
-        "    define('foo/bar', ['exports'], factory);",
+        "    define('/foo/bar', ['exports'], factory);",
         "  } else if (typeof exports !== 'undefined') {",
         "    factory(exports);",
         "  } else {",
@@ -268,7 +268,7 @@ suite("api", function () {
 
     test("system", function () {
       var expected = [
-        "System.register('foo/bar', [], function (_export) {",
+        "System.register('/foo/bar', [], function (_export) {",
         "  'use strict';",
         "",
         "  return {",

test/core/bin.js

@@ -9,6 +9,7 @@ var child          = require("child_process");
 var path           = require("path");
 var chai           = require("chai");
 var fs             = require("fs");
+var pathExists     = require("path-exists");
 var _              = require("lodash");
 
 var fixtureLoc = __dirname + "/fixtures/bin";
@@ -16,7 +17,7 @@ var tmpLoc = __dirname + "/tmp";
 
 var readDir = function (loc) {
   var files = {};
-  if (fs.existsSync(loc)) {
+  if (pathExists.sync(loc)) {
     _.each(readdir(loc), function (filename) {
       var contents = helper.readFile(loc + "/" + filename);
       files[filename] = contents;
@@ -112,7 +113,7 @@ var buildTest = function (binName, testName, opts) {
 
 var clear = function () {
   process.chdir(__dirname);
-  if (fs.existsSync(tmpLoc)) rimraf.sync(tmpLoc);
+  if (pathExists.sync(tmpLoc)) rimraf.sync(tmpLoc);
   fs.mkdirSync(tmpLoc);
   process.chdir(tmpLoc);
 };
@@ -132,11 +133,11 @@ _.each(fs.readdirSync(fixtureLoc), function (binName) {
       };
 
       var optionsLoc = testLoc + "/options.json"
-      if (fs.existsSync(optionsLoc)) _.merge(opts, require(optionsLoc));
+      if (pathExists.sync(optionsLoc)) _.merge(opts, require(optionsLoc));
 
       _.each(["stdout", "stdin", "stderr"], function (key) {
         var loc = testLoc + "/" + key + ".txt";
-        if (fs.existsSync(loc)) {
+        if (pathExists.sync(loc)) {
           opts[key] = helper.readFile(loc);
         } else {
           opts[key] = opts[key] || "";

test/core/fixtures/generation/types/Decorator/actual.js

@@ -0,0 +1,31 @@
+var obj = {
+  @foo
+  @bar
+  foo: "bar",
+
+  @foo
+  @bar
+  foo() {},
+
+  @foo
+  get foo() {},
+
+  @bar
+  set bar(foo) {}
+};
+
+class Foo {
+  @foo
+  @bar
+  foo() {}
+
+  @foo
+  @bar
+  foo() {}
+
+  @foo
+  get foo() {}
+
+  @bar
+  set bar(foo) {}
+}

test/core/fixtures/generation/types/Decorator/expected.js

@@ -0,0 +1,31 @@
+var obj = {
+  @foo
+  @bar
+  foo: "bar",
+
+  @foo
+  @bar
+  foo() {},
+
+  @foo
+  get foo() {},
+
+  @bar
+  set bar(foo) {}
+};
+
+class Foo {
+  @foo
+  @bar
+  foo() {}
+
+  @foo
+  @bar
+  foo() {}
+
+  @foo
+  get foo() {}
+
+  @bar
+  set bar(foo) {}
+}

test/core/generation.js

@@ -32,6 +32,7 @@ _.each(helper.get("generation"), function (testSuite) {
           strictMode: false,
           sourceType: "module",
           features: {
+            "es7.decorators": true,
             "es7.comprehensions": true,
             "es7.asyncFunctions": true,
             "es7.exportExtensions": true,

test/core/util.js

@@ -101,6 +101,11 @@ suite("util", function () {
     assert.strictEqual(util.booleanify("inline"), "inline");
   });
 
+  test("resolve", function () {
+    assert.notEqual(util.resolve("is-integer").indexOf("node_modules/is-integer"), -1);
+    assert.equal(util.resolve("foo-bar-module"), null);
+  });
+
   test("toIdentifier", function () {
     assert.equal(t.toIdentifier(t.identifier("swag")), "swag");
     assert.equal(t.toIdentifier("swag-lord"), "swagLord");

tools/build-plugins/protect.js

@@ -0,0 +1,23 @@
+var path = require("path");
+
+module.exports = function (babel) {
+  var t = babel.types;
+
+  return new babel.Plugin("protect-internal-apis", {
+    visitor: {
+      Program: function (node, parent, scope, file) {
+        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(protectLoc), [t.identifier("module")]))
+        ]);
+      }
+    }
+  });
+}