New: Add class decorators to AST (fixes #66) (#67)

Author: JamesHenry

lib/ast-converter.js

@@ -1317,7 +1317,10 @@ module.exports = function(ast, extra) {
                         loc: getLocFor(openBrace.getStart(), node.end, ast)
                     },
                     superClass: (superClass ? convertChild(superClass.types[0].expression) : null),
-                    implements: hasImplements ? heritageClauses[0].types.map(convertClassImplements) : []
+                    implements: hasImplements ? heritageClauses[0].types.map(convertClassImplements) : [],
+                    decorators: (node.decorators) ? node.decorators.map(function(d) {
+                        return convertChild(d.expression);
+                    }) : []
                 });
 
                 var filteredMembers = node.members.filter(isESTreeClassMember);

tests/fixtures/attach-comments/export-default-anonymous-class.result.js

@@ -5,6 +5,7 @@ module.exports = {
             "type": "ExportDefaultDeclaration",
             "declaration": {
                 "type": "ClassDeclaration",
+                "decorators": [],
                 "id": null,
                 "superClass": null,
                 "implements": [],

tests/fixtures/ecma-features-mix/classes-and-generators/classes-and-generators.result.js

@@ -3,6 +3,7 @@ module.exports = {
     "body": [
         {
             "type": "ClassDeclaration",
+            "decorators": [],
             "id": {
                 "type": "Identifier",
                 "name": "Foo",
@@ -331,4 +332,4 @@ module.exports = {
             }
         }
     ]
-};
+};

tests/fixtures/ecma-features-mix/classes-and-generators/computed-generator.result.js

@@ -3,6 +3,7 @@ module.exports = {
     "body": [
         {
             "type": "ClassDeclaration",
+            "decorators": [],
             "id": {
                 "type": "Identifier",
                 "name": "Foo",

tests/fixtures/ecma-features-mix/classes-and-generators/static-generators.result.js

@@ -3,6 +3,7 @@ module.exports = {
     "body": [
         {
             "type": "ClassDeclaration",
+            "decorators": [],
             "id": {
                 "type": "Identifier",
                 "name": "Foo",
@@ -349,4 +350,4 @@ module.exports = {
             }
         }
     ]
-};
+};

tests/fixtures/ecma-features-mix/modules-and-classes/class-default-anonymous-extends.result.js

@@ -5,6 +5,7 @@ module.exports = {
             "type": "ExportDefaultDeclaration",
             "declaration": {
                 "type": "ClassDeclaration",
+                "decorators": [],
                 "id": null,
                 "superClass": {
                     "type": "Identifier",
@@ -251,4 +252,4 @@ module.exports = {
             }
         }
     ]
-};
+};

tests/fixtures/ecma-features-mix/modules-and-classes/class-default-anonymous.result.js

@@ -5,6 +5,7 @@ module.exports = {
             "type": "ExportDefaultDeclaration",
             "declaration": {
                 "type": "ClassDeclaration",
+                "decorators": [],
                 "id": null,
                 "superClass": null,
                 "body": {
@@ -198,4 +199,4 @@ module.exports = {
             }
         }
     ]
-};
+};

tests/fixtures/ecma-features-mix/modules-and-classes/class-default-extends.result.js

@@ -5,6 +5,7 @@ module.exports = {
             "type": "ExportDefaultDeclaration",
             "declaration": {
                 "type": "ClassDeclaration",
+                "decorators": [],
                 "id": {
                     "type": "Identifier",
                     "name": "foo",
@@ -286,4 +287,4 @@ module.exports = {
             }
         }
     ]
-};
+};

tests/fixtures/ecma-features-mix/modules-and-classes/class-default.result.js

@@ -5,6 +5,7 @@ module.exports = {
             "type": "ExportDefaultDeclaration",
             "declaration": {
                 "type": "ClassDeclaration",
+                "decorators": [],
                 "id": {
                     "type": "Identifier",
                     "name": "foo",
@@ -233,4 +234,4 @@ module.exports = {
             }
         }
     ]
-};
+};

tests/fixtures/ecma-features-mix/modules-and-classes/class-extends.result.js

@@ -5,6 +5,7 @@ module.exports = {
             "type": "ExportNamedDeclaration",
             "declaration": {
                 "type": "ClassDeclaration",
+                "decorators": [],
                 "id": {
                     "type": "Identifier",
                     "name": "foo",
@@ -270,4 +271,4 @@ module.exports = {
             }
         }
     ]
-};
+};