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

Author: dg

src/PhpGenerator/ClassType.php

@@ -85,6 +85,15 @@ public function __toString(): string
 	}
 
 
+	/**
+	 * @internal use PhpNamespace::add()
+	 */
+	public function setNamespace(?PhpNamespace $val): void
+	{
+		$this->namespace = $val;
+	}
+
+
 	public function getNamespace(): ?PhpNamespace
 	{
 		return $this->namespace;

src/PhpGenerator/PhpNamespace.php

@@ -144,13 +144,26 @@ public function unresolveName(string $name): string
 	}
 
 
-	public function addClass(string $name): ClassType
+	/**
+	 * @return static
+	 */
+	public function add(ClassType $class): self
 	{
-		if (isset($this->classes[$name])) {
-			throw new Nette\InvalidArgumentException("Class $name was already added.");
+		$name = $class->getName();
+		if (!$name) {
+			throw new Nette\InvalidArgumentException('Class does not have a name.');
 		}
 		$this->addUse($this->name . '\\' . $name);
-		return $this->classes[$name] = new ClassType($name, $this);
+		$class->setNamespace($this);
+		$this->classes[$name] = $class;
+		return $this;
+	}
+
+
+	public function addClass(string $name): ClassType
+	{
+		$this->add($class = new ClassType($name));
+		return $class;
 	}
 
 

tests/PhpGenerator/PhpNamespace.add.phpt

@@ -0,0 +1,22 @@
+<?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')));
+
+Assert::same($namespace, $classA->getNamespace());
+Assert::same($namespace, $classB->getNamespace());