added PhpNamespace::add() [Closes #27]

Author: dg

readme.md

@@ -390,6 +390,10 @@ $namespace = new Nette\PhpGenerator\PhpNamespace('Foo');
 $class = $namespace->addClass('Task');
 $interface = $namespace->addInterface('Countable');
 $trait = $namespace->addTrait('NameAware');
+
+// or
+$class = new Nette\PhpGenerator\ClassType('Task');
+$namespace->add($class);
 ```
 
 If the class already exists, it will be overwritten.

src/PhpGenerator/PhpNamespace.php

@@ -145,10 +145,25 @@ public function unresolveName(string $name): string
 	}
 
 
-	public function addClass(string $name): ClassType
+	/**
+	 * @return static
+	 */
+	public function add(ClassType $class): self
 	{
+		$name = $class->getName();
+		if ($name === null) {
+			throw new Nette\InvalidArgumentException('Class does not have a name.');
+		}
 		$this->addUse($this->name . '\\' . $name);
-		return $this->classes[$name] = new ClassType($name, $this);
+		$this->classes[$name] = $class;
+		return $this;
+	}
+
+
+	public function addClass(string $name): ClassType
+	{
+		$this->add($class = new ClassType($name, $this));
+		return $class;
 	}
 
 

tests/PhpGenerator/PhpNamespace.add.phpt

@@ -0,0 +1,35 @@
+<?php
+
+declare(strict_types=1);
+
+use Nette\PhpGenerator\ClassType;
+use Nette\PhpGenerator\PhpNamespace;
+use Tester\Assert;
+
+require __DIR__ . '/../bootstrap.php';
+
+
+Assert::exception(function () {
+	(new PhpNamespace('Foo'))->add(new ClassType);
+}, Nette\InvalidArgumentException::class, 'Class does not have a name.');
+
+
+$namespace = (new PhpNamespace('Foo'))
+	->add($classA = new ClassType('A'))
+	->add($classB = new ClassType('B', new PhpNamespace('X')));
+
+
+same('namespace Foo;
+
+class A
+{
+}
+
+class B
+{
+}
+', (string) $namespace);
+
+// namespaces are not changed
+Assert::null($classA->getNamespace());
+Assert::same('X', $classB->getNamespace()->getName());