nstdio\yii2notymo\PushNotification

Complexity

Total Complexity

16

Size/Duplication

Total Lines	89
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	7

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
C	__construct()	0	30	7
A	autoDetectProvider()	0	8	3
A	getDbConnectionList()	0	7	1
B	send()	0	28	5

<?php
namespace nstdio\yii2notymo;

use nstdio\notymo\MessageInterface;
use nstdio\notymo\PushNotification as PushImpl;
use nstdio\yii2notymo\provider\DataProvider;
use Yii;
use yii\base\InvalidConfigException;
use yii\base\Object;
use yii\helpers\ArrayHelper;

/**
 * Class PushNotification
 *
 * @author Edgar Asatryan <nstdio@gmail.com>
 */
class PushNotification extends Object
{
    /**
     * @var DataProvider
     */
    public $dataProvider;

    /**
     * @var PushImpl
     */
    protected $push;

    public function __construct($config = [])
    {
        if (!isset($config['push'])) {
            throw new InvalidConfigException("push configuration required.");
        }
        if (!isset($config['dataProvider'])) {
            throw new InvalidConfigException("dataProvider configuration required.");
        }
        if (!isset($config['dataProvider']['class'])) {
            $config['dataProvider']['class'] = $this->autoDetectProvider();
        }
        if (isset($config['push']['skipApns'])) {
            $config['dataProvider']['apns'] = null;
        }
        if (isset($config['push']['skipGcm'])) {
            $config['dataProvider']['gcm'] = null;
        }

        $this->push = new PushImpl($config['push']);

        $this->dataProvider = Yii::createObject($config['dataProvider']);

        if (!($this->dataProvider instanceof DataProvider)) {
            throw new InvalidConfigException("dataProvider must extend " . DataProvider::className() . " class.");
        }

        unset($config['push'], $config['dataProvider']);

        parent::__construct($config);
    }

    protected function autoDetectProvider()
    {
        foreach ($this->getDbConnectionList() as $service => $providerName) {
            if (Yii::$app->get($service, false) !== null) {
                return $providerName;
            }
        }
    }

    public function getDbConnectionList()
    {
        return [
            'db'      => 'nstdio\yii2notymo\provider\SQLDataProvider',
            'mongodb' => 'nstdio\yii2notymo\provider\MongoDataProvider',
        ];
    }

    public function send(MessageInterface $message, $userId = null)
    {
        $oldTokens = $message->getToken();
        $tokens = $this->dataProvider->getTokens($userId);

        if ($this->dataProvider->apns !== null) {
            $apnsTokens = ArrayHelper::getColumn($tokens, $this->dataProvider->apns);
            if (!empty($apnsTokens)) {
                $message->setType(MessageInterface::TYPE_IOS);
                $message->addToken($apnsTokens);
                $this->push->enqueue($message);
            }
        }

        if ($this->dataProvider->gcm !== null) {
            $gcmTokens = ArrayHelper::getColumn($tokens, $this->dataProvider->gcm);

            if (!empty($gcmTokens)) {
                $gcmMessage = clone $message;
                $gcmMessage->setType(MessageInterface::TYPE_ANDROID);
                $gcmMessage->setToken(ArrayHelper::merge($oldTokens, $gcmTokens));

                $this->push->enqueue($gcmMessage);
            }
        }

        $this->push->send();
    }
}