facile-it / mongodb-bundle

src/Command/AbstractCommand.php

<?php

declare(strict_types=1);

namespace Facile\MongoDbBundle\Command;

use MongoDB\Database as Connection;
use Symfony\Component\Console\Command\Command;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Input\InputOption;
use Symfony\Component\Console\Output\OutputInterface;
use Symfony\Component\Console\Style\SymfonyStyle;
use Symfony\Component\DependencyInjection\ContainerInterface;

/**
 * Class AbstractCommand.
 */
abstract class AbstractCommand extends Command
{
    /** @var SymfonyStyle */
    protected $io;

    /** @var Connection */
    protected $connection;

    /** @var ContainerInterface */
    private $container;

    /**
     * AbstractCommand constructor.
     *
     * @param ContainerInterface $container
     */
    public function __construct(ContainerInterface $container, string $name = null)
    {
        parent::__construct($name);
        $this->container = $container;
    }

    protected function getContainer(): ContainerInterface
    {
        return $this->container;
    }

    /**
     * {@inheritdoc}
     */
    protected function configure()
    {
        parent::configure();
        $this
            ->addOption('connection', null, InputOption::VALUE_OPTIONAL, 'The connection to use for this command');
    }

    /**
     * {@inheritdoc}
     */
    protected function initialize(InputInterface $input, OutputInterface $output)
    {
        parent::initialize($input, $output);
        $this->io = new SymfonyStyle($input, $output);

        $connectionName = 'mongo.connection';

        if ($input->getOption('connection')) {
            $connectionName .= '.' . $input->getOption('connection');
        }

        if (! $this->container->has($connectionName)) {
            throw new \LogicException(sprintf('No connection named \'%s\' found', $input->getOption('connection')));

It seems like $input->getOption('connection') can also be of type string[]; however, parameter $values of sprintf() does only seem to accept double|integer|string, maybe add an additional type check?

        }

        $this->connection = $this->container->get($connectionName);
    }
}