datacounter.php

<?php
namespace Bitrix\Sender\Connector;
 
use Bitrix\Sender\Recipient;
 
class DataCounter
{
    public $data;
 
    public function __construct(array $data)
    {
        $this->data = array();
        $types = Recipient\Type::getList();
        foreach ($data as $typeId => $count)
        {
            if (is_numeric($typeId))
            {
                $typeId = (int) $typeId;
            }
            $typeId = $typeId ?: Recipient\Type::EMAIL;
            $typeId = is_numeric($typeId) ? $typeId : Recipient\Type::getId($typeId);
            if ($typeId && !in_array($typeId, $types))
            {
                continue;
            }
 
            $this->data[$typeId] = (int) $count;
        }
    }
 
    public function getSummary()
    {
        $summary = 0;
        foreach ($this->data as $typeId => $count)
        {
            $summary += $count;
        }
 
        return $summary;
    }
 
    public function getCount($typeId)
    {
        return isset($this->data[$typeId]) ? $this->data[$typeId] : 0;
    }
 
    public function leave($leaveTypeId = null)
    {
        if (!$leaveTypeId)
        {
            return $this;
        }
 
        foreach ($this->data as $typeId => $count)
        {
            if ($leaveTypeId == $typeId)
            {
                continue;
            }
 
            unset($this->data[$typeId]);
        }
 
        return $this;
    }
 
    public function getList()
    {
        return $this->data;
    }
 
    public function getArray()
    {
        $counters = array();
        foreach ($this->data as $typeId => $count)
        {
            $counters[] = array(
                'typeId' => $typeId,
                'typeCode' => Recipient\Type::getCode($typeId),
                'typeName' => Recipient\Type::getName($typeId),
                'count' => $count
            );
        }
 
        $result = self::getDefaultArray();
        $result['summary'] = $this->getSummary();
        $result['counters'] = $counters;
 
        return $result;
    }
 
    public function getArrayCounters()
    {
        $counters = array();
        foreach ($this->data as $typeId => $count)
        {
            $counters[$typeId] = $count;
        }
 
        return $counters;
    }
 
    public static function getDefaultArray()
    {
        return array(
            'summary' => 0,
            'counters' => array()
        );
    }
}