Add plugins parameter to Babel compiler.

Author: Andrey Fedoseev

CHANGES.rst

@@ -12,6 +12,7 @@ Dev
 - Add source maps support for Babel
 - Add `Handlebars <http://handlebarsjs.com/>`_ compiler
 - Add support for Django 1.9
+- Add ``plugins`` parameter to Babel compiler
 
 
 1.0.1

README.rst

@@ -194,13 +194,13 @@ Babel
 ``sourcemap_enabled``
   Boolean. Set to ``True`` to enable source maps. Default: ``False``
 
-``modules``
-  Babel `modules <https://babeljs.io/docs/usage/modules/>`_ command line option. Default: ``None`` (uses Babel's default option).
+``plugins``
+  Babel `plugins <http://babeljs.io/docs/plugins/>`_ command line option. Default: ``None`` (uses Babel's default option).
 
 Example::
 
     STATIC_PRECOMPILER_COMPILERS = (
-        ('static_precompiler.compilers.Babel', {"executable": "/usr/bin/babel", "sourcemap_enabled": True, "modules": "amd"}),
+        ('static_precompiler.compilers.Babel', {"executable": "/usr/bin/babel", "sourcemap_enabled": True, "plugins": "transform-react-jsx"}),
     )
 
 

circle.yml

@@ -3,7 +3,7 @@ dependencies:
     - sudo apt-get install python2.7-dev python3.4-dev
     - npm install -g coffee-script@1.7.1
     - npm install -g less@1.7.4
-    - npm install -g babel@4.3.0
+    - npm install -g babel@6.2.0
     - npm install -g stylus@0.50.0
     - npm install -g handlebars@4.0.2
     - gem install sass -v 3.4.9

static_precompiler/compilers/babel.py

@@ -1,6 +1,7 @@
 import json
 import os
 import posixpath
+import warnings
 
 from static_precompiler import exceptions, utils
 
@@ -17,23 +18,34 @@ class Babel(base.BaseCompiler):
     input_extension = "es6"
     output_extension = "js"
 
-    def __init__(self, executable="babel", sourcemap_enabled=False, modules=None):
+    def __init__(self, executable="babel", sourcemap_enabled=False, modules=None, plugins=None):
         self.executable = executable
         self.is_sourcemap_enabled = sourcemap_enabled
+        if modules:
+            warnings.warn("'modules' option is removed in Babel 6.0. Use `plugins` instead.", DeprecationWarning)
         self.modules = modules
+        self.plugins = plugins
         super(Babel, self).__init__()
 
+    def get_extra_args(self):
+        args = []
+
+        if self.modules is not None:
+            args += ["--modules", self.modules]
+
+        if self.plugins is not None:
+            args += ["--plugins", self.plugins]
+
+        return args
+
     def compile_file(self, source_path):
         args = [
             self.executable,
-        ]
+        ] + self.get_extra_args()
 
         if self.is_sourcemap_enabled:
             args.append("-s")
 
-        if self.modules is not None:
-            args.extend(["--modules", self.modules])
-
         full_output_path = self.get_full_output_path(source_path)
 
         full_output_dirname = os.path.dirname(full_output_path)
@@ -66,11 +78,8 @@ def compile_file(self, source_path):
 
     def compile_source(self, source):
         args = [
-            self.executable
-        ]
-
-        if self.modules is not None:
-            args.extend(["--modules", self.modules])
+            self.executable,
+        ] + self.get_extra_args()
 
         out, errors = utils.run_command(args, source)
         if errors:

static_precompiler/tests/static/scripts/broken.es6

@@ -0,0 +1 @@
+console.log("Hello, World!"

static_precompiler/tests/test_babel.py

@@ -20,10 +20,10 @@ def test_compile_file(monkeypatch, tmpdir):
     assert os.path.exists(full_output_path)
 
     with open(full_output_path) as compiled:
-        assert compiled.read() == """"use strict";
+        assert compiled.read() == """console.log("Hello, World!");\n"""
 
-console.log("Hello, World!");
-"""
+    with pytest.raises(exceptions.StaticCompilationError):
+        compiler.compile_file("scripts/broken.es6")
 
 
 def test_sourcemap(monkeypatch, tmpdir):
@@ -46,7 +46,7 @@ def test_compile_source():
 
     assert (
         clean_javascript(compiler.compile_source('console.log("Hello, World!");')) ==
-        """"use strict";\nconsole.log("Hello, World!");"""
+        """console.log("Hello, World!");"""
     )
 
     with pytest.raises(exceptions.StaticCompilationError):
@@ -55,5 +55,15 @@ def test_compile_source():
     # Test non-ascii
     assert (
         clean_javascript(compiler.compile_source('console.log("Привет, Мир!");')) ==
-        """"use strict";\nconsole.log("Привет, Мир!");"""
+        """console.log("Привет, Мир!");"""
     )
+
+
+def test_get_extra_args():
+
+    compiler = compilers.Babel(modules="foo", plugins="bar")
+
+    assert compiler.get_extra_args() == [
+        "--modules", "foo",
+        "--plugins", "bar",
+    ]