added PhpFile::removeNamespace()

nette · Apr 25, 2024 · 9afa824 · 9afa824
1 parent d815ee2
commit 9afa824
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/PhpGenerator/PhpFile.php b/src/PhpGenerator/PhpFile.php
@@ -110,6 +110,16 @@ public function addNamespace(string|PhpNamespace $namespace): PhpNamespace
 	}
 
 
+	/**
+	 * Removes the namespace from the file.
+	 */
+	public function removeNamespace(string|PhpNamespace $namespace): void
+	{
+		$name = $namespace instanceof PhpNamespace ? $namespace->getName() : $namespace;
+		unset($this->namespaces[$name]);
+	}
+
+
 	/** @return PhpNamespace[] */
 	public function getNamespaces(): array
 	{

diff --git a/tests/PhpGenerator/PhpFile.phpt b/tests/PhpGenerator/PhpFile.phpt
@@ -18,6 +18,10 @@ $file->addComment('This file is auto-generated. DO NOT EDIT!');
 $file->addComment('Hey there, I\'m here to document things.');
 
 
+$namespace = $file->addNamespace('Deleted');
+$namespace->addClass('Foo');
+$file->removeNamespace('Deleted');
+
 $namespaceFoo = $file->addNamespace('Foo');
 
 $classA = $namespaceFoo->addClass('A');