Include babel syntax plugins when tree shaking

main.js

@@ -1216,6 +1216,25 @@ addStealExtension(function(loader) {
 		};
 	}
 
+	// Collect syntax plugins, because we need to always include these.
+	var getSyntaxPlugins = (function(){
+		var plugins;
+		return function(babel) {
+			if(!plugins) {
+				plugins = [];
+				for(var p in babel.availablePlugins) {
+					if(p.indexOf("syntax-") === 0) {
+						plugins.push(babel.availablePlugins[p]);
+					}
+				}
+			}
+			return plugins;
+		};
+	})();
+
+
+
+
 	function applyBabelPlugin(load) {
 		var loader = this;
 		var pluginLoader = loader.pluginLoader || loader;
@@ -1225,9 +1244,8 @@ addStealExtension(function(loader) {
 			var babel = transpiler.Babel || transpiler.babel || transpiler;
 
 			try {
-				var babelPlugins = [
-					loader._getImportSpecifierPositionsPlugin.bind(null, load)
-				];
+				var babelPlugins = [].concat(getSyntaxPlugins(babel));
+				babelPlugins.push(loader._getImportSpecifierPositionsPlugin.bind(null, load));
 				if(treeShakingEnabled(loader, load)) {
 					babelPlugins.push(treeShakePlugin.bind(null, loader, load));
 				}

src/extension-tree-shaking.js

@@ -273,6 +273,25 @@ addStealExtension(function(loader) {
 		};
 	}
 
+	// Collect syntax plugins, because we need to always include these.
+	var getSyntaxPlugins = (function(){
+		var plugins;
+		return function(babel) {
+			if(!plugins) {
+				plugins = [];
+				for(var p in babel.availablePlugins) {
+					if(p.indexOf("syntax-") === 0) {
+						plugins.push(babel.availablePlugins[p]);
+					}
+				}
+			}
+			return plugins;
+		};
+	})();
+
+
+
+
 	function applyBabelPlugin(load) {
 		var loader = this;
 		var pluginLoader = loader.pluginLoader || loader;
@@ -282,9 +301,8 @@ addStealExtension(function(loader) {
 			var babel = transpiler.Babel || transpiler.babel || transpiler;
 
 			try {
-				var babelPlugins = [
-					loader._getImportSpecifierPositionsPlugin.bind(null, load)
-				];
+				var babelPlugins = [].concat(getSyntaxPlugins(babel));
+				babelPlugins.push(loader._getImportSpecifierPositionsPlugin.bind(null, load));
 				if(treeShakingEnabled(loader, load)) {
 					babelPlugins.push(treeShakePlugin.bind(null, loader, load));
 				}

steal-with-promises.js

@@ -7380,6 +7380,25 @@ addStealExtension(function(loader) {
 		};
 	}
 
+	// Collect syntax plugins, because we need to always include these.
+	var getSyntaxPlugins = (function(){
+		var plugins;
+		return function(babel) {
+			if(!plugins) {
+				plugins = [];
+				for(var p in babel.availablePlugins) {
+					if(p.indexOf("syntax-") === 0) {
+						plugins.push(babel.availablePlugins[p]);
+					}
+				}
+			}
+			return plugins;
+		};
+	})();
+
+
+
+
 	function applyBabelPlugin(load) {
 		var loader = this;
 		var pluginLoader = loader.pluginLoader || loader;
@@ -7389,9 +7408,8 @@ addStealExtension(function(loader) {
 			var babel = transpiler.Babel || transpiler.babel || transpiler;
 
 			try {
-				var babelPlugins = [
-					loader._getImportSpecifierPositionsPlugin.bind(null, load)
-				];
+				var babelPlugins = [].concat(getSyntaxPlugins(babel));
+				babelPlugins.push(loader._getImportSpecifierPositionsPlugin.bind(null, load));
 				if(treeShakingEnabled(loader, load)) {
 					babelPlugins.push(treeShakePlugin.bind(null, loader, load));
 				}

steal.js

@@ -6110,6 +6110,25 @@ addStealExtension(function(loader) {
 		};
 	}
 
+	// Collect syntax plugins, because we need to always include these.
+	var getSyntaxPlugins = (function(){
+		var plugins;
+		return function(babel) {
+			if(!plugins) {
+				plugins = [];
+				for(var p in babel.availablePlugins) {
+					if(p.indexOf("syntax-") === 0) {
+						plugins.push(babel.availablePlugins[p]);
+					}
+				}
+			}
+			return plugins;
+		};
+	})();
+
+
+
+
 	function applyBabelPlugin(load) {
 		var loader = this;
 		var pluginLoader = loader.pluginLoader || loader;
@@ -6119,9 +6138,8 @@ addStealExtension(function(loader) {
 			var babel = transpiler.Babel || transpiler.babel || transpiler;
 
 			try {
-				var babelPlugins = [
-					loader._getImportSpecifierPositionsPlugin.bind(null, load)
-				];
+				var babelPlugins = [].concat(getSyntaxPlugins(babel));
+				babelPlugins.push(loader._getImportSpecifierPositionsPlugin.bind(null, load));
 				if(treeShakingEnabled(loader, load)) {
 					babelPlugins.push(treeShakePlugin.bind(null, loader, load));
 				}

test/steal-with-promises.js

@@ -7380,6 +7380,25 @@ addStealExtension(function(loader) {
 		};
 	}
 
+	// Collect syntax plugins, because we need to always include these.
+	var getSyntaxPlugins = (function(){
+		var plugins;
+		return function(babel) {
+			if(!plugins) {
+				plugins = [];
+				for(var p in babel.availablePlugins) {
+					if(p.indexOf("syntax-") === 0) {
+						plugins.push(babel.availablePlugins[p]);
+					}
+				}
+			}
+			return plugins;
+		};
+	})();
+
+
+
+
 	function applyBabelPlugin(load) {
 		var loader = this;
 		var pluginLoader = loader.pluginLoader || loader;
@@ -7389,9 +7408,8 @@ addStealExtension(function(loader) {
 			var babel = transpiler.Babel || transpiler.babel || transpiler;
 
 			try {
-				var babelPlugins = [
-					loader._getImportSpecifierPositionsPlugin.bind(null, load)
-				];
+				var babelPlugins = [].concat(getSyntaxPlugins(babel));
+				babelPlugins.push(loader._getImportSpecifierPositionsPlugin.bind(null, load));
 				if(treeShakingEnabled(loader, load)) {
 					babelPlugins.push(treeShakePlugin.bind(null, loader, load));
 				}

test/tree_shake/syntax.html

@@ -0,0 +1,45 @@
+<!doctype html>
+<html lang="en">
+<head>
+	<meta charset="UTF-8">
+	<title>Tree shaking with advanced syntax</title>
+</head>
+<body>
+	<script>
+		window.assert = window.parent.assert;
+		window.done = window.parent.done;
+	</script>
+	<script
+		src="../../steal-with-promises.js"
+		data-base-url="."
+		data-config="package.json!npm"></script>
+	<script>
+		function assertSyntax() {
+			var mod = steal.loader.get("shake@1.0.0#mod");
+			var modB = steal.loader.get("shake@1.0.0#mod_b");
+
+			if(typeof window.assert !== "undefined") {
+				window.assert.equal(mod.a, "a", "a was not tree shaken");
+				window.assert.equal(mod.b, undefined, "b was tree shaken");
+				window.assert.equal(mod.d, "d", "d was not tree shaken");
+
+				window.assert.equal(modB, undefined, "b was not loaded at all");
+			} else {
+				console.log("mod", mod);
+				console.log("mod_b", modB);
+			}
+		}
+
+		steal.done()
+		.then(function() {
+			return steal.import("~/syntax_main");
+		})
+		.then(assertSyntax)
+		.then(function(){
+			if(window.assert) {
+				window.done();
+			}
+		});
+	</script>
+</body>
+</html>

test/tree_shake/syntax.js

@@ -0,0 +1,8 @@
+import { a, c } from "./mod";
+
+const obj = { a: 'b' };
+const another = { ...obj };
+
+export default function() {
+
+}

test/tree_shake/syntax_main.js

@@ -0,0 +1,4 @@
+import { a } from "./mod";
+import s from "./syntax";
+
+window.SYNTAX_APP = { a, s };