fix: normalize enum names (#29)

dylibso · Oct 14, 2024 · 1fb7126 · 1fb7126
1 parent c9de0eb
commit 1fb7126
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/template/src/pdk.ts.ejs b/template/src/pdk.ts.ejs
@@ -57,7 +57,14 @@ export class <%- schema.name %> {
  */
 export enum <%- schema.name %> {
   <% schema.enum.forEach(variant => { -%>
-    <%- variant %> = "<%- variant %>",
+    <%- variant
+        // "host:foo$bar" -> "host_Foo$bar"
+        .replace(/[^a-zA-Z0-9_$]+(.)?/g, (a, m) => '_' + (m ||'').toUpperCase())
+        // "13" -> "$13" ("$" is a valid identifier char; with apologies to jquery)
+        .replace(/^([^a-zA-Z_$])/, (a, m) => `$${m}`)
+        // "host_Foo_Bar" -> "Host_Foo_Bar"
+        .replace(/^(.)/, (a, m) => m.toUpperCase())
+    %> = "<%- variant %>",
   <% }) -%>
 }