ClassType: added addMember() [Closes #35]

Author: dg

readme.md

@@ -153,6 +153,19 @@ class Demo
 }
 ```
 
+You can also add existing `Method`, `Property` or `Constant` objects to the class:
+
+```php
+$method = new Nette\PhpGenerator\Method('getHandle');
+$property = new Nette\PhpGenerator\Property('handle');
+$const = new Nette\PhpGenerator\Constant('ROLE');
+
+$class = (new Nette\PhpGenerator\ClassType('Demo'))
+	->addMember($method)
+	->addMember($property)
+	->addMember($const);
+```
+
 Tabs versus spaces
 ------------------
 

src/PhpGenerator/ClassType.php

@@ -313,6 +313,32 @@ public function addTrait(string $name, array $resolutions = []): self
 	}
 
 
+	/**
+	 * @param  Method|Property|Constant  $member
+	 * @return static
+	 */
+	public function addMember($member): self
+	{
+		if ($member instanceof Method) {
+			if ($this->type === self::TYPE_INTERFACE) {
+				$member->setBody(null);
+			}
+			$this->methods[$member->getName()] = $member->setNamespace($this->namespace);
+
+		} elseif ($member instanceof Property) {
+			$this->properties[$member->getName()] = $member;
+
+		} elseif ($member instanceof Constant) {
+			$this->consts[$member->getName()] = $member;
+
+		} else {
+			throw new Nette\InvalidArgumentException('Argument must be Method|Property|Constant.');
+		}
+
+		return $this;
+	}
+
+
 	/**
 	 * @param  Constant[]|mixed[]  $consts
 	 * @return static

tests/PhpGenerator/ClassType.addMember.phpt

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+use Nette\PhpGenerator\ClassType;
+use Tester\Assert;
+
+
+require __DIR__ . '/../bootstrap.php';
+
+
+Assert::exception(function () {
+	(new ClassType('Example'))
+		->addMember(new stdClass);
+}, Nette\InvalidArgumentException::class, 'Argument must be Method|Property|Constant.');
+
+
+$class = (new ClassType('Example'))
+	->addMember($method = new Nette\PhpGenerator\Method('getHandle'))
+	->addMember($property = new Nette\PhpGenerator\Property('handle'))
+	->addMember($const = new Nette\PhpGenerator\Constant('ROLE'));
+
+Assert::same(['getHandle' => $method], $class->getMethods());
+Assert::same(['handle' => $property], $class->getProperties());
+Assert::same(['ROLE' => $const], $class->getConstants());
+Assert::same('', $method->getBody());
+
+
+$class = (new ClassType('Example'))
+	->setType('interface')
+	->addMember($method = new Nette\PhpGenerator\Method('getHandle'));
+
+Assert::null($method->getBody());