horsegen: generate enumerations and lists

horsegen/character.kk.template

@@ -9,6 +9,26 @@ import std/data/rb-map
 import horse/game-id
 pub import horse/character
 
+// Enumeration of all characters for type-safe programming.
+pub type character
+  {{- range $uma := $.Characters }}
+  {{ kkenum $uma.Name }}
+  {{- end }}
+
+// Get the character ID for a character.
+pub fun character-id(c: character): character-id
+  match c
+    {{- range $uma := $.Characters }}
+    {{ kkenum $uma.Name }} -> Character-id({{ $uma.ID }})
+    {{- end }}
+
+// List of all characters in ID order for easy iterating.
+pub val all = [
+  {{- range $uma := $.Characters }}
+  {{ kkenum $uma.Name }},
+  {{- end }}
+]
+
 val name2id: rbmap<string, character-id> = rb-map/empty()
   {{- range $uma := $.Characters }}
   .set({{ printf "%q" $uma.Name }}, Character-id({{ $uma.ID}}))