vendor/redjanym/fcm-bundle/Entity/DeviceNotification.php line 9

Open in your IDE?
  1. <?php
  3. namespace RedjanYm\FCMBundle\Entity;
  5. use Doctrine\Common\Collections\ArrayCollection;
  6. use sngrl\PhpFirebaseCloudMessaging\Notification;
  7. use RedjanYm\FCMBundle\Model\DeviceNotificationInterface;
  9. class DeviceNotification extends Notification implements DeviceNotificationInterface
  10. {
  12.     use NotificationData;
  14.     /**
  15.      * @var ArrayCollection $deviceToken
  16.      */
  17.     private $deviceTokens;
  19.     public function __construct($title = '', $body = '')
  20.     {
  21.         parent::__construct($title, $body);
  23.         $this->deviceTokens = new ArrayCollection();
  24.     }
  26.     /**
  27.      * {@inheritdoc}
  28.      */
  29.     public function setDeviceToken($token)
  30.     {
  31.         $this->deviceTokens[] = $token;
  33.         return $this;
  34.     }
  35.     
  36.     /**
  37.      * {@inheritdoc}
  38.      */
  39.     public function addDeviceToken($token)
  40.     {
  41.         if(is_string($token) === false){
  42.             throw new \InvalidArgumentException('Token must be a string!');
  43.         }
  45.         $this->deviceTokens->add($token);
  47.         return $this;
  48.     }
  50.     /**
  51.      * {@inheritdoc}
  52.      */
  53.     public function setDeviceTokens($tokens)
  54.     {
  55.         if(is_array($tokens) === false){
  56.             throw new \InvalidArgumentException('Tokens must be an array of strings!');
  57.         }
  59.         $this->deviceTokens = new ArrayCollection($tokens);
  61.         return $this;
  62.     }
  64.     /**
  65.      * {@inheritdoc}
  66.      */
  67.     public function getDeviceToken()
  68.     {
  69.         if ($this->deviceTokens->isEmpty() === false) {
  70.             return $this->deviceTokens->first();
  71.         } else {
  72.             throw new \InvalidArgumentException('The Mobile Notification must have a Device Token!');
  73.         }
  74.     }
  76.     /**
  77.      * {@inheritdoc}
  78.      */
  79.     public function getDeviceTokens()
  80.     {
  81.         if ($this->deviceTokens->isEmpty() === false) {
  82.             return $this->deviceTokens->toArray();
  83.         } else {
  84.             throw new \InvalidArgumentException('The Mobile Notification must have a Device Token!');
  85.         }
  86.     }
  87. }