ClassType: checks whether names are valid

Author: dg

src/PhpGenerator/ClassType.php

@@ -218,9 +218,10 @@ public function isAbstract(): bool
 	 */
 	public function setExtends($names): self
 	{
-		if (!is_string($names) && !(is_array($names) && Nette\Utils\Arrays::every($names, 'is_string'))) {
+		if (!is_string($names) && !is_array($names)) {
 			throw new Nette\InvalidArgumentException('Argument must be string or string[].');
 		}
+		$this->validate((array) $names);
 		$this->extends = $names;
 		return $this;
 	}
@@ -240,6 +241,7 @@ public function getExtends()
 	 */
 	public function addExtend(string $name): self
 	{
+		$this->validate([$name]);
 		$this->extends = (array) $this->extends;
 		$this->extends[] = $name;
 		return $this;
@@ -252,6 +254,7 @@ public function addExtend(string $name): self
 	 */
 	public function setImplements(array $names): self
 	{
+		$this->validate($names);
 		$this->implements = $names;
 		return $this;
 	}
@@ -271,6 +274,7 @@ public function getImplements(): array
 	 */
 	public function addImplement(string $name): self
 	{
+		$this->validate([$name]);
 		$this->implements[] = $name;
 		return $this;
 	}
@@ -282,6 +286,7 @@ public function addImplement(string $name): self
 	 */
 	public function setTraits(array $names): self
 	{
+		$this->validate($names);
 		$this->traits = array_fill_keys($names, []);
 		return $this;
 	}
@@ -301,6 +306,7 @@ public function getTraits(): array
 	 */
 	public function addTrait(string $name, array $resolutions = []): self
 	{
+		$this->validate([$name]);
 		$this->traits[$name] = $resolutions;
 		return $this;
 	}
@@ -459,4 +465,14 @@ public function addMethod(string $name): Method
 		return $this->methods[$name] = $method;
 	}
 
+
+	private function validate(array $names)
+	{
+		foreach ($names as $name) {
+			if (!Helpers::isNamespaceIdentifier($name, TRUE)) {
+				throw new Nette\InvalidArgumentException("Value '$name' is not valid class name.");
+			}
+		}
+	}
+
 }

tests/PhpGenerator/invalidNames.phpt

@@ -63,6 +63,36 @@ Assert::exception(function () {
 }, Nette\InvalidArgumentException::class);
 
 
+$class = new Nette\PhpGenerator\ClassType('Abc');
+Assert::exception(function () use ($class) {
+	$class->setExtends('*');
+}, Nette\InvalidArgumentException::class, "Value '*' is not valid class name.");
+
+Assert::exception(function () use ($class) {
+	$class->setExtends(['A', '*']);
+}, Nette\InvalidArgumentException::class, "Value '*' is not valid class name.");
+
+Assert::exception(function () use ($class) {
+	$class->addExtend('*');
+}, Nette\InvalidArgumentException::class, "Value '*' is not valid class name.");
+
+Assert::exception(function () use ($class) {
+	$class->setImplements(['A', '*']);
+}, Nette\InvalidArgumentException::class, "Value '*' is not valid class name.");
+
+Assert::exception(function () use ($class) {
+	$class->addImplement('*');
+}, Nette\InvalidArgumentException::class, "Value '*' is not valid class name.");
+
+Assert::exception(function () use ($class) {
+	$class->setTraits(['A', '*']);
+}, Nette\InvalidArgumentException::class, "Value '*' is not valid class name.");
+
+Assert::exception(function () use ($class) {
+	$class->addTrait('*');
+}, Nette\InvalidArgumentException::class, "Value '*' is not valid class name.");
+
+
 Assert::noError(function () {
 	new Nette\PhpGenerator\Property('Iñtërnâtiônàlizætiøn');
 });