Accept source map input for the visitor

README.md

@@ -88,6 +88,40 @@ If you don't provide options in your Babel config, the plugin will look for `exc
 
 You can also use [istanbul's ignore hints](https://github.com/gotwarlost/istanbul/blob/master/ignoring-code-for-coverage.md#ignoring-code-for-coverage-purposes) to specify specific lines of code to skip instrumenting.
 
+## Source Maps
+
+By default, this plugin will pick up inline source maps and attach them to the instrumented code such that code coverage can be remapped back to the original source, even for multi-step build processes. This can be memory intensive. Set `useInlineSourceMaps` to prevent this behavior.
+
+```json
+{
+  "env": {
+    "test": {
+      "plugins": [
+        ["istanbul", {
+          "useInlineSourceMaps": false
+        }]
+      ]
+    }
+  }
+}
+```
+
+If you're instrumenting code programatically, you can pass a source map explicitly.
+```js
+import babelPluginIstanbul from 'babel-plugin-istanbul';
+
+function instrument(sourceCode, sourceMap, fileName) {
+  return babel.transform(sourceCode, {
+    filename,
+    plugins: [
+      [babelPluginIstanbul, {
+        inputSourceMap: sourceMap
+      }]
+    ]
+  })
+}
+```
+
 ## Credit where credit is due
 
 The approach used in `babel-plugin-istanbul` was inspired by [Thai Pangsakulyanont](https://github.com/dtinth)'s original library [`babel-plugin-__coverage__`](https://github.com/dtinth/babel-plugin-__coverage__).

package.json

@@ -36,6 +36,11 @@
     "prepublish": "npm test && npm run release",
     "version": "standard-version"
   },
+  "standard": {
+    "ignore": [
+      "fixtures/has-inline-source-map.js"
+    ]
+  },
   "repository": {
     "type": "git",
     "url": "git+https://github.com/istanbuljs/babel-plugin-istanbul.git"

src/index.js

@@ -43,7 +43,14 @@ function makeVisitor ({types: t}) {
           if (shouldSkip(realPath, this.opts)) {
             return
           }
-          this.__dv__ = programVisitor(t, realPath)
+          let { inputSourceMap } = this.opts
+          if (this.opts.useInlineSourceMaps !== false) {
+            inputSourceMap = inputSourceMap || this.file.opts.inputSourceMap
+          }
+          this.__dv__ = programVisitor(t, realPath, {
+            coverageVariable: '__coverage__',
+            inputSourceMap
+          })
           this.__dv__.enter(path)
         },
         exit (path) {

test/babel-plugin-istanbul.js

@@ -32,6 +32,43 @@ describe('babel-plugin-istanbul', function () {
     })
   })
 
+  context('source maps', function () {
+    it('should use inline source map', function () {
+      var result = babel.transformFileSync('./fixtures/has-inline-source-map.js', {
+        plugins: [
+          [makeVisitor({types: babel.types}), {
+            include: ['fixtures/has-inline-source-map.js']
+          }]
+        ]
+      })
+      result.code.should.match(/inputSourceMap/)
+    })
+
+    it('should not use inline source map if inputSourceMap is set to false', function () {
+      var result = babel.transformFileSync('./fixtures/has-inline-source-map.js', {
+        plugins: [
+          [makeVisitor({types: babel.types}), {
+            include: ['fixtures/has-inline-source-map.js'],
+            useInlineSourceMaps: false
+          }]
+        ]
+      })
+      result.code.should.not.match(/inputSourceMap/)
+    })
+
+    it('should use provided source map', function () {
+      var result = babel.transformFileSync('./fixtures/has-inline-source-map.js', {
+        plugins: [
+          [makeVisitor({types: babel.types}), {
+            include: ['fixtures/has-inline-source-map.js'],
+            inputSourceMap: { asdfQwer: 'foo' }
+          }]
+        ]
+      })
+      result.code.should.match(/inputSourceMap:\s*{\s*asdfQwer: "foo"\s*}/)
+    })
+  })
+
   context('package.json "nyc" config', function () {
     it('should instrument file if shouldSkip returns false', function () {
       var result = babel.transformFileSync('./fixtures/should-cover.js', {