Accept filename-to-file-content hash as code parameter of generate.

2016-02-04 22:58:19 -08:00
parent 26dcd6400f
commit f007b51554
2 changed files with 51 additions and 2 deletions
--- a/packages/babel-generator/src/source-map.js
+++ b/packages/babel-generator/src/source-map.js
@@ -17,7 +17,13 @@ export default class SourceMap {
        sourceRoot: opts.sourceRoot
      });

-      this.map.setSourceContent(opts.sourceFileName, code);
+      if (typeof code === "string") {
+        this.map.setSourceContent(opts.sourceFileName, code);
+      } else if (typeof code === "object") {
+        Object.keys(code).forEach((sourceFileName) => {
+          this.map.setSourceContent(sourceFileName, code[sourceFileName]);
+        });
+      }
    } else {
      this.map = null;
    }