UserCollection.php

<?php
 
namespace Bitrix\Im\V2\Entity\User;
 
use Bitrix\Im\Model\StatusTable;
use Bitrix\Im\V2\Entity\EntityCollection;
use Bitrix\Main\ORM\Fields\Relations\Reference;
use Bitrix\Main\ORM\Query\Join;
use Bitrix\Main\UserTable;
 
class UserCollection extends EntityCollection
{
    public function __construct(array $usersIds = [])
    {
        parent::__construct();
 
        foreach ($usersIds as $userId)
        {
            $this[] = User::getInstance($userId);
        }
    }
 
    public function fillOnlineData(bool $withStatus = false): void
    {
        $idsUsersWithoutOnlineData = [];
 
        foreach ($this as $user)
        {
            if (!$user->isOnlineDataFilled($withStatus))
            {
                $idsUsersWithoutOnlineData[] = $user->getId();
            }
        }
 
        $idsUsersWithoutOnlineData = array_unique($idsUsersWithoutOnlineData);
 
        if (empty($idsUsersWithoutOnlineData))
        {
            return;
        }
 
        $select = $withStatus ? User::ONLINE_DATA_SELECTED_FIELDS : User::ONLINE_DATA_SELECTED_FIELDS_WITHOUT_STATUS;
        $query = UserTable::query()
            ->setSelect($select)
            ->whereIn('ID', $idsUsersWithoutOnlineData)
        ;
        if ($withStatus)
        {
            $query->registerRuntimeField(
                new Reference(
                    'STATUS',
                    StatusTable::class,
                    Join::on('this.ID', 'ref.USER_ID'),
                    ['join_type' => Join::TYPE_LEFT]
                )
            );
        }
        $statusesData = $query->fetchAll() ?: [];
 
        foreach ($statusesData as $statusData)
        {
            $this->getById((int)$statusData['USER_ID'])->setOnlineData($statusData, $withStatus);
        }
    }
 
    public function filterExtranet(): self
    {
        $filteredUsers = new static();
 
        foreach ($this as $user)
        {
            if (!$user->isExtranet())
            {
                $filteredUsers[] = $user;
            }
        }
 
        return $filteredUsers;
    }
 
    public static function filterOnlineUserId(array $userIds): array
    {
        if (empty($userIds))
        {
            return [];
        }
 
        $result = UserTable::query()
            ->setSelect(['ID'])
            ->whereIn('ID', $userIds)
            ->where('IS_ONLINE', true)
            ->fetchAll()
        ;
        $onlineUsers = [];
        foreach ($result as $row)
        {
            $onlineUsers[] = (int)$row['ID'];
        }
 
        return $onlineUsers;
    }
 
    public function toRestFormat(array $option = []): array
    {
        $this->fillOnlineData();
 
        return parent::toRestFormat($option);
    }
 
    public static function getRestEntityName(): string
    {
        return 'users';
    }
 
    public function offsetSet($offset, $value): void
    {
        if (!$value->isExist())
        {
            return;
        }
 
        parent::offsetSet($offset, $value);
    }
}