VisibilityAware: added new setter/getters for each visiblity

Author: dg

readme.md

@@ -78,7 +78,7 @@ We can add constants and properties:
 $class->addConstant('ID', 123);
 
 $class->addProperty('items', [1, 2, 3])
-	->setVisibility('private')
+	->setPrivate()
 	->setStatic()
 	->addComment('@var int[]');
 ```
@@ -99,7 +99,7 @@ $method = $class->addMethod('count')
 	->addComment('Count it.')
 	->addComment('@return int')
 	->setFinal()
-	->setVisibility('protected')
+	->setProtected()
 	->setBody('return count($items ?: $this->items);');
 
 $method->addParameter('items', []) // $items = []
@@ -130,7 +130,7 @@ PHP Generator supports all new PHP 7.3 and 7.4 features:
 $class = new Nette\PhpGenerator\ClassType('Demo');
 
 $class->addConstant('ID', 123)
-	->setVisibility('private'); // constant visiblity
+	->setPrivate(); // constant visiblity
 
 $class->addProperty('items')
 	->setType('array') // typed properites

src/PhpGenerator/ClassType.php

@@ -421,7 +421,7 @@ public function addMethod(string $name): Method
 		if ($this->type === self::TYPE_INTERFACE) {
 			$method->setBody(null);
 		} else {
-			$method->setVisibility(self::VISIBILITY_PUBLIC);
+			$method->setPublic();
 		}
 		return $this->methods[$name] = $method;
 	}

src/PhpGenerator/Traits/VisibilityAware.php

@@ -40,4 +40,46 @@ public function getVisibility(): ?string
 	{
 		return $this->visibility;
 	}
+
+
+	/** @return static */
+	public function setPublic(): self
+	{
+		$this->visibility = ClassType::VISIBILITY_PUBLIC;
+		return $this;
+	}
+
+
+	public function isPublic(): bool
+	{
+		return $this->visibility === ClassType::VISIBILITY_PUBLIC || $this->visibility === null;
+	}
+
+
+	/** @return static */
+	public function setProtected(): self
+	{
+		$this->visibility = ClassType::VISIBILITY_PROTECTED;
+		return $this;
+	}
+
+
+	public function isProtected(): bool
+	{
+		return $this->visibility === ClassType::VISIBILITY_PROTECTED;
+	}
+
+
+	/** @return static */
+	public function setPrivate(): self
+	{
+		$this->visibility = ClassType::VISIBILITY_PRIVATE;
+		return $this;
+	}
+
+
+	public function isPrivate(): bool
+	{
+		return $this->visibility === ClassType::VISIBILITY_PRIVATE;
+	}
 }

tests/PhpGenerator/ClassType.phpt

@@ -47,7 +47,7 @@ $class->addConstant('FORCE_ARRAY', new PhpLiteral('Nette\Utils\Json::FORCE_ARRAY
 	->addComment('Commented');
 
 $class->addProperty('handle')
-	->setVisibility('private')
+	->setPrivate()
 	->addComment('@var resource  orignal file handle');
 
 $class->addProperty('order')
@@ -69,6 +69,9 @@ Assert::true($class->hasProperty('sections'));
 Assert::false($class->hasProperty('unknown'));
 Assert::true($p->isStatic());
 Assert::null($p->getVisibility());
+Assert::false($p->isPrivate());
+Assert::false($p->isProtected());
+Assert::true($p->isPublic());
 
 $m = $class->addMethod('getHandle')
 	->addComment('Returns file handle.')
@@ -100,6 +103,9 @@ Assert::true($m->getReturnReference());
 Assert::false($m->isReturnNullable());
 Assert::null($m->getReturnType());
 Assert::same('protected', $m->getVisibility());
+Assert::false($m->isPrivate());
+Assert::true($m->isProtected());
+Assert::false($m->isPublic());
 
 $method = $class->addMethod('show')
 	->setAbstract(true);

tests/PhpGenerator/Printer.phpt

@@ -23,7 +23,7 @@ $class = (new ClassType('Example'))
 	->addComment("Description of class.\nThis is example\n");
 
 $class->addConstant('FORCE_ARRAY', new PhpLiteral('Nette\Utils\Json::FORCE_ARRAY'))
-	->setVisibility('private')
+	->setPrivate()
 	->addComment('Commented');
 
 $class->addConstant('MULTILINE_LONG', ['aaaaaaaa' => 1, 'bbbbbbbb' => 2, 'cccccccc' => 3, 'dddddddd' => 4, 'eeeeeeee' => 5, 'ffffffff' => 6]);