result.php

<?php
namespace Bitrix\Sender\Connector;
 
use Bitrix\Main\DB\ArrayResult;
use Bitrix\Main\DB\Result as DbResult;
use Bitrix\Sender\Recipient;
 
class Result
{
    public $resource;
 
    public $resourceCDBResult;
 
    protected $fields;
 
    protected $additionalFields;
 
    protected $fieldsDisallowed = ['NAME', 'USER_ID'];
 
    protected $dataTypeId = Recipient\Type::EMAIL;
 
    public function __construct($resource)
    {
        if(is_array($resource))
        {
            $isSingleArray = false;
            $arrayKeyList = array_keys($resource);
            foreach($arrayKeyList as $key)
            {
                if(is_string($key))
                {
                    $isSingleArray = true;
                    break;
                }
            }
 
            if($isSingleArray)
            {
                $resource = array($resource);
            }
 
            $this->resource = new ArrayResult($resource);
        }
        elseif($resource instanceof DbResult)
        {
            $this->resource = $resource;
        }
        elseif($resource instanceof \CAllDBResult)
        {
            $this->resourceCDBResult = $resource;
        }
 
        $this->fieldsDisallowed = array_merge(
            $this->fieldsDisallowed,
            Recipient\Type::getCodes()
        );
    }
 
    public function getDataTypeId()
    {
        return $this->dataTypeId;
    }
 
    public function setDataTypeId($dataTypeId)
    {
        $this->dataTypeId = $dataTypeId;
    }
 
    public function setFilterFields(array $fields)
    {
        $this->fields = $fields;
    }
 
    public function getFilterFields()
    {
        return $this->fields;
    }
 
    public function setAdditionalFields(array $additionalFields)
    {
        $this->additionalFields = $additionalFields;
    }
 
    public function fetchPlain()
    {
        $result = null;
        if($this->resource)
        {
            $result = $this->resource->fetch();
        }
        elseif($this->resourceCDBResult)
        {
            $result = $this->resourceCDBResult->Fetch();
        }
 
        return (is_array($result) && count($result) > 0) ? $result : null;
    }
 
    public function fetch()
    {
        $result = $this->fetchPlain();
        if($result)
        {
            $result = $this->fetchModifierFields($result);
        }
 
        return ($result && count($result) > 0) ? $result : null;
    }
 
    protected function fetchModifierFields(array $result)
    {
        $fieldsList = array();
        foreach($result as $key => $value)
        {
            if(is_object($value))
            {
                $value = (string) $value;
                $result[$key] = $value;
            }
 
            if(in_array($key, $this->fieldsDisallowed))
            {
                continue;
            }
 
            if($this->fields && in_array($key, $this->fields))
            {
                $fieldsList[$key] = $value;
            }
 
            unset($result[$key]);
        }
 
        if($this->additionalFields)
        {
            $fieldsList = $fieldsList + $this->additionalFields;
        }
 
        if(count($fieldsList) > 0)
        {
            $result['FIELDS'] = $fieldsList;
        }
 
        return $result;
    }
 
    public function getSelectedRowsCount()
    {
        if($this->resource)
        {
            return $this->resource->getSelectedRowsCount();
        }
        elseif($this->resourceCDBResult)
        {
            if(!($this->resourceCDBResult instanceof \CDBResultMysql))
            {
                $this->resourceCDBResult->NavStart(0);
            }
 
            return $this->resourceCDBResult->SelectedRowsCount();
        }
 
        return 0;
    }
 
    public function getQuerySql()
    {
        if (!$this->resource)
        {
            return null;
        }
 
        if (!$this->resource->getTrackerQuery())
        {
            return null;
        }
 
        return $this->resource->getTrackerQuery()->getSql();
    }
}