set icon badge number

Author: Azri Ismail

src/FirebaseMessage.php

@@ -60,13 +60,40 @@ public function title($value)
         return $this;
     }
 
+    /**
+     * Set notification sound
+     *
+     * @param string $value
+     *
+     * @return $this
+     */
     public function sound($value = 'default')
     {
         $this->notification->setSound($value);
 
         return $this;
     }
 
+    /**
+     * Set badge number
+     *
+     * @param int $integer
+     *
+     * @return $this
+     */
+    public function badge($integer)
+    {
+        $this->notification->setBadge($integer);
+
+        return $this;
+    }
+
+    /**
+     * Set additional data
+     *
+     * @param array $values
+     * @return $this
+     */
     public function data(array $values = [])
     {
         $this->data = $values;