support for class expressions (#345)

tomblind · Perryvw · commit 1ef66d1ddde4 · 2019-01-19T14:37:30.000+01:00
diff --git a/src/LuaTransformer.ts b/src/LuaTransformer.ts
@@ -229,7 +229,10 @@ export class LuaTransformer {
         }
     }
 
-    public transformClassDeclaration(statement: ts.ClassDeclaration, nameOverride?: tstl.Identifier): tstl.Statement[] {
+    public transformClassDeclaration(
+        statement: ts.ClassLikeDeclaration,
+        nameOverride?: tstl.Identifier
+    ): tstl.Statement[] {
         let className = statement.name ? this.transformIdentifier(statement.name) : nameOverride;
         if (!className) {
             throw TSTLErrors.MissingClassName(statement);
@@ -1496,11 +1499,9 @@ export class LuaTransformer {
                 // TODO move to extra function (consistency)
                 return undefined;
             case ts.SyntaxKind.ClassExpression:
-                /*this.namespace.push("");
-                const classDeclaration =  this.transformClassDeclaration(node as ts.ClassExpression, "_");
-                this.namespace.pop();
-                return `(function() ${classDeclaration}; return _ end)()`;*/
-                throw new Error("Not yet implemented");
+                const className = tstl.createIdentifier("____");
+                const classDeclaration =  this.transformClassDeclaration(expression as ts.ClassExpression, className);
+                return this.createImmediatelyInvokedFunctionExpression(classDeclaration, className, expression);
             default:
                 throw TSTLErrors.UnsupportedKind("expression", expression.kind, expression);
         }
