ContextCustomer.php

<?php
 
namespace Bitrix\Im\V2\Common;
 
use Bitrix\Im\User;
use Bitrix\Im\V2\Service\Context;
 
trait ContextCustomer
{
    protected ?Context $context = null;
 
    public function withContext(?Context $context): self
    {
        $copy = clone $this;
 
        return $copy->setContext($context);
    }
 
    public function withContextUser($user): self
    {
        if ($this->context)
        {
            $context = clone $this->context;
        }
        else
        {
            $context = new Context;
        }
 
        if ($user instanceof User)
        {
            $context->setUser($user);
        }
        elseif (is_numeric($user))
        {
            $context->setUserId((int)$user);
        }
 
        return $this->withContext($context);
    }
 
    public function setContext(?Context $context): self
    {
        $this->context = $context;
 
        return $this;
    }
 
    public function getContext(): Context
    {
        if ($this->context)
        {
            return $this->context;
        }
 
        return \Bitrix\Im\V2\Service\Locator::getContext();
    }
}