Merge pull request #154 from prezi/reintroduce-transitives

Kotlin enum support

Author: lptr

spaghetti-kotlin-support/src/main/groovy/com/prezi/spaghetti/kotlin/KotlinEnumGeneratorVisitor.groovy

@@ -17,6 +17,33 @@ class KotlinEnumGeneratorVisitor extends AbstractKotlinGeneratorVisitor {
 """class ${enumName} {
 	class object {
 ${values.join("\n")}
+		private val _values = arrayListOf(${node.values.join(", ")})
+		private val _names = arrayListOf(${node.values.collect { "\"${it}\"" }.join(", ")})
+
+		fun names():Array<String> = _names.copyToArray()
+
+		fun values():Array<${enumName}> = _values.copyToArray()
+
+		fun getName(value:${enumName}):String = _names.get(value as Int)
+
+		fun getValue(value:${enumName}):Int = value as Int
+
+		fun fromValue(value:Int):${enumName} {
+			if (value < 0 || value >= _values.size) {
+				throw IllegalArgumentException("Invalid value for ${enumName}: " + value)
+			}
+			return _values[value]
+		}
+
+		fun valueOf(name:String):${enumName} {
+			var result:MyEnum
+			when(name)
+			{
+${node.values.collect { "				\"${it}\" -> result = ${it}" }.join("\n")}
+				else -> throw IllegalArgumentException("Invalid name for ${enumName}: " + name)
+			}
+			return result
+		}
 	}
 }
 """
@@ -33,7 +60,7 @@ ${values.join("\n")}
 
 		@Override
 		String visitEnumValueNode(EnumValueNode node) {
-			return "\t\tval ${node.name} = ${index} as ${enumName}"
+			return "		val ${node.name} = ${index} as ${enumName}"
 		}
 	}
 }

spaghetti-kotlin-support/src/test/groovy/com/prezi/spaghetti/kotlin/KotlinEnumGeneratorVisitorTest.groovy

@@ -32,6 +32,35 @@ class KotlinEnumGeneratorVisitorTest extends AstTestBase {
 		[deprecated("escape \\"this\\"!")]
 		val BELA = 1 as MyEnum
 		val GEZA = 2 as MyEnum
+		private val _values = arrayListOf(ALMA, BELA, GEZA)
+		private val _names = arrayListOf("ALMA", "BELA", "GEZA")
+
+		fun names():Array<String> = _names.copyToArray()
+
+		fun values():Array<MyEnum> = _values.copyToArray()
+
+		fun getName(value:MyEnum):String = _names.get(value as Int)
+
+		fun getValue(value:MyEnum):Int = value as Int
+
+		fun fromValue(value:Int):MyEnum {
+			if (value < 0 || value >= _values.size) {
+				throw IllegalArgumentException("Invalid value for MyEnum: " + value)
+			}
+			return _values[value]
+		}
+
+		fun valueOf(name:String):MyEnum {
+			var result:MyEnum
+			when(name)
+			{
+				"ALMA" -> result = ALMA
+				"BELA" -> result = BELA
+				"GEZA" -> result = GEZA
+				else -> throw IllegalArgumentException("Invalid name for MyEnum: " + name)
+			}
+			return result
+		}
 	}
 }
 """