fixed #180 - worklog visibility error.

src/Issue/Comment.php

@@ -4,6 +4,8 @@
 
 class Comment implements \JsonSerializable
 {
+    use VisibilityTrait;
+
     /** @var string */
     public $self;
 
@@ -35,31 +37,6 @@ public function setBody($body)
         return $this;
     }
 
-    /**
-     * @param Visibility $type
-     * @param null       $value
-     *
-     * @return $this
-     */
-    public function setVisibility($type, $value = null)
-    {
-        if (is_null($this->visibility)) {
-            $this->visibility = new Visibility();
-        }
-
-        if (is_array($type)) {
-            $this->visibility->setType($type['type']);
-            $this->visibility->setValue($type['value']);
-        } elseif ($type instanceof Visibility) {
-            $this->visibility = $type;
-        } else {
-            $this->visibility->setType($type);
-            $this->visibility->setValue($value);
-        }
-
-        return $this;
-    }
-
     public function jsonSerialize()
     {
         return array_filter(get_object_vars($this));

src/Issue/VisibilityTrait.php

@@ -0,0 +1,32 @@
+<?php
+
+namespace JiraRestApi\Issue;
+
+trait VisibilityTrait {
+
+    /**
+     *
+     * @param Visibility $type
+     * @param null       $value
+     *
+     * @return $this
+     */
+    public function setVisibility($type, $value = null)
+    {
+        if (is_null($this->visibility)) {
+            $this->visibility = new Visibility();
+        }
+
+        if (is_array($type)) {
+            $this->visibility->setType($type['type']);
+            $this->visibility->setValue($type['value']);
+        } elseif ($type instanceof Visibility) {
+            $this->visibility = $type;
+        } else {
+            $this->visibility->setType($type);
+            $this->visibility->setValue($value);
+        }
+
+        return $this;
+    }
+}

src/Issue/Worklog.php

@@ -11,6 +11,7 @@
 class Worklog
 {
     use ClassSerialize;
+    use VisibilityTrait;
 
     /**
      * @var int id of worklog
@@ -58,7 +59,7 @@ class Worklog
     public $timeSpentSeconds;
 
     /**
-     * @var array
+     * @var \JiraRestApi\Issue\Visibility
      */
     public $visibility;
 
@@ -153,22 +154,4 @@ public function setTimeSpentSeconds($timeSpentSeconds)
 
         return $this;
     }
-
-    /**
-     * Function to set visibility of worklog.
-     *
-     * @param string $type  value can be group or role
-     * @param string $value
-     *
-     * @return Worklog
-     */
-    public function setVisibility($type, $value)
-    {
-        $this->visibility = [
-            'type'  => $type,
-            'value' => $value,
-        ];
-
-        return $this;
-    }
 }