ClassType: added removeImplement() & removeTrait() [Closes #53]

Author: dg

src/PhpGenerator/ClassType.php

@@ -268,6 +268,17 @@ public function addImplement(string $name): self
 	}
 
 
+	/** @return static */
+	public function removeImplement(string $name): self
+	{
+		$key = array_search($name, $this->implements, true);
+		if ($key !== false) {
+			unset($this->implements[$key]);
+		}
+		return $this;
+	}
+
+
 	/**
 	 * @param  string[]  $names
 	 * @return static
@@ -303,6 +314,14 @@ public function addTrait(string $name, array $resolutions = []): self
 	}
 
 
+	/** @return static */
+	public function removeTrait(string $name): self
+	{
+		unset($this->traits[$name]);
+		return $this;
+	}
+
+
 	/**
 	 * @param  Method|Property|Constant  $member
 	 * @return static

tests/PhpGenerator/ClassType.phpt

@@ -123,6 +123,14 @@ $method->addParameter('res', null)
 		->setReference(true)
 		->setType(Type::ARRAY);
 
+
+$class->addTrait('foo');
+$class->removeTrait('foo');
+
+$class->addImplement('foo');
+$class->removeImplement('foo');
+
+
 sameFile(__DIR__ . '/expected/ClassType.expect', (string) $class);