Add has* methods to ClassBuilder - Close #48

Author: sandrokeil

src/Builder/ClassBuilder.php

@@ -122,9 +122,13 @@ public function setExtends(string $extends): self
         return $this;
     }
 
-    public function setImplements(string ...$interfaces): self
+    public function setImplements(string ...$implements): self
     {
-        $this->implements = $interfaces;
+        $this->implements = [];
+
+        foreach ($implements as $implement) {
+            $this->implements[$implement] = $implement;
+        }
 
         return $this;
     }
@@ -144,6 +148,11 @@ public function addImplement(string ...$implements): self
         return $this;
     }
 
+    public function hasImplement(string $implement): bool
+    {
+        return isset($this->implements[$implement]);
+    }
+
     public function setNamespace(string $namespace): self
     {
         $this->namespace = $namespace;
@@ -198,6 +207,11 @@ public function removeNamespaceImport(string ...$namespaceImports): self
         return $this;
     }
 
+    public function hasNamespaceImport(string $namespace): bool
+    {
+        return isset($this->namespaceImports[$namespace]);
+    }
+
     /**
      * @deprecated Use setNamespaceImports()
      * @param string ...$namespaces
@@ -255,6 +269,11 @@ public function removeTrait(string ...$traits): self
         return $this;
     }
 
+    public function hasTrait(string $trait): bool
+    {
+        return isset($this->traits[$trait]);
+    }
+
     /**
      * @deprecated Use setTraits()
      * @param string ...$traits
@@ -273,7 +292,11 @@ public function setUseTrait(string ...$traits): self
      */
     public function setConstants(ClassConstBuilder ...$constants): self
     {
-        $this->constants = $constants;
+        $this->constants = [];
+
+        foreach ($constants as $constant) {
+            $this->constants[$constant->getName()] = $constant;
+        }
 
         return $this;
     }
@@ -308,6 +331,11 @@ public function removeConstant(string ...$constants): self
         return $this;
     }
 
+    public function hasConstant(string $constantName): bool
+    {
+        return isset($this->constants[$constantName]);
+    }
+
     /**
      * Replacing will not work on existing files
      *
@@ -355,6 +383,11 @@ public function removeProperty(string ...$propertyNames): self
         return $this;
     }
 
+    public function hasProperty(string $propertyName): bool
+    {
+        return isset($this->properties[$propertyName]);
+    }
+
     public function setMethods(ClassMethodBuilder ...$methods): self
     {
         $this->methods = [];
@@ -396,6 +429,11 @@ public function removeMethod(string ...$methodNames): self
         return $this;
     }
 
+    public function hasMethod(string $methodName): bool
+    {
+        return isset($this->methods[$methodName]);
+    }
+
     public function getNamespace(): ?string
     {
         return $this->namespace;

tests/Builder/ClassBuilderTest.php

@@ -52,6 +52,12 @@ public function it_generates_class_for_empty_file(): void
             ->setTraits('\\My\\TestTrait')
             ->setConstants(ClassConstBuilder::fromScratch('PRIV', 'private')->setPrivate());
 
+        $this->assertTrue($classFactory->hasConstant('PRIV'));
+        $this->assertTrue($classFactory->hasImplement('\\Iterator'));
+        $this->assertTrue($classFactory->hasImplement('Bar'));
+        $this->assertTrue($classFactory->hasTrait('\\My\\TestTrait'));
+        $this->assertTrue($classFactory->hasNamespaceImport('Foo\\Bar'));
+
         $nodeTraverser = new NodeTraverser();
         $classFactory->injectVisitors($nodeTraverser, $this->parser);
 

tests/Builder/ClassMethodBuilderTest.php

@@ -52,6 +52,9 @@ public function it_generates_method_for_empty_class(): void
             ClassMethodBuilder::fromScratch('doSomething')->setReturnType('void')->setFinal(true)
         );
 
+        $this->assertTrue($classFactory->hasMethod('setActive'));
+        $this->assertTrue($classFactory->hasMethod('doSomething'));
+
         $methods = $classFactory->getMethods();
 
         $this->assertCount(2, $methods);

tests/Builder/ClassPropertyBuilderTest.php

@@ -46,6 +46,8 @@ public function it_generates_property_for_empty_class(): void
         $classFactory = ClassBuilder::fromScratch('TestClass', 'My\\Awesome\\Service');
         $classFactory->setProperties(ClassPropertyBuilder::fromScratch('aggregateId', 'string'));
 
+        $this->assertTrue($classFactory->hasProperty('aggregateId'));
+
         $nodeTraverser = new NodeTraverser();
         $classFactory->injectVisitors($nodeTraverser, $this->parser);