magento2/app/code/Magento/GraphQl/Model/Query/ContextFactory.php

<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
declare(strict_types=1);

namespace Magento\GraphQl\Model\Query;

use Magento\Authorization\Model\UserContextInterface;
use Magento\Framework\Api\ExtensionAttributesFactory;
use Magento\Framework\App\ObjectManager;
use Magento\Framework\Exception\LocalizedException;
use Magento\Framework\ObjectManager\ResetAfterRequestInterface;
use Magento\Framework\ObjectManagerInterface;

/**
 * @inheritdoc
 */
class ContextFactory implements ContextFactoryInterface, ResetAfterRequestInterface
{
    /**
     * @var ExtensionAttributesFactory
     */
    private $extensionAttributesFactory;

    /**
     * @var ObjectManagerInterface
     */
    private $objectManager;

    /**
     * @var ContextParametersProcessorInterface[]
     */
    private $contextParametersProcessors;

    /**
     * @var ContextInterface
     */
    private $context;

    /**
     * @param ExtensionAttributesFactory $extensionAttributesFactory
     * @param ObjectManagerInterface $objectManager
     * @param ContextParametersProcessorInterface[] $contextParametersProcessors
     */
    public function __construct(
        ExtensionAttributesFactory $extensionAttributesFactory,
        ObjectManagerInterface $objectManager,
        array $contextParametersProcessors = []
    ) {
        $this->extensionAttributesFactory = $extensionAttributesFactory;
        $this->objectManager = $objectManager;
        $this->contextParametersProcessors = $contextParametersProcessors;
    }

    /**
     * @inheritdoc
     */
    public function create(?UserContextInterface $userContext = null): ContextInterface
    {
        $contextParameters = $this->objectManager->create(ContextParametersInterface::class);
        foreach ($this->contextParametersProcessors as $contextParametersProcessor) {
            if (!$contextParametersProcessor instanceof ContextParametersProcessorInterface) {
                throw new LocalizedException(
                    __('ContextParametersProcessors must implement %1', ContextParametersProcessorInterface::class)
                );
            }
            if ($contextParametersProcessor instanceof UserContextParametersProcessorInterface) {
                $contextParametersProcessor->setUserContext(
                    $userContext ?? $this->objectManager->create(UserContextInterface::class)
                );
            }
            $contextParameters = $contextParametersProcessor->execute($contextParameters);
        }

        $extensionAttributes = $this->extensionAttributesFactory->create(
            ContextInterface::class,
            [
                'data' => $contextParameters->getExtensionAttributesData(),
            ]
        );

        $context = $this->objectManager->create(
            ContextInterface::class,
            [
                'userType' => $contextParameters->getUserType(),
                'userId' => $contextParameters->getUserId(),
                'extensionAttributes' => $extensionAttributes,
            ]
        );
        $this->context = $context;
        return $context;
    }

    /**
     * @inheritdoc
     */
    public function get(): ContextInterface
    {
        if (!$this->context) {
            $this->create();
        }
        return $this->context;
    }

    /**
     * @inheritDoc
     */
    public function _resetState(): void
    {
        $this->context = null;
    }
}