add AllEnum() function (#1079)

volatiletech · Feb 1, 2022 · d6e9344 · d6e9344
1 parent d5757fd
commit d6e9344
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/templates/main/singleton/boil_types.go.tpl b/templates/main/singleton/boil_types.go.tpl
@@ -88,7 +88,11 @@ It only titlecases the EnumValue portion if it's snake-cased.
 				-}}
 
 				{{- if $enumFirstIter -}}
+					{{$enumType := "string" }}
+					{{$allvals := "\n"}}
+
 					{{if $.AddEnumTypes}}
+						{{- $enumType = $enumName -}}
 						type {{$enumName}} string
 					{{end}}
 
@@ -100,9 +104,14 @@ It only titlecases the EnumValue portion if it's snake-cased.
 						{{- if shouldTitleCaseEnum $valStripped -}}
 							{{$enumValue = titleCase $valStripped}}
 						{{- end -}}
-						{{$enumName}}{{$enumValue}} {{if $.AddEnumTypes}}{{$enumName}}{{end}} = "{{$val}}"
+						{{$enumName}}{{$enumValue}} {{$enumType}} = "{{$val}}"
+						{{$allvals = printf "%s%s%s,\n" $allvals $enumName $enumValue -}}
 					{{end -}}
 					)
+
+					func All{{$enumName}}() []{{$enumType}} {
+						return []{{$enumType}}{ {{$allvals}} }
+					}
 				{{- end -}}
 
 				{{if $.AddEnumTypes}}