ClassType: added removeMethod(), removeProperty(), removeConstant()

Author: dg

readme.md

@@ -119,6 +119,8 @@ It results in:
 
 If the property, constant, method or parameter already exist, it will be overwritten.
 
+Members can be removed using `removeProperty()`, `removeConstant()`, `removeMethod()` or `removeParameter()`.
+
 PHP Generator supports all new PHP 7.2 features:
 
 ```php

src/PhpGenerator/ClassType.php

@@ -369,6 +369,16 @@ public function addConstant(string $name, $value): Constant
 	}
 
 
+	/**
+	 * @return static
+	 */
+	public function removeConstant(string $name): self
+	{
+		unset($this->consts[$name]);
+		return $this;
+	}
+
+
 	/**
 	 * @param  Property[]  $props
 	 * @return static
@@ -413,6 +423,17 @@ public function addProperty(string $name, $value = null): Property
 	}
 
 
+	/**
+	 * @param  string  $name without $
+	 * @return static
+	 */
+	public function removeProperty(string $name): self
+	{
+		unset($this->properties[$name]);
+		return $this;
+	}
+
+
 	/**
 	 * @param  Method[]  $methods
 	 * @return static
@@ -460,6 +481,16 @@ public function addMethod(string $name): Method
 	}
 
 
+	/**
+	 * @return static
+	 */
+	public function removeMethod(string $name): self
+	{
+		unset($this->methods[$name]);
+		return $this;
+	}
+
+
 	private function validate(array $names): void
 	{
 		foreach ($names as $name) {

tests/PhpGenerator/ClassType.phpt

@@ -123,3 +123,24 @@ Assert::exception(function () {
 	$class = new ClassType;
 	$class->addMethod('method')->setVisibility('unknown');
 }, Nette\InvalidArgumentException::class, 'Argument must be public|protected|private.');
+
+
+// remove members
+$class = new ClassType('Example');
+$class->addConstant('a', 1);
+$class->addConstant('b', 1);
+$class->removeConstant('b')->removeConstant('c');
+
+Assert::same(['a'], array_keys($class->getConstants()));
+
+$class->addProperty('a');
+$class->addProperty('b');
+$class->removeProperty('b')->removeProperty('c');
+
+Assert::same(['a'], array_keys($class->getProperties()));
+
+$class->addMethod('a');
+$class->addMethod('b');
+$class->removeMethod('b')->removeMethod('c');
+
+Assert::same(['a'], array_keys($class->getMethods()));