LiquidBox\Silex\Provider\MongoDbServiceProvider

Complexity

Total Complexity

40

Size/Duplication

Total Lines	243
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	2

Test Coverage

Coverage

100%

Importance

Changes

0

14 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	9	3
A	buildConnectionString()	0	13	3
A	buildUri()	0	8	3
A	getArg()	0	4	2
A	getArgConnectionAuthority()	0	9	2
A	getArgConnectionHost()	0	12	4
A	getArgConnectionHosts()	0	6	2
A	getArgUri()	0	8	3
A	getDefaultArg()	0	8	2
A	loadParameters()	0	15	4
B	loadSingletonParameters()	0	17	5
A	sanitizeUri()	0	8	3
A	setDefaultArg()	0	6	2
B	register()	0	30	2

<?php
/**
 * MongoDB service provider for the Silex micro-framework.
 *
 * @see http://php.net/manual/set.mongodb.php
 */

namespace LiquidBox\Silex\Provider;

use MongoDB;
use Pimple\Container;
use Pimple\ServiceProviderInterface;

/**
 * MongoDb Provider.
 *
 * @author Jonathan-Paul Marois <jonathanpaul.marois@gmail.com>
 */
class MongoDbServiceProvider implements ServiceProviderInterface
{
    /**
     * @var array Buffer.
     */
    private $args = array();

    /**
     * @var array
     */
    private $defaultArgs = array();

    /**
     * @var int|string
     */
    private $defaultConnection = 0;

    /**
     * @var string
     */
    private $id = 'mongodb';

    /**
     * @var string
     */
    private $instances = 'mongodb.clients';

    /**
     * @var bool
     */
    private $isLoaded = false;

    /**
     * @var array
     */
    private $parameters = array();

    /**
     * @param string $id
     * @param string $instances
     */
    public function __construct($id = null, $instances = null)
    {
        if (strlen($id)) {
            $this->id = $id;
        }
        if (strlen($instances)) {
            $this->instances = $instances;
        }
    }

    /**
     * @return string
     */
    private function buildConnectionString(array $connection)
    {
        $uri = $this->getArgConnectionAuthority($connection) . '/';

        if (!empty($connection['db'])) {
            $uri .= $connection['db'];
        }
        if (!empty($connection['options'])) {
            $uri .= '?' . http_build_query($connection['options']);
        }

        return rtrim($uri, '/');
    }

    /**
     * @param array|string $connection
     *
     * @return string
     */
    private function buildUri($connection)
    {
        if (!empty($connection)) {
            return 'mongodb://' . (
                is_array($connection) ? $this->buildConnectionString($connection) : $this->sanitizeUri($connection)
            );
        }
    }

    /**
     * @param \Pimple\Container $app
     * @param string            $name
     *
     * @return string
     */
    private function getArg(Container $app, $name)
    {
        return isset($this->args[$name]) ? $this->args[$name] : $this->getDefaultArg($app, $name);
    }

    private function getArgConnectionAuthority(array $connection)
    {
        if (empty($connection['user'])) {
            return $this->getArgConnectionHost($connection);
        }

        return $connection['user'] . ':' . rawurlencode($connection['pwd']) . '@' .
            $this->getArgConnectionHost($connection);
    }

    /**
     * @return string
     */
    private function getArgConnectionHost(array $connection = array())
    {
        if (!empty($connection['hosts'])) {
            return $this->getArgConnectionHosts($connection);
        }

        if (empty($connection['host'])) {
            $connection['host'] = '127.0.0.1';
        }

        return empty($connection['port']) ? $connection['host'] : $connection['host'] . ':' . $connection['port'];
    }

    private function getArgConnectionHosts(array $connection)
    {
        return implode(',', array_map(function ($hosts) {
            return $this->getArgConnectionHost(is_array($hosts) ? $hosts : array('host' => $hosts));
        }, $connection['hosts']));
    }

    /**
     * @return string
     */
    private function getArgUri()
    {
        if (!empty($this->args['uri'])) {
            return $this->args['uri'];
        } elseif (!empty($this->args['connection'])) {
            return $this->args['connection'];
        }
    }

    /**
     * @param \Pimple\Container $app
     * @param string            $name
     *
     * @return string|array
     */
    private function getDefaultArg(Container $app, $name)
    {
        if (!isset($this->defaultArgs[$name])) {
            $this->setDefaultArg($app, $name);
        }

        return $this->defaultArgs[$name];
    }

    private function loadParameters(Container $app)
    {
        if ($this->isLoaded) {
            return;
        }

        $this->isLoaded = true;

        if (empty($app['mongodb.uri']) || !is_array($app['mongodb.uri'])) {
            $this->loadSingletonParameters($app);
        } else {
            $this->parameters = $app['mongodb.uri'];
            $this->defaultConnection = array_keys($this->parameters)[0];
        }
    }

    private function loadSingletonParameters(Container $app)
    {
        $this->parameters[0] = array();

        if (!empty($app['mongodb.uri'])) {
            $this->parameters[0]['uri'] = $app['mongodb.uri'];
        } elseif (!empty($app['mongodb.connection'])) {
            $this->parameters[0]['connection'] = $app['mongodb.connection'];
        }

        if (!empty($app['mongodb.uri_options'])) {
            $this->parameters[0]['uri_options'] = $app['mongodb.uri_options'];
        }
        if (!empty($app['mongodb.driver_options'])) {
            $this->parameters[0]['driver_options'] = $app['mongodb.driver_options'];
        }
    }

    /**
     * @param string $uri
     *
     * @return string
     */
    private function sanitizeUri($uri)
    {
        if ('mongodb:' == substr($uri, 0, 8)) {
            $uri = substr($uri, 8);
        }

        return ('//' == substr($uri, 0, 2)) ? substr($uri, 2) : $uri;
    }

    /**
     * @param \Pimple\Container $app
     * @param string            $name
     */
    private function setDefaultArg(Container $app, $name)
    {
        $this->defaultArgs[$name] = empty($app['mongodb.' . $name]) ?
            array('uri_options' => array(), 'driver_options' => array())[$name] :
            $app['mongodb.' . $name];
    }

    public function register(Container $app)
    {
        $app[$this->id] = function () use ($app) {
            $this->loadParameters($app);

            return $app[$this->instances][$this->defaultConnection];
        };
        $app[$this->instances] = function () use ($app) {
            $this->loadParameters($app);

            $instances = new Container();
            foreach ($this->parameters as $client => $this->args) {
                $instances[$client] = function () use ($app) {
                    return $app['mongodb.client'](
                        $this->getArgUri(),
                        $this->getArg($app, 'uri_options'),
                        $this->getArg($app, 'driver_options')
                    );
                };
            }

            return $instances;
        };

        $app['mongodb.client'] = $app->protect(
            function ($uri = null, array $uriOptions = array(), array $driverOptions = array()) {
                return new MongoDB\Client($this->buildUri($uri), $uriOptions, $driverOptions);
            }
        );
    }
}