|
7 | 7 | use GraphQL\SchemaGenerator\CodeGenerator\ArgumentsObjectClassBuilder; |
8 | 8 | use GraphQL\SchemaGenerator\CodeGenerator\EnumObjectBuilder; |
9 | 9 | use GraphQL\SchemaGenerator\CodeGenerator\InputObjectClassBuilder; |
| 10 | +use GraphQL\SchemaGenerator\CodeGenerator\InterfaceObjectBuilder; |
10 | 11 | use GraphQL\SchemaGenerator\CodeGenerator\ObjectBuilderInterface; |
11 | 12 | use GraphQL\SchemaGenerator\CodeGenerator\QueryObjectClassBuilder; |
12 | 13 | use GraphQL\SchemaGenerator\CodeGenerator\UnionObjectBuilder; |
@@ -44,7 +45,7 @@ class SchemaClassGenerator |
44 | 45 | *AND complete covering the schema scanner class |
45 | 46 | * @var array |
46 | 47 | */ |
47 | | - private $generatedObjects; |
| 48 | + private $generatedObjects; |
48 | 49 |
|
49 | 50 | /** |
50 | 51 | * SchemaClassGenerator constructor. |
@@ -141,6 +142,7 @@ protected function generateObject(string $objectName, string $objectKind): bool |
141 | 142 | { |
142 | 143 | switch ($objectKind) { |
143 | 144 | case FieldTypeKindEnum::OBJECT: |
| 145 | + case FieldTypeKindEnum::INTERFACE_OBJECT: |
144 | 146 | return $this->generateQueryObject($objectName); |
145 | 147 | case FieldTypeKindEnum::INPUT_OBJECT: |
146 | 148 | return $this->generateInputObject($objectName); |
@@ -168,7 +170,15 @@ protected function generateQueryObject(string $objectName): bool |
168 | 170 | $this->generatedObjects[$objectName] = true; |
169 | 171 | $objectArray = $this->schemaInspector->getObjectSchema($objectName); |
170 | 172 | $objectName = $objectArray['name']; |
171 | | - $objectBuilder = new QueryObjectClassBuilder($this->writeDir, $objectName, $this->generationNamespace); |
| 173 | + |
| 174 | + if ($objectArray['kind'] === FieldTypeKindEnum::INTERFACE_OBJECT) { |
| 175 | + $objectBuilder = new InterfaceObjectBuilder($this->writeDir, $objectName, $this->generationNamespace); |
| 176 | + foreach ($objectArray['possibleTypes'] as $possibleType) { |
| 177 | + $objectBuilder->addImplementation($possibleType['name']); |
| 178 | + } |
| 179 | + } else { |
| 180 | + $objectBuilder = new QueryObjectClassBuilder($this->writeDir, $objectName, $this->generationNamespace); |
| 181 | + } |
172 | 182 |
|
173 | 183 | $this->appendQueryObjectFields($objectBuilder, $objectName, $objectArray['fields']); |
174 | 184 | $objectBuilder->build(); |
@@ -274,6 +284,32 @@ protected function generateUnionObject(string $objectName): bool |
274 | 284 | return true; |
275 | 285 | } |
276 | 286 |
|
| 287 | + /** |
| 288 | + * @param string $objectName |
| 289 | + * |
| 290 | + * @return bool |
| 291 | + */ |
| 292 | + protected function generateInterfaceObject(string $objectName): bool |
| 293 | + { |
| 294 | + if (array_key_exists($objectName, $this->generatedObjects)) { |
| 295 | + return true; |
| 296 | + } |
| 297 | + |
| 298 | + $this->generatedObjects[$objectName] = true; |
| 299 | + |
| 300 | + $objectArray = $this->schemaInspector->getObjectSchema($objectName); |
| 301 | + $objectName = $objectArray['name']; |
| 302 | + $objectBuilder = new UnionObjectBuilder($this->writeDir, $objectName, $this->generationNamespace); |
| 303 | + |
| 304 | + foreach ($objectArray['possibleTypes'] as $possibleType) { |
| 305 | + $this->generateObject($possibleType['name'], $possibleType['kind']); |
| 306 | + $objectBuilder->addPossibleType($possibleType['name']); |
| 307 | + } |
| 308 | + $objectBuilder->build(); |
| 309 | + |
| 310 | + return true; |
| 311 | + } |
| 312 | + |
277 | 313 | /** |
278 | 314 | * @param string $argsObjectName |
279 | 315 | * @param array $arguments |
|
0 commit comments