ClassType: added new setter/getters for each type

Author: dg

readme.md

@@ -238,8 +238,8 @@ Interface or Trait
 
 ```php
 $class = new Nette\PhpGenerator\ClassType('DemoInterface');
-$class->setType('interface');
-// or $class->setType('trait');
+$class->setInterface();
+// or $class->setTrait();
 ```
 
 Trait Resolutions and Visibility

src/PhpGenerator/ClassType.php

@@ -121,6 +121,48 @@ public function getName(): ?string
 	}
 
 
+	/** @return static */
+	public function setClass(): self
+	{
+		$this->type = self::TYPE_CLASS;
+		return $this;
+	}
+
+
+	public function isClass(): bool
+	{
+		return $this->type === self::TYPE_CLASS;
+	}
+
+
+	/** @return static */
+	public function setInterface(): self
+	{
+		$this->type = self::TYPE_INTERFACE;
+		return $this;
+	}
+
+
+	public function isInterface(): bool
+	{
+		return $this->type === self::TYPE_INTERFACE;
+	}
+
+
+	/** @return static */
+	public function setTrait(): self
+	{
+		$this->type = self::TYPE_TRAIT;
+		return $this;
+	}
+
+
+	public function isTrait(): bool
+	{
+		return $this->type === self::TYPE_TRAIT;
+	}
+
+
 	/** @return static */
 	public function setType(string $type): self
 	{
@@ -268,7 +310,7 @@ public function addTrait(string $name, array $resolutions = []): self
 	public function addMember($member): self
 	{
 		if ($member instanceof Method) {
-			if ($this->type === self::TYPE_INTERFACE) {
+			if ($this->isInterface()) {
 				$member->setBody(null);
 			}
 			$this->methods[$member->getName()] = $member;
@@ -418,7 +460,7 @@ public function getMethod(string $name): Method
 	public function addMethod(string $name): Method
 	{
 		$method = new Method($name);
-		if ($this->type === self::TYPE_INTERFACE) {
+		if ($this->isInterface()) {
 			$method->setBody(null);
 		} else {
 			$method->setPublic();

src/PhpGenerator/Factory.php

@@ -25,8 +25,8 @@ public function fromClassReflection(\ReflectionClass $from): ClassType
 			? new ClassType
 			: new ClassType($from->getShortName(), new PhpNamespace($from->getNamespaceName()));
 		$class->setType($from->isInterface() ? $class::TYPE_INTERFACE : ($from->isTrait() ? $class::TYPE_TRAIT : $class::TYPE_CLASS));
-		$class->setFinal($from->isFinal() && $class->getType() === $class::TYPE_CLASS);
-		$class->setAbstract($from->isAbstract() && $class->getType() === $class::TYPE_CLASS);
+		$class->setFinal($from->isFinal() && $class->isClass());
+		$class->setAbstract($from->isAbstract() && $class->isClass());
 
 		$ifaces = $from->getInterfaceNames();
 		foreach ($ifaces as $iface) {

src/PhpGenerator/PhpNamespace.php

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

tests/PhpGenerator/ClassType.interface.phpt

@@ -15,7 +15,7 @@ require __DIR__ . '/../bootstrap.php';
 
 $interface = new ClassType('IExample');
 $interface
-	->setType('interface')
+	->setInterface()
 	->addExtend('IOne')
 	->addExtend('ITwo')
 	->addComment('Description of interface');

tests/PhpGenerator/ClassType.phpt

@@ -18,6 +18,9 @@ $class = new ClassType('Example');
 
 Assert::false($class->isFinal());
 Assert::false($class->isAbstract());
+Assert::true($class->isClass());
+Assert::false($class->isInterface());
+Assert::false($class->isTrait());
 Assert::same([], $class->getExtends());
 Assert::same([], $class->getTraits());
 Assert::same([], $class->getTraitResolutions());