added parameter $overwrite to addMember(), addMethod(), addProperty(), addConstant(), addCase() [Closes #152]

Author: dg

src/PhpGenerator/ClassType.php

@@ -177,7 +177,7 @@ public function removeImplement(string $name): static
 	}
 
 
-	public function addMember(Method|Property|Constant|TraitUse $member): static
+	public function addMember(Method|Property|Constant|TraitUse $member, bool $overwrite = false): static
 	{
 		$name = $member->getName();
 		[$type, $n] = match (true) {
@@ -186,7 +186,7 @@ public function addMember(Method|Property|Constant|TraitUse $member): static
 			$member instanceof Property => ['properties', $name],
 			$member instanceof TraitUse => ['traits', $name],
 		};
-		if (isset($this->$type[$n])) {
+		if (!$overwrite && isset($this->$type[$n])) {
 			throw new Nette\InvalidStateException("Cannot add member '$name', because it already exists.");
 		}
 		$this->$type[$n] = $member;

src/PhpGenerator/EnumType.php

@@ -99,9 +99,9 @@ public function getCases(): array
 
 
 	/** Adds case to enum */
-	public function addCase(string $name, string|int|Literal|null $value = null): EnumCase
+	public function addCase(string $name, string|int|Literal|null $value = null, bool $overwrite = false): EnumCase
 	{
-		if (isset($this->cases[$name])) {
+		if (!$overwrite && isset($this->cases[$name])) {
 			throw new Nette\InvalidStateException("Cannot add cases '$name', because it already exists.");
 		}
 		return $this->cases[$name] = (new EnumCase($name))
@@ -116,7 +116,7 @@ public function removeCase(string $name): static
 	}
 
 
-	public function addMember(Method|Constant|EnumCase|TraitUse $member): static
+	public function addMember(Method|Constant|EnumCase|TraitUse $member, bool $overwrite = false): static
 	{
 		$name = $member->getName();
 		[$type, $n] = match (true) {
@@ -125,7 +125,7 @@ public function addMember(Method|Constant|EnumCase|TraitUse $member): static
 			$member instanceof TraitUse => ['traits', $name],
 			$member instanceof EnumCase => ['cases', $name],
 		};
-		if (isset($this->$type[$n])) {
+		if (!$overwrite && isset($this->$type[$n])) {
 			throw new Nette\InvalidStateException("Cannot add member '$name', because it already exists.");
 		}
 		$this->$type[$n] = $member;

src/PhpGenerator/InterfaceType.php

@@ -51,14 +51,14 @@ public function addExtend(string $name): static
 	}
 
 
-	public function addMember(Method|Constant $member): static
+	public function addMember(Method|Constant $member, bool $overwrite = false): static
 	{
 		$name = $member->getName();
 		[$type, $n] = match (true) {
 			$member instanceof Constant => ['consts', $name],
 			$member instanceof Method => ['methods', strtolower($name)],
 		};
-		if (isset($this->$type[$n])) {
+		if (!$overwrite && isset($this->$type[$n])) {
 			throw new Nette\InvalidStateException("Cannot add member '$name', because it already exists.");
 		}
 		$this->$type[$n] = $member;

src/PhpGenerator/TraitType.php

@@ -22,7 +22,7 @@ final class TraitType extends ClassLike
 	use Traits\PropertiesAware;
 	use Traits\TraitsAware;
 
-	public function addMember(Method|Property|Constant|TraitUse $member): static
+	public function addMember(Method|Property|Constant|TraitUse $member, bool $overwrite = false): static
 	{
 		$name = $member->getName();
 		[$type, $n] = match (true) {
@@ -31,7 +31,7 @@ public function addMember(Method|Property|Constant|TraitUse $member): static
 			$member instanceof Property => ['properties', $name],
 			$member instanceof TraitUse => ['traits', $name],
 		};
-		if (isset($this->$type[$n])) {
+		if (!$overwrite && isset($this->$type[$n])) {
 			throw new Nette\InvalidStateException("Cannot add member '$name', because it already exists.");
 		}
 		$this->$type[$n] = $member;

src/PhpGenerator/Traits/ConstantsAware.php

@@ -48,9 +48,9 @@ public function getConstant(string $name): Constant
 	}
 
 
-	public function addConstant(string $name, mixed $value): Constant
+	public function addConstant(string $name, mixed $value, bool $overwrite = false): Constant
 	{
-		if (isset($this->consts[$name])) {
+		if (!$overwrite && isset($this->consts[$name])) {
 			throw new Nette\InvalidStateException("Cannot add constant '$name', because it already exists.");
 		}
 		return $this->consts[$name] = (new Constant($name))

src/PhpGenerator/Traits/MethodsAware.php

@@ -53,10 +53,10 @@ public function getMethod(string $name): Method
 	}
 
 
-	public function addMethod(string $name): Method
+	public function addMethod(string $name, bool $overwrite = false): Method
 	{
 		$lower = strtolower($name);
-		if (isset($this->methods[$lower])) {
+		if (!$overwrite && isset($this->methods[$lower])) {
 			throw new Nette\InvalidStateException("Cannot add method '$name', because it already exists.");
 		}
 		$method = new Method($name);

src/PhpGenerator/Traits/PropertiesAware.php

@@ -49,9 +49,9 @@ public function getProperty(string $name): Property
 
 
 	/** @param  string  $name  without $ */
-	public function addProperty(string $name, mixed $value = null): Property
+	public function addProperty(string $name, mixed $value = null, bool $overwrite = false): Property
 	{
-		if (isset($this->properties[$name])) {
+		if (!$overwrite && isset($this->properties[$name])) {
 			throw new Nette\InvalidStateException("Cannot add property '$name', because it already exists.");
 		}
 		return $this->properties[$name] = func_num_args() > 1

tests/PhpGenerator/ClassType.addMember.phpt

@@ -29,3 +29,6 @@ Assert::exception(
 	Nette\InvalidStateException::class,
 	"Cannot add member 'FOO', because it already exists.",
 );
+
+$class->addMember($new = new Nette\PhpGenerator\Method('FOO'), overwrite: true);
+Assert::same($new, $class->getMethod('FOO'));

tests/PhpGenerator/ClassType.phpt

@@ -217,6 +217,21 @@ Assert::exception(
 );
 
 
+// overwrite
+$class = new ClassType('Example');
+$class->addConstant('a', 1);
+$new = $class->addConstant('a', 1, overwrite: true);
+Assert::same($new, $class->getConstant('a'));
+
+$class->addProperty('a');
+$new = $class->addProperty('a', overwrite: true);
+Assert::same($new, $class->getProperty('a'));
+
+$class->addMethod('a');
+$new = $class->addMethod('a', overwrite: true);
+Assert::same($new, $class->getMethod('a'));
+
+
 // remove members
 $class = new ClassType('Example');
 $class->addConstant('a', 1);