added addComment(), setComment() and getComment()

nette · Jun 17, 2016 · bbc8189 · bbc8189
1 parent 6245050
commit bbc8189
Show file tree

Hide file tree

Showing 4 changed files with 123 additions and 3 deletions.
diff --git a/src/PhpGenerator/ClassType.php b/src/PhpGenerator/ClassType.php
@@ -333,6 +333,36 @@ public function addTrait($trait)
  }
 
 
+ /**
+ * @param string|NULL
+ * @return self
+ */
+ public function setComment($val)
+ {
+ $this->documents = $val ? array((string) $val) : array();
+ return $this;
+ }
+
+
+ /**
+ * @return string|NULL
+ */
+ public function getComment()
+ {
+ return implode($this->documents) ?: NULL;
+ }
+
+
+ /**
+ * @param string
+ * @return self
+ */
+ public function addComment($val)
+ {
+ return $this->addDocument($val);
+ }
+
+
  /**
  * @param string[]
  * @return self

diff --git a/src/PhpGenerator/Method.php b/src/PhpGenerator/Method.php
@@ -374,6 +374,36 @@ public function isVariadic()
  }
 
 
+ /**
+ * @param string|NULL
+ * @return self
+ */
+ public function setComment($val)
+ {
+ $this->documents = $val ? array((string) $val) : array();
+ return $this;
+ }
+
+
+ /**
+ * @return string|NULL
+ */
+ public function getComment()
+ {
+ return implode($this->documents) ?: NULL;
+ }
+
+
+ /**
+ * @param string
+ * @return self
+ */
+ public function addComment($val)
+ {
+ return $this->addDocument($val);
+ }
+
+
  /**
  * @param string[]
  * @return self

diff --git a/src/PhpGenerator/PhpFile.php b/src/PhpGenerator/PhpFile.php
@@ -29,11 +29,32 @@ class PhpFile extends Object
 
 
  /**
- * @return string[]
+ * @param string|NULL
+ * @return self
  */
- public function getDocuments()
+ public function setComment($val)
  {
- return $this->documents;
+ $this->documents = $val ? array((string) $val) : array();
+ return $this;
+ }
+
+
+ /**
+ * @return string|NULL
+ */
+ public function getComment()
+ {
+ return implode($this->documents) ?: NULL;
+ }
+
+
+ /**
+ * @param string
+ * @return self
+ */
+ public function addComment($val)
+ {
+ return $this->addDocument($val);
  }
 
 
@@ -48,6 +69,15 @@ public function setDocuments(array $documents)
  }
 
 
+ /**
+ * @return string[]
+ */
+ public function getDocuments()
+ {
+ return $this->documents;
+ }
+
+
  /**
  * @param string
  * @return self

diff --git a/src/PhpGenerator/Property.php b/src/PhpGenerator/Property.php
@@ -137,6 +137,36 @@ public function getVisibility()
  }
 
 
+ /**
+ * @param string|NULL
+ * @return self
+ */
+ public function setComment($val)
+ {
+ $this->documents = $val ? array((string) $val) : array();
+ return $this;
+ }
+
+
+ /**
+ * @return string|NULL
+ */
+ public function getComment()
+ {
+ return implode($this->documents) ?: NULL;
+ }
+
+
+ /**
+ * @param string
+ * @return self
+ */
+ public function addComment($val)
+ {
+ return $this->addDocument($val);
+ }
+
+
  /**
  * @param string[]
  * @return self