ClassType: setInterface() & setTrait() replaced with interface() & trait() factories

Author: dg

readme.md

@@ -219,12 +219,12 @@ echo $printer->printClass($class); // 4 spaces indentation
 Interface or Trait
 ------------------
 
-You can create interfaces and traits in a similar way, just change the type:
+You can create interfaces and traits:
 
 ```php
-$class = new Nette\PhpGenerator\ClassType('DemoInterface');
-$class->setInterface();
-// or $class->setTrait();
+$interface = Nette\PhpGenerator\ClassType::interface('MyInterface');
+$trait = Nette\PhpGenerator\ClassType::trait('MyTrait');
+// in a similar way $class = Nette\PhpGenerator\ClassType::class('MyClass');
 ```
 
 Literals

src/PhpGenerator/ClassType.php

@@ -68,6 +68,24 @@ final class ClassType
 	private $methods = [];
 
 
+	public static function class(string $name = null, PhpNamespace $namespace = null): self
+	{
+		return new self($name, $namespace);
+	}
+
+
+	public static function interface(string $name = null, PhpNamespace $namespace = null): self
+	{
+		return (new self($name, $namespace))->setType(self::TYPE_INTERFACE);
+	}
+
+
+	public static function trait(string $name = null, PhpNamespace $namespace = null): self
+	{
+		return (new self($name, $namespace))->setType(self::TYPE_TRAIT);
+	}
+
+
 	/**
 	 * @param  string|object  $class
 	 */
@@ -131,7 +149,7 @@ public function getName(): ?string
 	}
 
 
-	/** @return static */
+	/** @deprecated */
 	public function setClass(): self
 	{
 		$this->type = self::TYPE_CLASS;

src/PhpGenerator/PhpNamespace.php

@@ -179,13 +179,13 @@ public function addClass(string $name): ClassType
 
 	public function addInterface(string $name): ClassType
 	{
-		return $this->addClass($name)->setInterface();
+		return $this->addClass($name)->setType(ClassType::TYPE_INTERFACE);
 	}
 
 
 	public function addTrait(string $name): ClassType
 	{
-		return $this->addClass($name)->setTrait();
+		return $this->addClass($name)->setType(ClassType::TYPE_TRAIT);
 	}
 
 

tests/PhpGenerator/ClassType.interface.phpt

@@ -13,9 +13,8 @@ use Tester\Assert;
 require __DIR__ . '/../bootstrap.php';
 
 
-$interface = new ClassType('IExample');
+$interface = ClassType::interface('IExample');
 $interface
-	->setInterface()
 	->addExtend('IOne')
 	->addExtend('ITwo')
 	->addComment('Description of interface');