blacklist.php

<?php
 
namespace Bitrix\Mail;
 
use Bitrix\Mail\Internals\Entity\BlacklistEmail;
use Bitrix\Main\Application;
use Bitrix\Main\Entity;
use Bitrix\Main\Localization;
use Bitrix\Main\ORM\Query\Query;
 
Localization\Loc::loadMessages(__FILE__);
 
class BlacklistTable extends Entity\DataManager
{
 
    public static function getFilePath()
    {
        return __FILE__;
    }
 
    public static function getTableName()
    {
        return 'b_mail_blacklist';
    }
 
    public static function getObjectClass()
    {
        return BlacklistEmail::class;
    }
 
    public static function getMap()
    {
        return [
            'ID' => [
                'data_type' => 'integer',
                'primary' => true,
                'autocomplete' => true,
            ],
            'SITE_ID' => [
                'data_type' => 'string',
                'required' => true,
            ],
            'MAILBOX_ID' => [
                'data_type' => 'integer',
            ],
            'USER_ID' => [
                'data_type' => 'integer',
            ],
            'ITEM_TYPE' => [
                'data_type' => 'integer',
                'required' => true,
            ],
            'ITEM_VALUE' => [
                'data_type' => 'string',
                'required' => true,
                'fetch_data_modification' => function ()
                {
                    return [
                        function ($value, $query, $data)
                        {
                            if (Blacklist\ItemType::DOMAIN == $data['ITEM_TYPE'])
                            {
                                $value = sprintf('@%s', $value);
                            }
 
                            return $value;
                        },
                    ];
                },
            ],
        ];
    }
 
    public static function replace($siteId, $mailboxId, array $list)
    {
        global $DB;
 
        if ($mailboxId > 0)
        {
            $DB->query(sprintf("DELETE FROM b_mail_blacklist WHERE MAILBOX_ID = %u", $mailboxId));
        }
        else
        {
            $DB->query(sprintf("DELETE FROM b_mail_blacklist WHERE SITE_ID = '%s' AND MAILBOX_ID = 0", $DB->forSql($siteId)));
        }
 
        if (!empty($list))
        {
            foreach ($list as $item)
            {
                static::add([
                    'SITE_ID' => $siteId,
                    'MAILBOX_ID' => $mailboxId,
                    'ITEM_TYPE' => Blacklist\ItemType::resolveByValue($item),
                    'ITEM_VALUE' => $item,
                ]);
            }
        }
    }
 
    public static function addMailsBatch(array $list, $userId = null)
    {
        if (empty($list))
        {
            return 0;
        }
        if (is_null($userId))
        {
            $userId = 0;
        }
        $sqlHelper = Application::getConnection()->getSqlHelper();
        $addList = [];
        foreach ($list as $index => $item)
        {
            $itemToAdd = [
                'SITE_ID' => SITE_ID,
                'MAILBOX_ID' => 0,
                'USER_ID' => $userId,
                'ITEM_TYPE' => Blacklist\ItemType::resolveByValue($item),
                'ITEM_VALUE' => $item,
            ];
            $addList[] = $itemToAdd;
        }
 
        if (count($addList) === 0)
        {
            return 0;
        }
        $keys = implode(', ', array_keys(current($addList)));
        $values = [];
        foreach ($addList as $item)
        {
            $values[] = implode(
                ", ",
                [
                    "'" . $sqlHelper->forSql($item['SITE_ID']) . "'",
                    (int)$item['MAILBOX_ID'],
                    (int)$item['USER_ID'],
                    $item['ITEM_TYPE'],
                    "'" . $sqlHelper->forSql($item['ITEM_VALUE']) . "'",
                ]
            );
        }
        $values = implode('), (', $values);
 
        $tableName = static::getTableName();
        $sql = $sqlHelper->getInsertIgnore($tableName, "($keys)", " VALUES($values)");
        Application::getConnection()->query($sql);
        return Application::getConnection()->getAffectedRowsCount();
    }
 
    public static function getUserAddressesListQuery($userId, $includeAddressesForAllUsers = true)
    {
        $filter = ['LOGIC' => 'OR'];
        $userMailboxes = \Bitrix\Mail\MailboxTable::getUserMailboxes();
        if (!empty($userMailboxes))
        {
            $mailboxesIds = array_column($userMailboxes, 'ID');
            if ($includeAddressesForAllUsers)
            {
                $mailboxesIds[] = 0;
            }
            $filter[] = [
                '@MAILBOX_ID' => $mailboxesIds,
                '=USER_ID' => 0,
            ];
        }
        $newStyleAddressesFilter = [];
        $userIds = [$userId];
        $newStyleAddressesFilter['=MAILBOX_ID'] = 0;
 
        if ($includeAddressesForAllUsers)
        {
            $userIds[] = 0;
        }
        $newStyleAddressesFilter['@USER_ID'] = $userIds;
        $mailsQuery = \Bitrix\Mail\BlacklistTable::query()
            ->addSelect('*');
        $filter[] = $newStyleAddressesFilter;
        return $mailsQuery->setFilter([$filter]);
    }
 
    public static function deleteList($filter)
    {
        $entity = static::getEntity();
        $connection = $entity->getConnection();
        return $connection->query(sprintf(
            'DELETE FROM %s WHERE %s',
            $connection->getSqlHelper()->quote($entity->getDbTableName()),
            Query::buildFilterSql(
                $entity,
                $filter
            )
        ));
    }
}