Remove enum support from import method in favor of dedicated importEnum method (#1)

* Initial plan

* Initial plan to extract enum handling from import method

Co-authored-by: ruudk <104180+ruudk@users.noreply.github.com>

* Extract enum handling to separate importEnum method

Co-authored-by: ruudk <104180+ruudk@users.noreply.github.com>

* Fix importEnum to call import method and remove enum support from import

Co-authored-by: ruudk <104180+ruudk@users.noreply.github.com>

* Inline alias variable in importEnum method

Co-authored-by: ruudk <104180+ruudk@users.noreply.github.com>

* Revert composer.lock to original state

Co-authored-by: ruudk <104180+ruudk@users.noreply.github.com>

---------

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: ruudk <104180+ruudk@users.noreply.github.com>

Author: Copilot

src/CodeGenerator.php

@@ -178,9 +178,17 @@ private function findAvailableAlias(Alias | FullyQualified | FunctionName | Name
     }
 
     /**
-     * Imports a class, function, or enum and returns the alias to use in the generated code
+     * Imports an enum and returns the fully qualified reference to use in the generated code
      */
-    public function import(FullyQualified | FunctionName | string | UnitEnum $fqcnOrEnum) : string
+    public function importEnum(UnitEnum $enum) : string
+    {
+        return sprintf('%s::%s', $this->import($enum::class), $enum->name);
+    }
+
+    /**
+     * Imports a class or function and returns the alias to use in the generated code
+     */
+    public function import(FullyQualified | FunctionName | string $fqcnOrEnum) : string
     {
         if ($fqcnOrEnum instanceof FunctionName) {
             $alias = $this->findAvailableAlias($fqcnOrEnum, $fqcnOrEnum->shortName);
@@ -189,14 +197,6 @@ public function import(FullyQualified | FunctionName | string | UnitEnum $fqcnOr
             return $alias;
         }
 
-        if ($fqcnOrEnum instanceof UnitEnum) {
-            $fqcn = new FullyQualified($fqcnOrEnum::class);
-            $alias = $this->findAvailableAlias($fqcn, $fqcn->className->name);
-            $this->imports[$alias] = $fqcn;
-
-            return sprintf('%s::%s', $alias, $fqcnOrEnum->name);
-        }
-
         $fqcn = FullyQualified::maybeFromString($fqcnOrEnum);
         $alias = $this->findAvailableAlias($fqcn, $fqcn->className->name);
         $this->imports[$alias] = $fqcn;

tests/CodeGeneratorTest.php

@@ -191,7 +191,7 @@ public function testImportFunction() : void
 
     public function testImportEnum() : void
     {
-        $reference = $this->generator->import(TestEnum::OPTION_ONE);
+        $reference = $this->generator->importEnum(TestEnum::OPTION_ONE);
 
         self::assertSame('TestEnum::OPTION_ONE', $reference);
 
@@ -208,6 +208,25 @@ public function testImportEnum() : void
         );
     }
 
+    public function testImportEnumDirectly() : void
+    {
+        $reference = $this->generator->importEnum(TestEnum::OPTION_TWO);
+
+        self::assertSame('TestEnum::OPTION_TWO', $reference);
+
+        $this->assertDumpFile(
+            <<<'PHP'
+                <?php
+
+                declare(strict_types=1);
+
+                use Ruudk\CodeGenerator\Fixtures\TestEnum;
+
+                PHP,
+            [],
+        );
+    }
+
     public function testImportSameNamespace() : void
     {
         $this->generator = new CodeGenerator('App\\Models');