feat: Implement setNamespace method

README.md

@@ -28,6 +28,10 @@ new PHPFileBuilder(app_path('Models/User.php'))
 
 ### Features
 
+#### setNamespace
+
+Set or replace the namespace declaration in a class, trait, enum, or interface file. If the file has no namespace, it will be added. If the namespace is already the same, no changes are made.
+
 #### setProperty
 
 Add new class property with the passed value and passed access level in case property does not exist in the class. Otherwise

src/Builders/PHPFileBuilder.php

@@ -19,6 +19,7 @@
 use RonasIT\Larabuilder\Visitors\PropertyVisitors\AddArrayPropertyItem;
 use RonasIT\Larabuilder\Visitors\PropertyVisitors\RemoveArrayPropertyItem;
 use RonasIT\Larabuilder\Visitors\PropertyVisitors\SetProperty;
+use RonasIT\Larabuilder\Visitors\SetNamespace;
 
 class PHPFileBuilder
 {
@@ -44,6 +45,13 @@ public function __construct(
         $this->traverser = new NodeTraverser();
     }
 
+    public function setNamespace(string $namespace): self
+    {
+        $this->traverser->addVisitor(new SetNamespace($namespace));
+
+        return $this;
+    }
+
     public function setProperty(string $name, mixed $value, AccessModifierEnum $accessModifier = AccessModifierEnum::Public): self
     {
         $this->traverser->addVisitor(new SetProperty($name, $value, $accessModifier));

src/Visitors/SetNamespace.php

@@ -0,0 +1,56 @@
+<?php
+
+namespace RonasIT\Larabuilder\Visitors;
+
+use PhpParser\Node\Name;
+use PhpParser\Node\Stmt\Class_;
+use PhpParser\Node\Stmt\Declare_;
+use PhpParser\Node\Stmt\Enum_;
+use PhpParser\Node\Stmt\Interface_;
+use PhpParser\Node\Stmt\Namespace_;
+use PhpParser\Node\Stmt\Trait_;
+
+class SetNamespace extends AbstractNodeVisitor
+{
+    protected array $allowedParentNodesTypes = [
+        Class_::class,
+        Trait_::class,
+        Enum_::class,
+        Interface_::class,
+    ];
+
+    public function __construct(
+        protected string $namespace,
+    ) {
+    }
+
+    public function afterTraverse(array $nodes): ?array
+    {
+        parent::afterTraverse($nodes);
+
+        $declares = [];
+
+        foreach ($nodes as $key => $node) {
+            if ($node instanceof Namespace_) {
+                return $this->updateNamespace($node);
+            }
+
+            if ($node instanceof Declare_) {
+                $declares[] = $node;
+
+                unset($nodes[$key]);
+            }
+        }
+
+        return [...$declares, new Namespace_(new Name($this->namespace), array_values($nodes))];
+    }
+
+    protected function updateNamespace(Namespace_ $node): ?array
+    {
+        if ($node->name->toString() !== $this->namespace) {
+            $node->name = new Name($this->namespace);
+        }
+
+        return null;
+    }
+}

tests/PHPFileBuilderTest.php

@@ -20,6 +20,59 @@ class PHPFileBuilderTest extends TestCase
 {
     use PHPFileBuilderTestMockTrait;
 
+    public function testSetNamespaceOnFileWithoutNamespace(): void
+    {
+        $file = $this->generateOriginalStructurePath('class_without_namespace.php');
+
+        $this->mockNativeFunction(
+            'RonasIT\Larabuilder\Builders',
+            $this->callFilePutContent($file, 'namespace_set.php'),
+        );
+
+        new PHPFileBuilder($file)
+            ->setNamespace('App\\Models')
+            ->save();
+    }
+
+    public function testSetNamespaceReplacesExistingNamespace(): void
+    {
+        $file = $this->generateOriginalStructurePath('enum.php');
+
+        $this->mockNativeFunction(
+            'RonasIT\Larabuilder\Builders',
+            $this->callFilePutContent($file, 'namespace_update.php'),
+        );
+
+        new PHPFileBuilder($file)
+            ->setNamespace('App\\Models')
+            ->save();
+    }
+
+    public function testSetNamespaceDoesNothingWhenSameNamespace(): void
+    {
+        $file = $this->generateOriginalStructurePath('class.php');
+
+        $this->mockNativeFunction(
+            'RonasIT\Larabuilder\Builders',
+            $this->callFilePutContent($file, 'class_unchanged.php'),
+        );
+
+        new PHPFileBuilder($file)
+            ->setNamespace('RonasIT\Larabuilder\Tests\Support')
+            ->save();
+    }
+
+    public function testSetNamespaceNotClassTraitEnumInterface(): void
+    {
+        $file = $this->generateOriginalStructurePath('bootstrap_empty.php');
+
+        $this->assertExceptionThrew(InvalidStructureTypeException::class, "'SetNamespace' operation may only be applied to: Class, Trait, Enum, Interface.");
+
+        new PHPFileBuilder($file)
+            ->setNamespace('RonasIT\Larabuilder\Tests\Support')
+            ->save();
+    }
+
     public function testSetProperty(): void
     {
         $file = $this->generateOriginalStructurePath('class_with_properties.php');

tests/Support/OriginStructures/class_without_namespace.php

@@ -0,0 +1,7 @@
+<?php
+
+declare(strict_types=1);
+
+class SomeClass
+{
+}

tests/fixtures/PHPFileBuilderTest/namespace_set.php

@@ -0,0 +1,9 @@
+<?php
+
+declare(strict_types=1);
+
+namespace App\Models;
+
+class SomeClass
+{
+}

tests/fixtures/PHPFileBuilderTest/namespace_update.php

@@ -0,0 +1,14 @@
+<?php
+
+namespace App\Models;
+
+enum SomeEnum
+{
+    case First;
+    case Second;
+
+    public static function toArray(): array
+    {
+        return self::cases();
+    }
+}