fix swift generics

apioo · Oct 9, 2024 · 7ba8909 · 7ba8909
1 parent c5b3c17
commit 7ba8909
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/src/Generator/Swift.php b/src/Generator/Swift.php
@@ -53,10 +53,20 @@ protected function newNormalizer(): NormalizerInterface
 
     protected function writeStruct(Code\Name $name, array $properties, ?string $extends, ?array $generics, ?array $templates, StructDefinitionType $origin): string
     {
-        $code = 'class ' . $name->getClass() . ': ';
+        $code = 'class ' . $name->getClass();
+
+        if (!empty($generics)) {
+            $code.= $this->generator->getGenericDefinition($generics);
+        }
+
+        $code.= ': ';
 
         if (!empty($extends)) {
-            $code.= $extends . ' ';
+            $code.= $extends;
+            if (!empty($templates)) {
+                $code.= $this->generator->getGenericDefinition($templates);
+            }
+            $code.= ' ';
         } else {
             $code.= 'Codable ';
         }

diff --git a/tests/Generator/resource/swift/swift_oop.swift b/tests/Generator/resource/swift/swift_oop.swift
@@ -16,7 +16,7 @@ class Student: Human {
     }
 }
 
-class Map: Codable {
+class Map<P, T>: Codable {
     var totalResults: Int
     var parent: P
     var entries: Array<T>
@@ -28,13 +28,13 @@ class Map: Codable {
     }
 }
 
-class StudentMap: Map {
+class StudentMap: Map<Human, Student> {
 
     enum CodingKeys: String, CodingKey {
     }
 }
 
-class HumanMap: Map {
+class HumanMap: Map<Human, Human> {
 
     enum CodingKeys: String, CodingKey {
     }