<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/

namespace Magento\Setup\Model;

use Symfony\Component\Console\Application;
use Magento\Framework\Console\CommandListInterface;
use Magento\Framework\ObjectManagerInterface;
use Laminas\ServiceManager\ServiceLocatorInterface;
use Magento\Setup\Mvc\Bootstrap\InitParamListener;

/**
* Object manager provider
*
* Links Laminas Framework's service locator and Magento object manager.
* Guaranties single object manager per application run.
* Hides complexity of creating Magento object manager
*/
class ObjectManagerProvider
{
/**
* @var ServiceLocatorInterface
*/
private $serviceLocator;

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

/**
* @var Bootstrap
*/
private $bootstrap;

/**
* @param ServiceLocatorInterface $serviceLocator
* @param Bootstrap $bootstrap
*/
public function __construct(
ServiceLocatorInterface $serviceLocator,
Bootstrap $bootstrap
) {
$this->serviceLocator = $serviceLocator;
$this->bootstrap = $bootstrap;
}

/**
* Retrieve object manager.
*
* @return ObjectManagerInterface
* @throws \Magento\Setup\Exception
*/
public function get()
{
if (null === $this->objectManager) {
$initParams = $this->serviceLocator->get(InitParamListener::BOOTSTRAP_PARAM);
$factory = $this->getObjectManagerFactory($initParams);
$this->objectManager = $factory->create($initParams);
if (PHP_SAPI == 'cli') {
$this->createCliCommands();
}
}
return $this->objectManager;
}

/**
* Creates cli commands and initialize them with application instance
*
* @return void
*/
private function createCliCommands()
{
/** @var CommandListInterface $commandList */
$commandList = $this->objectManager->create(CommandListInterface::class);
$application = $this->serviceLocator->get(Application::class);
foreach ($commandList->getCommands() as $command) {
$application->add($command);
}
}

/**
* Causes object manager to be reinitialized the next time it is retrieved.
*
* @return void
*/
public function reset()
{
$this->objectManager = null;
}

/**
* Sets object manager
*
* @param ObjectManagerInterface $objectManager
* @return void
*/
public function setObjectManager(ObjectManagerInterface $objectManager)
{
$this->objectManager = $objectManager;
}

/**
* Returns ObjectManagerFactory
*
* @param array $initParams
* @return \Magento\Framework\App\ObjectManagerFactory
*/
public function getObjectManagerFactory($initParams = [])
{
return $this->bootstrap->createObjectManagerFactory(
BP,
$initParams
);
}
}