Obfuscate and overwrite source files when dest not specified

Author: tomasz-oponowicz

tasks/javascript_obfuscator.js

@@ -16,12 +16,35 @@ function concat(grunt, paths) {
   }
 
   return paths.map(function(path) {
-
-    // Read file source.
     return grunt.file.read(path);
   }).join(grunt.util.normalizelf(';'));
 }
 
+function obfuscate(grunt, originalSource, destFile, options) {
+  var obfuscatedSource = JavaScriptObfuscator.obfuscate(originalSource, options).getObfuscatedCode();
+  grunt.file.write(destFile, obfuscatedSource);
+  grunt.log.writeln('File "' + destFile + '" obfuscated.');
+}
+
+function obfuscateSingleTarget(grunt, srcFiles, destFile, options) {
+  var source;
+
+  try {
+    source = concat(grunt, srcFiles);
+  } catch (error) {
+    return grunt.fail.warn(error);
+  }
+
+  obfuscate(grunt, source, destFile, options);
+}
+
+function obfuscateMultiplesTargets(grunt, destFiles, options) {
+  destFiles.forEach(function (destFile) {
+    var source = grunt.file.read(destFile);
+    obfuscate(grunt, source, destFile, options);
+  });
+}
+
 module.exports = function(grunt) {
   grunt.registerMultiTask('javascript_obfuscator', 'Obfuscates JavaScript files.', function() {
 
@@ -33,22 +56,12 @@ module.exports = function(grunt) {
       throw new Error('Target files not found.');
     }
 
-    files.forEach(function(f) {
-      var src;
-
-      try {
-        src = concat(grunt, f.src);
-      } catch (e) {
-        return grunt.fail.warn(e);
+    files.forEach(function(file) {
+      if (file.dest) {
+        obfuscateSingleTarget(grunt, file.src, file.dest, options);
+      } else {
+        obfuscateMultiplesTargets(grunt, file.src, options);
       }
-
-      src = JavaScriptObfuscator.obfuscate(src, options).getObfuscatedCode();
-
-      // Write the destination file.
-      grunt.file.write(f.dest, src);
-
-      // Print a success message.
-      grunt.log.writeln('File "' + f.dest + '" created.');
     });
   });
 };

test/fixtures/gruntfile_overwrite_without_dest.js

@@ -0,0 +1,6 @@
+module.exports = require('./gruntfile_generic').setup({
+    options: {},
+    overwrite_without_dest: {
+        src: '../../tmp/overwrite_without_dest_*'
+    }
+});

test/javascript_obfuscator_test.js

@@ -131,6 +131,40 @@ exports.javascript_obfuscator = {
             // should be obfuscated
             test.ok(obfuscated.indexOf('FIRST_STRING_LITERAL') === -1, '`FIRST_STRING_LITERAL` should be obfuscated');
 
+            test.done();
+        });
+    },
+
+    overwrite_without_dest: function(test) {
+        test.expect(8);
+
+        grunt.file.copy('test/fixtures/first_sample.js', 'tmp/overwrite_without_dest_first.js');
+        grunt.file.copy('test/fixtures/second_sample.js', 'tmp/overwrite_without_dest_second.js');
+
+        helper.callGruntfile('fixtures/gruntfile_overwrite_without_dest.js', function (error, stdout, stderr) {
+            test.equal(error, null, "Command should not fail.");
+            test.equal(stderr, '', "Standard error stream should be empty.");
+
+            var obfuscated;
+
+            obfuscated = grunt.file.read('tmp/overwrite_without_dest_first.js');
+
+            // should NOT be obfuscated
+            test.ok(obfuscated.indexOf('console') > -1, '`console` shouldn\'t be obfuscated');
+            test.ok(obfuscated.indexOf('FIRST_LOCAL_VARIABLE') > -1, '`FIRST_LOCAL_VARIABLE` shouldn\'t be obfuscated');
+
+            // should be obfuscated
+            test.ok(obfuscated.indexOf('FIRST_STRING_LITERAL') === -1, '`FIRST_STRING_LITERAL` should be obfuscated');
+
+            obfuscated = grunt.file.read('tmp/overwrite_without_dest_second.js');
+
+            // should NOT be obfuscated
+            test.ok(obfuscated.indexOf('console') > -1, '`console` shouldn\'t be obfuscated');
+            test.ok(obfuscated.indexOf('SECOND_LOCAL_VARIABLE') > -1, '`SECOND_LOCAL_VARIABLE` shouldn\'t be obfuscated');
+
+            // should be obfuscated
+            test.ok(obfuscated.indexOf('SECOND_STRING_LITERAL') === -1, '`SECOND_STRING_LITERAL` should be obfuscated');
+
             test.done();
         });
     }