JobFactory.php 7.16 KB
<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
namespace Magento\Setup\Model\Cron;

use Magento\Backend\Console\Command\CacheDisableCommand;
use Magento\Backend\Console\Command\CacheEnableCommand;
use Magento\Framework\ObjectManagerInterface;
use Magento\Setup\Console\Command\ModuleDisableCommand;
use Magento\Setup\Console\Command\ModuleEnableCommand;
use Magento\Setup\Console\Command\UpgradeCommand;
use Zend\ServiceManager\ServiceLocatorInterface;
use Magento\Setup\Console\Command\MaintenanceDisableCommand;
use Magento\Setup\Console\Command\MaintenanceEnableCommand;

/**
 * Factory class to create jobs
 *
 * @SuppressWarnings(PHPMD.CouplingBetweenObjects)
 */
class JobFactory
{
    /**
     * Name of jobs
     */
    const JOB_UPGRADE = 'setup:upgrade';
    const JOB_DB_ROLLBACK = 'setup:rollback';
    const JOB_COMPONENT_UNINSTALL = 'setup:component:uninstall';
    const JOB_MODULE_ENABLE = 'setup:module:enable';
    const JOB_MODULE_DISABLE = 'setup:module:disable';
    const JOB_STATIC_REGENERATE = 'setup:static:regenerate';
    const JOB_ENABLE_CACHE = 'setup:cache:enable';
    const JOB_DISABLE_CACHE = 'setup:cache:disable';
    const JOB_MAINTENANCE_MODE_ENABLE = 'setup:maintenance:enable';
    const JOB_MAINTENANCE_MODE_DISABLE = 'setup:maintenance:disable';

    /**
     * @var ServiceLocatorInterface
     */
    private $serviceLocator;

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

    /**
     * Create job instance.
     *
     * @param string $name
     * @param array $params
     * @return AbstractJob
     * @throws \RuntimeException
     * @SuppressWarnings(PHPMD.ExcessiveMethodLength)
     * @SuppressWarnings(PHPMD.CyclomaticComplexity)
     */
    public function create($name, array $params = [])
    {
        $cronStatus = $this->serviceLocator->get(\Magento\Setup\Model\Cron\Status::class);
        $statusStream = fopen($cronStatus->getStatusFilePath(), 'a+');
        $logStream = fopen($cronStatus->getLogFilePath(), 'a+');
        $streamOutput = new MultipleStreamOutput([$statusStream, $logStream]);
        $objectManagerProvider = $this->serviceLocator->get(\Magento\Setup\Model\ObjectManagerProvider::class);
        /** @var ObjectManagerInterface $objectManager */
        $objectManager = $objectManagerProvider->get();
        switch ($name) {
            case self::JOB_UPGRADE:
                return new JobUpgrade(
                    $this->serviceLocator->get(UpgradeCommand::class),
                    $objectManagerProvider,
                    $streamOutput,
                    $this->serviceLocator->get(\Magento\Setup\Model\Cron\Queue::class),
                    $cronStatus,
                    $name,
                    $params
                );
                break;
            case self::JOB_DB_ROLLBACK:
                return new JobDbRollback(
                    $objectManager->get(\Magento\Framework\Setup\BackupRollbackFactory::class),
                    $streamOutput,
                    $cronStatus,
                    $objectManagerProvider,
                    $name,
                    $params
                );
                break;
            case self::JOB_STATIC_REGENERATE:
                return new JobStaticRegenerate(
                    $objectManagerProvider,
                    $streamOutput,
                    $cronStatus,
                    $name,
                    $params
                );
                break;
            case self::JOB_COMPONENT_UNINSTALL:
                $moduleUninstall = new Helper\ModuleUninstall(
                    $this->serviceLocator->get(\Magento\Setup\Model\ModuleUninstaller::class),
                    $this->serviceLocator->get(\Magento\Setup\Model\ModuleRegistryUninstaller::class),
                    $objectManager->get(\Magento\Framework\Module\PackageInfoFactory::class)
                );
                $themeUninstall = new Helper\ThemeUninstall(
                    $objectManager->get(\Magento\Theme\Model\Theme\ThemeUninstaller::class),
                    $objectManager->get(\Magento\Theme\Model\Theme\ThemePackageInfo::class)
                );
                return new JobComponentUninstall(
                    $objectManager->get(\Magento\Framework\Composer\ComposerInformation::class),
                    $moduleUninstall,
                    $themeUninstall,
                    $objectManagerProvider,
                    $streamOutput,
                    $this->serviceLocator->get(\Magento\Setup\Model\Cron\Queue::class),
                    $cronStatus,
                    $this->serviceLocator->get(\Magento\Setup\Model\Updater::class),
                    $name,
                    $params
                );
                break;
            case self::JOB_MODULE_ENABLE:
                return new JobModule(
                    $this->serviceLocator->get(ModuleEnableCommand::class),
                    $objectManagerProvider,
                    $streamOutput,
                    $cronStatus,
                    $name,
                    $params
                );
                break;
            case self::JOB_MODULE_DISABLE:
                return new JobModule(
                    $this->serviceLocator->get(ModuleDisableCommand::class),
                    $objectManagerProvider,
                    $streamOutput,
                    $cronStatus,
                    $name,
                    $params
                );
                break;
            case self::JOB_ENABLE_CACHE:
                return new JobSetCache(
                    $objectManager->get(CacheEnableCommand::class),
                    $objectManagerProvider,
                    $streamOutput,
                    $cronStatus,
                    $name,
                    $params
                );
                break;
            case self::JOB_DISABLE_CACHE:
                return new JobSetCache(
                    $objectManager->get(CacheDisableCommand::class),
                    $objectManagerProvider,
                    $streamOutput,
                    $cronStatus,
                    $name
                );
                break;
            case self::JOB_MAINTENANCE_MODE_ENABLE:
                return new JobSetMaintenanceMode(
                    $this->serviceLocator->get(MaintenanceEnableCommand::class),
                    $objectManagerProvider,
                    $streamOutput,
                    $cronStatus,
                    $name,
                    $params
                );
                break;
            case self::JOB_MAINTENANCE_MODE_DISABLE:
                return new JobSetMaintenanceMode(
                    $this->serviceLocator->get(MaintenanceDisableCommand::class),
                    $objectManagerProvider,
                    $streamOutput,
                    $cronStatus,
                    $name,
                    $params
                );
                break;
            default:
                throw new \RuntimeException(sprintf('"%s" job is not supported.', $name));
                break;
        }
    }
}