PhpNamespace: added removeClass() & removeFunction()

Author: dg

src/PhpGenerator/PhpNamespace.php

@@ -281,6 +281,13 @@ public function addEnum(string $name): ClassType
 	}
 
 
+	public function removeClass(string $name): self
+	{
+		unset($this->classes[strtolower($name)]);
+		return $this;
+	}
+
+
 	public function addFunction(string $name): GlobalFunction
 	{
 		$lower = strtolower($name);
@@ -292,6 +299,13 @@ public function addFunction(string $name): GlobalFunction
 	}
 
 
+	public function removeFunction(string $name): self
+	{
+		unset($this->functions[strtolower($name)]);
+		return $this;
+	}
+
+
 	/** @return ClassType[] */
 	public function getClasses(): array
 	{

tests/PhpGenerator/PhpNamespace.phpt

@@ -22,3 +22,11 @@ Assert::same($namespace, $interfaceB->getNamespace());
 
 Assert::count(2, $namespace->getClasses());
 Assert::type(Nette\PhpGenerator\ClassType::class, $namespace->getClasses()['A']);
+$namespace->removeClass('a');
+Assert::count(1, $namespace->getClasses());
+
+$function = $namespace->addFunction('foo');
+Assert::count(1, $namespace->getFunctions());
+Assert::same($function, $namespace->getFunctions()['foo']);
+$namespace->removeFunction('FOO');
+Assert::count(0, $namespace->getFunctions());