esrap 2.1.1 → 2.1.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "esrap",
-  "version": "2.1.1",
+  "version": "2.1.3",
   "description": "Parse in reverse",
   "repository": {
     "type": "git",

package/src/languages/ts/index.js

@@ -501,6 +501,12 @@ export default (options = {}) => {
 		 * @param {Context} context
 		 */
 		'ClassDeclaration|ClassExpression': (node, context) => {
+			if (node.decorators) {
+				for (const decorator of node.decorators) {
+					context.visit(decorator);
+				}
+			}
+
 			if (node.declare) {
 				context.write('declare ');
 			}
@@ -956,13 +962,28 @@ export default (options = {}) => {
 		},
 
 		ExportNamedDeclaration(node, context) {
-			context.write('export ');
-
 			if (node.declaration) {
-				context.visit(node.declaration);
+				// Check if declaration has decorators (ClassDeclaration, ClassExpression can have them)
+				const decl = /** @type {any} */ (node.declaration);
+				if (decl.decorators && decl.decorators.length > 0) {
+					for (const decorator of decl.decorators) {
+						context.visit(decorator);
+					}
+					context.write('export ');
+					// Temporarily remove decorators so ClassDeclaration doesn't print them again
+					const savedDecorators = decl.decorators;
+					decl.decorators = [];
+					context.visit(node.declaration);
+					decl.decorators = savedDecorators;
+				} else {
+					context.write('export ');
+					context.visit(node.declaration);
+				}
 				return;
 			}
 
+			context.write('export ');
+
 			if (node.exportKind === 'type') {
 				context.write('type ');
 			}