Adding arrow functions

2015-01-11 20:07:19 +00:00
parent 9d7d365c2b
commit fa5c4f2d03
6 changed files with 129 additions and 2 deletions
--- a/lib/transform.js
+++ b/lib/transform.js
@@ -166,7 +166,11 @@ TreeTransformer.prototype = new TreeWalker;
    _(AST_Lambda, function(self, tw){
        if (self.name) self.name = self.name.transform(tw);
        self.argnames = do_list(self.argnames, tw);
-        self.body = do_list(self.body, tw);
+        if (self.body instanceof AST_Node) {
+            self.body = self.body.transform(tw);
+        } else {
+            self.body = do_list(self.body, tw);
+        }
    });

    _(AST_Call, function(self, tw){