fix proxy schema generator

apioo · Sep 8, 2023 · 401005e · 401005e
1 parent d52ad02
commit 401005e
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/Generator/Proxy/Schema.php b/src/Generator/Proxy/Schema.php
@@ -50,7 +50,18 @@ public function __construct(string $type, ?string $config = null)
     public function generate(SpecificationInterface $specification): Generator\Code\Chunks|string
     {
         $schema = new \PSX\Schema\Schema(TypeFactory::getAny(), $specification->getDefinitions());
+        $generator = $this->factory->getGenerator($this->type, $this->config);
 
-        return $this->factory->getGenerator($this->type, $this->config)->generate($schema);
+        $result = $generator->generate($schema);
+        if ($result instanceof Generator\Code\Chunks && $generator instanceof Generator\FileAwareInterface) {
+            $chunks = new Generator\Code\Chunks();
+            foreach ($result->getChunks() as $identifier => $content) {
+                $chunks->append($generator->getFileName($identifier), $generator->getFileContent($content));
+            }
+
+            return $chunks;
+        } else {
+            return $result;
+        }
     }
 }