added ClassType::add() [Closes #35]

readme.md

@@ -402,6 +402,7 @@ class Demo implements A
 }
 ```
 
+TODO: import()
 
 Generate using Reflection
 -------------------------

src/PhpGenerator/ClassType.php

@@ -400,6 +400,32 @@ public function addMethod(string $name): Method
 	}
 
 
+	/**
+	 * @param  Method|Property|Constant  $member
+	 * @return static
+	 */
+	public function add($member): self
+	{
+		if ($member instanceof Method) {
+			if ($this->type === 'interface') {
+				$member->setBody(null);
+			}
+			$this->methods[$member->getName()] = $member;
+
+		} 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;
+	}
+
+
 	private function validate(array $names): void
 	{
 		foreach ($names as $name) {

tests/PhpGenerator/ClassType.add.phpt

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