feat(DtsGenerator): Added AllRoadizNodesSources Typescript type

roadiz · Dec 2, 2023 · c249f37 · c249f37
1 parent a62c34f
commit c249f37
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/lib/DtsGenerator/src/Generators/DeclarationGenerator.php b/lib/DtsGenerator/src/Generators/DeclarationGenerator.php
@@ -40,9 +40,24 @@ public function getContents(): string
             $blocks[] = $this->generatorFactory->createForNodeType($nodeType)->getContents();
         }
 
+        $blocks[] = $this->getAllTypesInterface();
+
         return implode(PHP_EOL . PHP_EOL, $blocks);
     }
 
+    private function getAllTypesInterface(): string
+    {
+        $nodeTypeNames = array_map(function (NodeTypeInterface $nodeType) {
+            return $nodeType->getSourceEntityClassName();
+        }, $this->nodeTypes);
+
+        $nodeTypeNames = implode(' | ', $nodeTypeNames);
+
+        return <<<EOT
+export type AllRoadizNodesSources = {$nodeTypeNames};
+EOT;
+    }
+
     private function getHeader(): string
     {
         return <<<EOT

diff --git a/lib/DtsGenerator/src/Generators/NodeTypeGenerator.php b/lib/DtsGenerator/src/Generators/NodeTypeGenerator.php
@@ -62,7 +62,7 @@ public function getContents(): string
     protected function getInterfaceBody(): string
     {
         $lines = [
-            'interface ' . $this->nodeType->getSourceEntityClassName() . ' extends RoadizNodesSources {',
+            'export interface ' . $this->nodeType->getSourceEntityClassName() . ' extends RoadizNodesSources {',
             $this->getFieldsContents(),
             '}'
         ];