From a1239e5f5a2d2c5e63a4417558197e10d0535d8b Mon Sep 17 00:00:00 2001
From: Sebastian McKenzie <sebmck@gmail.com>
Date: Thu, 4 Dec 2014 10:34:52 +1100
Subject: [PATCH] add `ast` return transform option

---
 lib/6to5/file.js     | 24 +++++++++++++-----------
 lib/6to5/register.js |  3 ++-
 2 files changed, 15 insertions(+), 12 deletions(-)

diff --git a/lib/6to5/file.js b/lib/6to5/file.js
index 24d872ffa7..de9957fe79 100644
--- a/lib/6to5/file.js
+++ b/lib/6to5/file.js
@@ -41,7 +41,8 @@ File.normaliseOptions = function (opts) {
     filename:     "unknown",
     modules:      "common",
     runtime:      false,
-    code:         true
+    code:         true,
+    ast:          true
   });
 
   // normalise windows path separators to unix
@@ -186,15 +187,18 @@ File.prototype.generate = function () {
   var opts = this.opts;
   var ast  = this.ast;
 
-  if (!opts.code) {
-    return {
-      code: "",
-      map:  null,
-      ast:  ast
-    };
-  }
+  var result = {
+    code: "",
+    map: null,
+    ast: null
+  };
 
-  var result = generate(ast, opts, this.code);
+  if (opts.ast) result.ast = ast;
+  if (!opts.code) return result;
+
+  var _result = generate(ast, opts, this.code);
+  result.code = _result.code;
+  result.map  = _result.map;
 
   if (this.shebang) {
     // add back shebang
@@ -205,8 +209,6 @@ File.prototype.generate = function () {
     result.code += "\n" + util.sourceMapToComment(result.map);
   }
 
-  result.ast = result;
-
   return result;
 };
 
diff --git a/lib/6to5/register.js b/lib/6to5/register.js
index 02a90fa0ee..efdec2ad90 100644
--- a/lib/6to5/register.js
+++ b/lib/6to5/register.js
@@ -89,7 +89,8 @@ var loader = function (m, filename) {
     whitelist: whitelist,
     blacklist: blacklist,
     sourceMap: true,
-    modules:   "commonInterop"
+    modules:   "commonInterop",
+    ast:       false
   }, transformOpts));
 
   if (cache) {