TypeScriptToLua
diff --git a/‎src/Transpiler.ts‎
Lines changed: 7 additions & 3 deletions b/‎src/Transpiler.ts‎
Lines changed: 7 additions & 3 deletions
diff --git a/‎test/translation/lua/enum.lua‎
Lines changed: 2 additions & 1 deletion b/‎test/translation/lua/enum.lua‎
Lines changed: 2 additions & 1 deletion
diff --git a/‎test/translation/lua/enumMembersOnly.lua‎
Lines changed: 1 addition & 0 deletions b/‎test/translation/lua/enumMembersOnly.lua‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎test/translation/ts/enum.ts‎
Lines changed: 3 additions & 2 deletions b/‎test/translation/ts/enum.ts‎
Lines changed: 3 additions & 2 deletions
diff --git a/‎test/translation/ts/enumMembersOnly.ts‎
Lines changed: 3 additions & 2 deletions b/‎test/translation/ts/enumMembersOnly.ts‎
Lines changed: 3 additions & 2 deletions
diff --git a/‎test/unit/enum.spec.ts‎
Lines changed: 18 additions & 4 deletions b/‎test/unit/enum.spec.ts‎
Lines changed: 18 additions & 4 deletions
@@ -282,7 +282,7 @@ export abstract class LuaTranspiler {
     }
 
     public transpileEnum(node: ts.EnumDeclaration): string {
-        let val = 0;
+        let val: number | string = 0;
         let result = "";
 
         const type = this.checker.getTypeAtLocation(node);
@@ -298,8 +298,10 @@ export abstract class LuaTranspiler {
             if (member.initializer) {
                 if (ts.isNumericLiteral(member.initializer)) {
                     val = parseInt(member.initializer.text);
+                } else if (ts.isStringLiteral(member.initializer)) {
+                    val = `"${member.initializer.text}"`;
                 } else {
-                    throw new TranspileError("Only numeric initializers allowed for enums.", node);
+                    throw new TranspileError("Only numeric or string initializers allowed for enums.", node);
                 }
             }
 
@@ -313,7 +315,9 @@ export abstract class LuaTranspiler {
                 result += this.indent + `${defName}=${val}\n`;
             }
 
-            val++;
+            if (typeof val === "number") {
+              val++;
+            }
         });
         return result;
     }
 
@@ -1,4 +1,5 @@
 TestEnum={}
 TestEnum.val1=0
 TestEnum.val2=2
-TestEnum.val3=3
+TestEnum.val3=3
+TestEnum.val4="bye"
@@ -1,4 +1,5 @@
 val1=0
 val2=2
 val3=3
+val4="bye"
 local a = val1
@@ -1,5 +1,6 @@
 enum TestEnum {
     val1 = 0,
     val2 = 2,
-    val3
-}
+    val3,
+    val4 = "bye"
+}
@@ -2,7 +2,8 @@
 enum TestEnum {
     val1 = 0,
     val2 = 2,
-    val3
+    val3,
+    val4 = "bye"
 }
 
-const a = TestEnum.val1;
+const a = TestEnum.val1;
@@ -2,8 +2,8 @@ import { Expect, Test, TestCase } from "alsatian";
 import * as util from "../src/util"
 
 export class EnumTests {
-    @Test("Unsuported enum")
-    public unsuportedEnum() {
+    @Test("String enum")
+    public stringEnum() {
         // Transpile & Assert
         Expect(() => {
             let lua = util.transpileString(
@@ -13,6 +13,20 @@ export class EnumTests {
                     val3 = "bye"
                 }`
             );
-        }).toThrowError(Error, "Only numeric initializers allowed for enums.");
-    }
+        }).not.toThrow();
+	}
+
+	@Test("Unsuported enum")
+	public unsuportedEnum() {
+		// Transpile & Assert
+		Expect(() => {
+			let lua = util.transpileString(
+				`enum TestEnum {
+                    val1 = [],
+                    val2 = "ok",
+                    val3 = "bye"
+                }`
+			);
+		}).toThrowError(Error, "Only numeric or string initializers allowed for enums.");
+	}
 }
Original file line number	Diff line number	Diff line change
`@@ -282,7 +282,7 @@ export abstract class LuaTranspiler {`
`282`	`282`	`}`
`283`	`283`
`284`	`284`	`public transpileEnum(node: ts.EnumDeclaration): string {`
`285`		`- let val = 0;`
	`285`	`+ let val: number \| string = 0;`
`286`	`286`	`let result = "";`
`287`	`287`
`288`	`288`	`const type = this.checker.getTypeAtLocation(node);`
`@@ -298,8 +298,10 @@ export abstract class LuaTranspiler {`
`298`	`298`	`if (member.initializer) {`
`299`	`299`	`if (ts.isNumericLiteral(member.initializer)) {`
`300`	`300`	`val = parseInt(member.initializer.text);`
	`301`	`+ } else if (ts.isStringLiteral(member.initializer)) {`
	`302`	+ val = `"${member.initializer.text}"`;
`301`	`303`	`} else {`
`302`		`- throw new TranspileError("Only numeric initializers allowed for enums.", node);`
	`304`	`+ throw new TranspileError("Only numeric or string initializers allowed for enums.", node);`
`303`	`305`	`}`
`304`	`306`	`}`
`305`	`307`
`@@ -313,7 +315,9 @@ export abstract class LuaTranspiler {`
`313`	`315`	result += this.indent + `${defName}=${val}\n`;
`314`	`316`	`}`
`315`	`317`
`316`		`- val++;`
	`318`	`+ if (typeof val === "number") {`
	`319`	`+ val++;`
	`320`	`+ }`
`317`	`321`	`});`
`318`	`322`	`return result;`
`319`	`323`	`}`
-Original file line number
+Diff line change
@@ @@ -1,4 +1,5 @@ @@
 val1=0
 val2=2
 val3=3
 +val4="bye"
 local a = val1