source map support for stack traces

sane · Jan 16, 2016 · f45c198 · f45c198
1 parent 59c6ff0
commit f45c198
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 4 deletions.
diff --git a/index.js b/index.js
@@ -1,4 +1,6 @@
-var path = require('path');
+var path               = require('path'),
+    sourceMapSupport   = require('source-map-support'),
+    babelRegisterCache = require('babel-register/lib/cache');
 
 module.exports = function(sails) {
 
@@ -36,6 +38,35 @@ module.exports = function(sails) {
         sails.log.verbose("Babel hook deactivated.");
       } else {
 
+        if (sails.config.environment === 'development') {
+
+          sourceMapSupport.install({
+            retrieveSourceMap: function(file) {
+
+              var cache = babelRegisterCache.get(),
+                  sourceMap = null;
+
+              Object.keys(cache).some(function(hash) {
+
+                var fileCache = cache[hash]; 
+
+                if (fileCache.options.filename != file) {
+                  return false;
+                }
+
+                sourceMap = {
+                  url: file,
+                  map: fileCache.map
+                };
+
+                return true;
+              });
+
+              return sourceMap;
+            }
+          });
+        }
+
         if (config.polyfill) {
           require("babel-polyfill");
         }
@@ -48,7 +79,5 @@ module.exports = function(sails) {
         sails.log.verbose("Babel hook activated. Enjoy ES6/7 power in your Sails app.");
       }
     },
-
   };
-
 };
diff --git a/package.json b/package.json
@@ -34,7 +34,8 @@
   "dependencies": {
     "babel-register": "^6.3.13",
     "babel-preset-es2015": "^6.3.13",
-    "babel-polyfill": "^6.3.14"
+    "babel-polyfill": "^6.3.14",
+    "source-map-support": "^0.4.0"
   },
   "devDependencies": {
     "glob": "^5.0.3",