--prepend-code option

Author: ConnorShride

_run.js

@@ -111,6 +111,7 @@ if (tasks.length > 1) // If batch mode
 	else
 		console.log(`Analyzing ${tasks.length} items with ${q.concurrency} threads (use --threads to change this value)`)
 
+// queue the input files for analysis
 const outputDir = argv["output-dir"] || "./";
 tasks.forEach(({filepath, filename}) => q.push(cb => analyze(filepath, filename, cb)));
 
@@ -125,7 +126,9 @@ q.on("success", () => {
 q.start();
 
 function analyze(filepath, filename, cb) {
+
 	let directory = path.join(outputDir, filename + ".results");
+
 	// Find a suitable directory name
 	for (let i = 1; fs.existsSync(directory); i++)
 		directory = path.join(outputDir, filename + "." + i + ".results");

analyze.js

@@ -307,6 +307,35 @@ cc decoder.c -o decoder
 }
 
 code = rewrite(code);
+
+// prepend extra JS containing mock objects in the given file(s) onto the code
+if (argv["prepended-code"]) {
+
+	var prependedCode = ""
+	var files = []
+
+	// get all the files in the directory and sort them alphebetically
+	if (fs.lstatSync(argv["prepended-code"]).isDirectory()) {
+
+		dir_files = fs.readdirSync(argv["prepended-code"]); 
+		for (var i = 0; i < dir_files.length; i++) {
+			files.push(path.join(argv["prepended-code"], dir_files[i]))
+		}
+
+		// make sure we're adding mock code in the right order
+		files.sort()
+	}
+	else {
+		files.push(argv["prepended-code"])
+	}
+
+	for (var i = 0; i < files.length; i++) {
+		prependedCode += fs.readFileSync(files[i], 'utf-8') + "\n\n"
+	}
+
+	code = prependedCode + "\n\n" + code
+}
+
 lib.logJS(code);
 code = fs.readFileSync(path.join(__dirname, "patch.js"), "utf8") + code;
 

argv.js

@@ -41,7 +41,6 @@ if (argv._unknown != null && argv._unknown.some(isFlag)) {
 
 module.exports = {
 	flags,
-
 	run: getArgs(flags.run),
 	export: getArgs(flags.export),
 };

flags.json

@@ -62,6 +62,11 @@
             "type": "Boolean",
             "description": "More aggressive preprocessing. Often results in better code, but can break on some edge cases (eg. redefining prototypes)"
         },
+        {
+            "name": "prepended-code",
+            "type": "String",
+            "description": "Input file or directory containing code that should be prepended to the JS file(s) we're analyzing. If directory is given,  prepends contents of all files in the directory in "
+        },
         {
             "name": "no-kill",
             "type": "Boolean",