support string enums as string union type

webrpc · Feb 16, 2024 · ea281f5 · ea281f5
1 parent 0d4f888
commit ea281f5
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 5 deletions.
diff --git a/_examples/node-ts/service.ridl b/_examples/node-ts/service.ridl
@@ -8,6 +8,10 @@ enum Kind: uint32
   - USER
   - ADMIN
 
+enum Intent: string
+  - openSession
+  - closeSession
+  - validateSession
 
 struct User
   - ID: uint64

diff --git a/types.go.tmpl b/types.go.tmpl
@@ -12,11 +12,19 @@
 {{range $_i, $type := $types -}}
 
 {{if isEnumType $type }}
-export enum {{$type.Name}} {
-{{- range $i, $field := $type.Fields}}
-  {{- if $i}},{{end}}
-  {{$field.Name}} = '{{$field.Name}}'
-{{- end}}
+  {{- if eq $type.Type.Expr "string" }}
+    export {{$type.Name}} =
+    {{- range $i, $field := $type.Fields}}
+      {{- if $i}}|{{end}} '{{$field.Name}}'
+    {{- end}}
+  {{ else }}
+    export enum {{$type.Name}} {
+    {{- range $i, $field := $type.Fields}}
+      {{- if $i}},{{end}}
+      {{$field.Name}} = '{{$field.Name}}'
+    {{- end}}
+  {{ end -}}
+{{- if $type.}}
 }
 {{end -}}