Return default export classes

Perryvw · Perryvw · commit 4f1fd4188fc6 · 2018-02-04T14:29:25.000+01:00
diff --git a/dist/Transpiler.js b/dist/Transpiler.js
@@ -784,6 +784,12 @@ var LuaTranspiler = /** @class */ (function () {
         node.members.filter(ts.isMethodDeclaration).forEach(function (method) {
             result += _this.transpileMethodDeclaration(method, className + ".");
         });
+        // Check if the class should be returned
+        var isExport = node.modifiers && node.modifiers.some(function (_) { return _.kind == ts.SyntaxKind.ExportKeyword; });
+        var isDefault = node.modifiers && node.modifiers.some(function (_) { return _.kind == ts.SyntaxKind.DefaultKeyword; });
+        if (isExport && isDefault) {
+            result += this.indent + ("return " + className + "\n");
+        }
         return result;
     };
     LuaTranspiler.prototype.transpileConstructor = function (node, className, instanceFields) {
diff --git a/src/Transpiler.ts b/src/Transpiler.ts
@@ -852,6 +852,13 @@ export class LuaTranspiler {
             result += this.transpileMethodDeclaration(method, `${className}.`);
         });
 
+        // Check if the class should be returned
+        const isExport = node.modifiers && node.modifiers.some(_ => _.kind == ts.SyntaxKind.ExportKeyword);
+        const isDefault = node.modifiers && node.modifiers.some(_ => _.kind == ts.SyntaxKind.DefaultKeyword);
+        if (isExport && isDefault) {
+            result += this.indent + `return ${className}\n`;
+        }
+
         return result;
     }
 
