Update to allow for much more specific rules for including the nodent

runtime in compiled files. See README.md for more details.

Author: matAtWork

README.md

@@ -46,7 +46,8 @@ With options:
 				"compiler": {
 					"promises": true,
 					"generators": false
-				}
+				},
+				"runtimePattern":null
 			}]
 		]
 	}
@@ -73,10 +74,27 @@ but since much of the parsing is done by Babel some are unused.
 	compiler:{
 		promises:true,    // Use nodent's "Promises" mode. Set to false if your execution environment does not support Promises.
 		generators:false  // Transform into 'Generators'.
-	}
+	},
+	runtimePattern:null   // See below
 
 For more information on the compiler options, see [ES7 and Promises](https://github.com/matatbread/nodent#es7-and-promises) in the nodent documentation.
 
+__runtimePattern__ 
+By default, fast-async will put the nodent runtime into every file it compiles. If your project is made up of more than one file, the constant
+redefinition of the runtime is a waste of time and space. You can specify that you want the runtime in particular file(s) by setting the 'runtimePattern' to a regular expression (in quotes). Only files that match the regular expression will have the runtime defined (which is global, so you only need it once). For example:
+
+	  "babel": {
+	    "plugins": [
+	      "syntax-async-functions",
+	      ["fast-async",{
+			"runtimePattern":"test-input\\.js"
+		  }]
+	    ]
+	  }
+
+Alternatively, if you set runtimePattern to `"directive"`, the statement `"use runtime-nodent";` will be replaced with the runtime during compilation.
+
+
 Useful Links
 ------------
 

package.json

@@ -1,8 +1,8 @@
 {
   "name": "fast-async",
-  "version": "6.0.19",
+  "version": "6.0.20",
   "dependencies": {
-    "nodent": "^2.5.0"
+    "nodent": "^2.5.1"
   },
   "description": "fast-async/await transformer Babel plugin",
   "main": "plugin.js",

plugin.js

@@ -50,7 +50,27 @@ module.exports = function (types) {
 
 					return ast ;
 				}
-				pr.ast.body.unshift(getRuntime('Function.prototype.$asyncbind',Function.prototype.$asyncbind)) ;
+
+				if (!state.opts.runtimePattern) {
+	                pr.ast.body.unshift(getRuntime('Function.prototype.$asyncbind',Function.prototype.$asyncbind)) ;
+				} else {
+				    if (state.opts.runtimePattern==='directive') {
+	                    if (path.node.directives) {
+	                        for (var i=0; i<path.node.directives.length; i++) {
+	                            if (path.node.directives[i].value.type==="DirectiveLiteral" && path.node.directives[i].value.value==="use runtime-nodent") {
+	                                pr.ast.body.unshift(getRuntime('Function.prototype.$asyncbind',Function.prototype.$asyncbind)) ;
+	                                path.node.directives.splice(i,1) ;
+	                                break ;
+	                            }
+	                        }
+	                    }
+				    } else {
+	                    var pattern = new RegExp(state.opts.runtimePattern) ;
+	                    if (state.file.parserOpts.filename.match(pattern)) {
+	                        pr.ast.body.unshift(getRuntime('Function.prototype.$asyncbind',Function.prototype.$asyncbind)) ;
+	                    }
+				    }
+				}
 			}
 		}
 	};

tests/package.json

@@ -1,6 +1,6 @@
 {
   "name": "fast-async-test",
-  "version": "6.0.11",
+  "version": "6.0.12",
   "dependencies": {
     "babel-core": "^6.7.4",
     "babel-plugin-syntax-async-functions": "^6.5.0",
@@ -20,7 +20,9 @@
   "babel": {
     "plugins": [
       "syntax-async-functions",
-      "fast-async"
+      ["fast-async",{
+		"runtimePattern":"directive"
+	  }]
     ]
   }
 }

tests/test.js

@@ -22,7 +22,7 @@ var testCode = require('fs').readFileSync(__dirname+'/test-input.js').toString()
 var babel = require("babel-core") ;
 
 var transformers = {
-	'fast-async':{plugins:[[require('../plugin.js'),{env:{dontMapStackTraces:true},compiler:{promises:true}}]]},
+	'fast-async':{plugins:[[require('../plugin.js'),{runtimePatten:'directive',env:{dontMapStackTraces:true},compiler:{promises:true}}]]},
 };
 
 var requires ;