feat: refactor swift enum

rudderlabs · Feb 3, 2025 · fa7be85 · fa7be85
1 parent 169505f
commit fa7be85
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/src/generators/swift/swift.ts b/src/generators/swift/swift.ts
@@ -37,6 +37,8 @@ type SwiftPropertyContext = {
   enumName?: string;
   // The formatted enum values
   enumValues?: any;
+  // The type of the enum values
+  enumType?: string;
 };
 
 type SwiftTrackCallContext = {
@@ -206,6 +208,7 @@ function defaultPropertyContext(
     hasEnum: !!hasEnum,
     enumName: sanitizeEnumKey(schema.name) + '_' + getEnumPropertyTypes(schema),
     enumValues: hasEnum && 'enum' in schema ? convertToEnum(schema.enum!, type) : undefined,
+    enumType: type === 'Any' ? 'String' : type,
   };
 }
 

diff --git a/src/generators/swift/templates/analytics.swift.hbs b/src/generators/swift/templates/analytics.swift.hbs
@@ -10,7 +10,7 @@ class RudderTyperAnalytics {
     {{#each properties}}
     {{#if hasEnum}}
 	{{#uniqueEnum enumName enumValues}}
-	public enum {{enumName}}: {{../type}} {
+	enum {{enumName}}: {{../enumType}} {
 	    {{enumValues}}
 	}
 

diff --git a/src/generators/swift/templates/class.swift.hbs b/src/generators/swift/templates/class.swift.hbs
@@ -6,6 +6,14 @@ import Foundation
 
 class {{name}}: RudderTyperSerializable {
 {{#each properties}}
+    {{#if hasEnum}}
+    {{#uniqueEnum enumName enumValues}}
+    enum {{enumName}}: {{../enumType}} {
+        {{enumValues}}
+    }
+    {{/uniqueEnum}}
+    {{/if}}
+
 {{#if description}}
 /// {{description}}
 {{/if}}