Merge pull request #6 from nickdima/sourcemaps

add source map support

Author: sanex3339

index.js

@@ -1,5 +1,5 @@
 "use strict";
-var JavaScriptObfuscator = require('javascript-obfuscator'), multimatch = require('multimatch'), RawSource = require('webpack-core/lib/RawSource');
+var JavaScriptObfuscator = require('javascript-obfuscator'), multimatch = require('multimatch'), RawSource = require('webpack-core/lib/RawSource'), SourceMapSource = require("webpack-core/lib/SourceMapSource"), transferSourceMap = require("multi-stage-sourcemap").transfer;
 var WebpackObfuscator = (function () {
     function WebpackObfuscator(options, excludes) {
         this.options = {};
@@ -24,8 +24,34 @@ var WebpackObfuscator = (function () {
                     if (_this.shouldExclude(file, _this.excludes)) {
                         return;
                     }
-                    var asset = compilation.assets[file], obfuscationResult = JavaScriptObfuscator.obfuscate(asset.source(), _this.options);
-                    compilation.assets[file] = new RawSource(obfuscationResult.toString());
+                    var asset = compilation.assets[file];
+                    var input, inputSourceMap;
+                    if (_this.options.sourceMap !== false) {
+                        if (asset.sourceAndMap) {
+                            var sourceAndMap = asset.sourceAndMap();
+                            inputSourceMap = sourceAndMap.map;
+                            input = sourceAndMap.source;
+                        }
+                        else {
+                            inputSourceMap = asset.map();
+                            input = asset.source();
+                        }
+                        if (inputSourceMap) {
+                            _this.options.sourceMap = true;
+                        }
+                    }
+                    else {
+                        input = asset.source();
+                    }
+                    var obfuscationResult = JavaScriptObfuscator.obfuscate(input, _this.options);
+                    if (_this.options.sourceMap) {
+                        var obfuscationSourceMap = obfuscationResult.getSourceMap();
+                        var transferedSourceMap = transferSourceMap({ fromSourceMap: obfuscationSourceMap, toSourceMap: inputSourceMap });
+                        compilation.assets[file] = new SourceMapSource(obfuscationResult.toString(), file, JSON.parse(transferedSourceMap), asset.source(), inputSourceMap);
+                    }
+                    else {
+                        compilation.assets[file] = new RawSource(obfuscationResult.toString());
+                    }
                 });
                 callback();
             });

index.ts

@@ -2,7 +2,9 @@
 
 let JavaScriptObfuscator = require('javascript-obfuscator'),
     multimatch: any = require('multimatch'),
-    RawSource: any = require('webpack-core/lib/RawSource');
+    RawSource: any = require('webpack-core/lib/RawSource'),
+    SourceMapSource: any = require("webpack-core/lib/SourceMapSource"),
+    transferSourceMap = require("multi-stage-sourcemap").transfer;
 
 class WebpackObfuscator {
     public options: any = {};
@@ -41,14 +43,36 @@ class WebpackObfuscator {
                     if (this.shouldExclude(file, this.excludes)) {
                         return;
                     }
+                    let asset = compilation.assets[file];
+                    let input, inputSourceMap;
+                    if (this.options.sourceMap !== false) {
+                    	if (asset.sourceAndMap) {
+                    		let sourceAndMap = asset.sourceAndMap();
+                    		inputSourceMap = sourceAndMap.map;
+                    		input = sourceAndMap.source;
+                    	} else {
+                    		inputSourceMap = asset.map();
+                    		input = asset.source();
+                    	}
+                      if (inputSourceMap) {
+                        this.options.sourceMap = true;
+                      }
+                    } else {
+                    	input = asset.source();
+                    }
 
-                    let asset = compilation.assets[file],
-                        obfuscationResult: any = JavaScriptObfuscator.obfuscate(
-                            asset.source(),
+                    let obfuscationResult: any = JavaScriptObfuscator.obfuscate(
+                            input,
                             this.options
                         );
-
-                    compilation.assets[file] = new RawSource(obfuscationResult.toString());
+                    if (this.options.sourceMap) {
+                      let obfuscationSourceMap: any = obfuscationResult.getSourceMap();
+                      let transferedSourceMap: any = transferSourceMap({fromSourceMap: obfuscationSourceMap, toSourceMap: inputSourceMap});
+                      compilation.assets[file] = new SourceMapSource(obfuscationResult.toString(), file, JSON.parse(transferedSourceMap), asset.source(), inputSourceMap);
+                    }
+                    else {
+                      compilation.assets[file] = new RawSource(obfuscationResult.toString());
+                    }
                 });
 
                 callback();

package.json

@@ -15,6 +15,7 @@
   "main": "index.js",
   "dependencies": {
     "javascript-obfuscator": "^0.7.0-dev.3",
+    "multi-stage-sourcemap": "^0.2.1",
     "multimatch": "^2.0.0",
     "webpack-core": "^0.6.8"
   },