1 files changed, 57 insertions, 0 deletions
diff --git a/app/src/main/kotlin/com/pitchedapps/frost/utils/EnumUtils.kt b/app/src/main/kotlin/com/pitchedapps/frost/utils/EnumUtils.kt
new file mode 100644
index 00000000..d20d1573
--- /dev/null
+++ b/app/src/main/kotlin/com/pitchedapps/frost/utils/EnumUtils.kt
@@ -0,0 +1,57 @@
+package com.pitchedapps.frost.utils
+
+import android.content.Intent
+import android.os.BaseBundle
+
+/**
+ * Created by Allan Wang on 29/12/17.
+ *
+ * Helper to set enum using its name rather than the serialized version
+ * Name is used in case the enum is involved in persistent data, where updates may shift indices
+ */
+interface EnumBundle<E : Enum<E>> {
+
+    val bundleContract: EnumBundleCompanion<E>
+
+    val name: String
+
+    val ordinal: Int
+
+    fun put(intent: Intent) {
+        intent.putExtra(bundleContract.argTag, name)
+    }
+
+    fun put(bundle: BaseBundle?) {
+        bundle?.putString(bundleContract.argTag, name)
+    }
+}
+
+interface EnumBundleCompanion<E : Enum<E>> {
+
+    val argTag: String
+
+    val values: Array<E>
+
+    val valueMap: Map<String, E>
+
+    operator fun get(name: String?) = if (name == null) null else valueMap[name]
+
+    operator fun get(bundle: BaseBundle?) = get(bundle?.getString(argTag))
+
+    operator fun get(intent: Intent?) = get(intent?.getStringExtra(argTag))
+
+}
+
+open class EnumCompanion<E : Enum<E>>(
+        override final val argTag: String,
+        override final val values: Array<E>) : EnumBundleCompanion<E> {
+
+    override final val valueMap: Map<String, E> = values.map { it.name to it }.toMap()
+
+    override final fun get(name: String?) = super.get(name)
+
+    override final fun get(bundle: BaseBundle?) = super.get(bundle)
+
+    override final fun get(intent: Intent?) = super.get(intent)
+
+}
+\ No newline at end of file