Doctrine\Migrations\Configuration\Connection\Loader\ConnectionHelperLoader

Complexity

Total Complexity

5

Size/Duplication

Total Lines	33
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	2
A	getConnection()	0	11	3

<?php

declare(strict_types=1);

namespace Doctrine\Migrations\Configuration\Connection\Loader;

use Doctrine\DBAL\Connection;
use Doctrine\DBAL\Tools\Console\Helper\ConnectionHelper;
use Doctrine\Migrations\Configuration\Connection\ConnectionLoader;
use Symfony\Component\Console\Helper\HelperSet;

/**
 * The ConnectionHelperLoader is responsible for loading a Doctrine\DBAL\Connection from a Symfony Console HelperSet.
 *
 * @internal
 */
final class ConnectionHelperLoader implements ConnectionLoader
{
    /** @var string */
    private $helperName;

    /** @var  HelperSet */
    private $helperSet;

    /** @var ConnectionLoader */
    private $fallback;

    public function __construct(string $helperName, ConnectionLoader $fallback, ?HelperSet $helperSet = null)
    {
        $this->helperSet  = $helperSet ?: new HelperSet();
        $this->helperName = $helperName;
        $this->fallback   = $fallback;
    }

    /**
     * Read the input and return a Configuration, returns null if the config
     * is not supported.
     */
    public function getConnection() : Connection
    {
        if ($this->helperSet->has($this->helperName)) {
            $connectionHelper = $this->helperSet->get($this->helperName);

            if ($connectionHelper instanceof ConnectionHelper) {
                return $connectionHelper->getConnection();
            }
        }

        return $this->fallback->getConnection();
    }
}