[WIP]Refactor to Traits and add some little more Parameters

src/OneSignalMessage.php

@@ -3,32 +3,20 @@
 namespace NotificationChannels\OneSignal;
 
 use Illuminate\Support\Arr;
+use NotificationChannels\OneSignal\Traits\Categories\AppearanceHelpers;
+use NotificationChannels\OneSignal\Traits\Categories\AttachmentHelpers;
+use NotificationChannels\OneSignal\Traits\Categories\ButtonHelpers;
+use NotificationChannels\OneSignal\Traits\Categories\DeliveryHelpers;
+use NotificationChannels\OneSignal\Traits\Categories\GroupingHelpers;
+
 
 class OneSignalMessage
 {
-    /** @var string */
-    protected $body;
-
-    /** @var string */
-    protected $subject;
-
-    /** @var string */
-    protected $url;
-
-    /** @var string */
-    protected $icon;
-
-    /** @var array */
-    protected $data = [];
 
-    /** @var array */
-    protected $buttons = [];
+    use AppearanceHelpers, AttachmentHelpers, ButtonHelpers, DeliveryHelpers, GroupingHelpers;
 
     /** @var array */
-    protected $webButtons = [];
-
-    /** @var array */
-    protected $extraParameters = [];
+    protected $payload = [];
 
     /**
      * @param string $body
@@ -45,212 +33,131 @@ public static function create($body = '')
      */
     public function __construct($body = '')
     {
-        $this->body = $body;
+        $this->setBody($body);
     }
 
     /**
      * Set the message body.
      *
-     * @param string $value
+     * @param mixed $value
      *
+     * @deprecated use setBody instead
+     *             
      * @return $this
      */
     public function body($value)
     {
-        $this->body = $value;
-
-        return $this;
+        return $this->setBody($value);
     }
 
     /**
-     * Set the message icon.
+     * Set the message body.
      *
-     * @param string $value
+     * @param mixed $value
      *
      * @return $this
      */
-    public function icon($value)
+    public function setBody($value)
     {
-        $this->icon = $value;
-
-        return $this;
+        return $this->setParameter('contents', $this->parseValueToArray($value));
     }
 
     /**
      * Set the message subject.
      *
-     * @param string $value
+     * @param mixed $value
+     *
+     * @deprecated Use setSubject instead
      *
      * @return $this
      */
     public function subject($value)
     {
-        $this->subject = $value;
-
-        return $this;
+        return $this->setParameter('headings', $this->parseValueToArray($value));
     }
 
     /**
-     * Set the message url.
+     * Set the message subject.
      *
-     * @param string $value
+     * @param mixed $value
      *
      * @return $this
      */
-    public function url($value)
+    public function setSubject($value)
     {
-        $this->url = $value;
-
-        return $this;
+        return $this->setSubject($value);
     }
 
     /**
-     * Set the iOS badge increment count.
+     * @param mixed $value
      *
-     * @param int $count
-     *
-     * @return $this
+     * @return array
      */
-    public function incrementIosBadgeCount($count = 1)
+    protected function parseValueToArray($value)
     {
-        return $this->setParameter('ios_badgeType', 'Increase')
-                    ->setParameter('ios_badgeCount', $count);
+        return (is_array($value)) ? $value : ['en' => $value];
     }
 
     /**
-     * Set the iOS badge decrement count.
-     *
-     * @param int $count
+     * Set the message url.
      *
-     * @return $this
-     */
-    public function decrementIosBadgeCount($count = 1)
-    {
-        return $this->setParameter('ios_badgeType', 'Increase')
-                    ->setParameter('ios_badgeCount', -1 * $count);
-    }
-
-    /**
-     * Set the iOS badge count.
+     * @param string $value
      *
-     * @param int $count
+     * @deprecated use setUrl Instead
      *
      * @return $this
      */
-    public function setIosBadgeCount($count)
+    public function url($value)
     {
-        return $this->setParameter('ios_badgeType', 'SetTo')
-                    ->setParameter('ios_badgeCount', $count);
+        return $this->setUrl($value);
     }
 
     /**
      * Set additional data.
      *
      * @param string $key
-     * @param string $value
+     * @param mixed  $value
      *
      * @return $this
      */
-    public function setData($key, $value)
+    public function setData(string $key, $value)
     {
-        $this->data[$key] = $value;
-
-        return $this;
+        return $this->setParameter("data.{$key}", $value);
     }
 
     /**
-     * Set additional parameters.
+     * Set parameters.
      *
      * @param string $key
-     * @param string $value
-     *
-     * @return $this
-     */
-    public function setParameter($key, $value)
-    {
-        $this->extraParameters[$key] = $value;
-
-        return $this;
-    }
-
-    /**
-     * Add a web button to the message.
-     *
-     * @param OneSignalWebButton $button
+     * @param mixed  $value
      *
      * @return $this
      */
-    public function webButton(OneSignalWebButton $button)
+    public function setParameter(string $key, $value)
     {
-        $this->webButtons[] = $button->toArray();
+        Arr::set($this->payload, $key, $value);
 
         return $this;
     }
 
     /**
-     * Add a native button to the message.
+     * Get parameters.
      *
-     * @param OneSignalButton $button
-     *
-     * @return $this
-     */
-    public function button(OneSignalButton $button)
-    {
-        $this->buttons[] = $button->toArray();
-
-        return $this;
-    }
-
-    /**
-     * Set an image to all possible attachment variables.
-     * @param string $imageUrl
+     * @param string $key
+     * @param mixed  $default
      *
-     * @return $this
+     * @return mixed
      */
-    public function setImageAttachments($imageUrl)
+    public function getParameter(string $key, $default = null)
     {
-        $this->extraParameters['ios_attachments']['id1'] = $imageUrl;
-        $this->extraParameters['big_picture'] = $imageUrl;
-        $this->extraParameters['adm_big_picture'] = $imageUrl;
-        $this->extraParameters['chrome_big_picture'] = $imageUrl;
-
-        return $this;
+        return Arr::get($this->payload, $key, $default);
     }
 
     /**
      * @return array
      */
     public function toArray()
     {
-        $message = [
-            'contents' => ['en' => $this->body],
-            'headings' => $this->subjectToArray(),
-            'url' => $this->url,
-            'buttons' => $this->buttons,
-            'web_buttons' => $this->webButtons,
-            'chrome_web_icon' => $this->icon,
-            'chrome_icon' => $this->icon,
-            'adm_small_icon' => $this->icon,
-            'small_icon' => $this->icon,
-        ];
-
-        foreach ($this->extraParameters as $key => $value) {
-            Arr::set($message, $key, $value);
-        }
-
-        foreach ($this->data as $data => $value) {
-            Arr::set($message, 'data.'.$data, $value);
-        }
-
-        return $message;
-    }
-
-    protected function subjectToArray()
-    {
-        if ($this->subject === null) {
-            return [];
-        }
-
-        return ['en' => $this->subject];
+        return $this->payload;
     }
 }