Refactor

Author: ruudk

composer.json

@@ -5,7 +5,7 @@
     "type": "library",
     "require": {
         "php": "^8.4",
-        "ruudk/code-generator": "^0.3.0",
+        "ruudk/code-generator": "^0.4.3",
         "symfony/console": "^7.3",
         "symfony/filesystem": "^7.3",
         "symfony/finder": "^7.3",

composer.lock

@@ -5,7 +5,8 @@
 namespace Ruudk\GraphQLCodeGenerator\Console;
 
 use Ruudk\GraphQLCodeGenerator\Config;
-use Ruudk\GraphQLCodeGenerator\GraphQLCodeGenerator;
+use Ruudk\GraphQLCodeGenerator\Executor\PlanExecutor;
+use Ruudk\GraphQLCodeGenerator\Planner;
 use Symfony\Component\Console\Attribute\AsCommand;
 use Symfony\Component\Console\Attribute\Option;
 use Symfony\Component\Console\Command\Command;
@@ -99,8 +100,8 @@ public function __invoke(
             $io->write(sprintf('Generating code for <info>%s</info>... ', $configItem->namespace));
 
             try {
-                $generator = new GraphQLCodeGenerator($configItem);
-                $files = $generator->generate();
+                $plan = new Planner($configItem)->plan();
+                $files = new PlanExecutor($configItem)->execute($plan);
 
                 // Clear output directory
                 $filesystem->remove($configItem->outputDir);

src/Console/GenerateCommand.php

@@ -0,0 +1,153 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Ruudk\GraphQLCodeGenerator\Executor;
+
+use JsonException;
+use LogicException;
+use Ruudk\GraphQLCodeGenerator\Config;
+use Ruudk\GraphQLCodeGenerator\Generator\DataClassGenerator;
+use Ruudk\GraphQLCodeGenerator\Generator\EnumTypeGenerator;
+use Ruudk\GraphQLCodeGenerator\Generator\ErrorClassGenerator;
+use Ruudk\GraphQLCodeGenerator\Generator\ExceptionClassGenerator;
+use Ruudk\GraphQLCodeGenerator\Generator\InputTypeGenerator;
+use Ruudk\GraphQLCodeGenerator\Generator\NodeNotFoundExceptionGenerator;
+use Ruudk\GraphQLCodeGenerator\Generator\OperationClassGenerator;
+use Ruudk\GraphQLCodeGenerator\Planner\OperationPlan;
+use Ruudk\GraphQLCodeGenerator\Planner\Plan\DataClassPlan;
+use Ruudk\GraphQLCodeGenerator\Planner\Plan\EnumClassPlan;
+use Ruudk\GraphQLCodeGenerator\Planner\Plan\InputClassPlan;
+use Ruudk\GraphQLCodeGenerator\Planner\Plan\NodeNotFoundExceptionPlan;
+use Ruudk\GraphQLCodeGenerator\Planner\PlannerResult;
+use Ruudk\GraphQLCodeGenerator\TypeInitializer\BackedEnumTypeInitializer;
+use Ruudk\GraphQLCodeGenerator\TypeInitializer\CollectionTypeInitializer;
+use Ruudk\GraphQLCodeGenerator\TypeInitializer\DelegatingTypeInitializer;
+use Ruudk\GraphQLCodeGenerator\TypeInitializer\NullableTypeInitializer;
+use Ruudk\GraphQLCodeGenerator\TypeInitializer\ObjectTypeInitializer;
+use Ruudk\GraphQLCodeGenerator\TypeMapper;
+use Symfony\Component\TypeInfo\Type as SymfonyType;
+
+final class PlanExecutor
+{
+    private readonly DataClassGenerator $dataClassGenerator;
+    private readonly EnumTypeGenerator $enumTypeGenerator;
+    private readonly OperationClassGenerator $operationClassGenerator;
+    private readonly ErrorClassGenerator $errorClassGenerator;
+    private readonly ExceptionClassGenerator $exceptionClassGenerator;
+    private readonly InputTypeGenerator $inputTypeGenerator;
+    private readonly NodeNotFoundExceptionGenerator $nodeNotFoundExceptionGenerator;
+
+    public function __construct(
+        Config $config,
+    ) {
+        // Initialize type initializer
+        $typeInitializer = new DelegatingTypeInitializer(
+            new NullableTypeInitializer(),
+            new CollectionTypeInitializer(),
+            new BackedEnumTypeInitializer($config->addUnknownCaseToEnums, $config->namespace),
+            new ObjectTypeInitializer(),
+            ...$config->typeInitializers,
+        );
+
+        // Initialize scalars
+        $scalars = [
+            'ID' => SymfonyType::string(),
+            'String' => SymfonyType::string(),
+            'Int' => SymfonyType::int(),
+            'Float' => SymfonyType::float(),
+            'Boolean' => SymfonyType::bool(),
+            ...$config->scalars,
+        ];
+
+        // Initialize enum types
+        $enumTypes = $config->enumTypes;
+
+        // Initialize input object types
+        $inputObjectTypes = $config->inputObjectTypes;
+
+        // Initialize object types
+        $objectTypes = $config->objectTypes;
+
+        // Create TypeMapper
+        $typeMapper = new TypeMapper(
+            $scalars,
+            $enumTypes,
+            $inputObjectTypes,
+            $objectTypes,
+        );
+
+        // Initialize all generators
+        $this->dataClassGenerator = new DataClassGenerator($config, $typeInitializer);
+        $this->enumTypeGenerator = new EnumTypeGenerator($config);
+        $this->operationClassGenerator = new OperationClassGenerator($config);
+        $this->errorClassGenerator = new ErrorClassGenerator($config);
+        $this->exceptionClassGenerator = new ExceptionClassGenerator($config);
+        $this->inputTypeGenerator = new InputTypeGenerator($config, $typeMapper);
+        $this->nodeNotFoundExceptionGenerator = new NodeNotFoundExceptionGenerator($config);
+    }
+
+    /**
+     * @throws LogicException
+     * @throws JsonException
+     * @return array<string, string>
+     */
+    public function execute(PlannerResult $plan) : array
+    {
+        $files = [];
+
+        foreach ($plan->classes as $relativePath => $class) {
+            $files[$relativePath] = $this->generateClass($class);
+        }
+
+        foreach ($plan->operations as $operation) {
+            $files = array_merge($files, $this->generateOperation($operation));
+        }
+
+        return $files;
+    }
+
+    /**
+     * @throws LogicException
+     */
+    private function generateClass(object $class) : string
+    {
+        return match ($class::class) {
+            DataClassPlan::class => $this->dataClassGenerator->generate($class),
+
+            EnumClassPlan::class => $this->enumTypeGenerator->generate($class),
+
+            InputClassPlan::class => $this->inputTypeGenerator->generate($class),
+
+            NodeNotFoundExceptionPlan::class => $this->nodeNotFoundExceptionGenerator->generate(),
+
+            default => throw new LogicException('Unknown class type: ' . $class::class),
+        };
+    }
+
+    /**
+     * @throws JsonException
+     * @return array<string, string>
+     */
+    private function generateOperation(OperationPlan $operation) : array
+    {
+        $files = [];
+
+        // Generate operation class
+        $files[$operation->operationClass->relativePath] = $this->operationClassGenerator->generate(
+            $operation->operationClass,
+        );
+
+        // Generate error class
+        $files[$operation->errorClass->relativePath] = $this->errorClassGenerator->generate(
+            $operation->errorClass,
+        );
+
+        // Generate exception class
+        $files[$operation->exceptionClass->relativePath] = $this->exceptionClassGenerator->generate(
+            $operation->exceptionClass,
+        );
+
+        return $files;
+    }
+}