LiquidBox\Silex\Provider\MongoDbServiceProvider

Complexity

Total Complexity

45

Size/Duplication

Total Lines	262
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	1

Importance

Changes

0

13 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	9	3
A	buildUri()	0	4	1
A	getArg()	0	4	2
A	getArgUri()	0	10	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	34	4
A	buildAuthority()	0	14	3
B	buildConnectionString()	0	22	6
C	buildHost()	0	24	7

<?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(
        'uri_options' => array(),
        'driver_options' => 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;
        }
    }

    private function buildAuthority(array $connection)
    {
        $authority = '';

        if (!empty($connection['password'])) {
            $authority .= $connection['username'] . ':' .$connection['password'] . '@';
        } elseif (!empty($connection['username'])) {
            $authority .= $connection['username'] . '@';
        }

        $authority .= $this->buildHost($connection);

        return $authority;
    }

    /**
     * @return string
     */
    private function buildConnectionString(array $connection)
    {
        $uri = '//';

        if (isset($connection[0]) && count($connection) == 1) {
            $uri .= $this->sanitizeUri($connection[0]);
        } else {
            $uri .= $this->buildAuthority($connection);

            if (!empty($connection['database'])) {
                $uri .= '/' . $connection['database'];

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

        return $uri;
    }

    private function buildHost(array $components)
    {
        $hosts = array();

        if (isset($components['hosts'])) {
            $hosts[] = $this->buildHost($components['hosts']);
        } else {
            $hosts[] = empty($components['host']) ? '127.0.0.1' : $components['host'];
        }
        
        foreach ($components as $component) {
            if (is_array($component)) {
                $hosts = empty($component['host']) ? '127.0.0.1' : $component['host'];

                if (!empty($component['port'])) {
                    $hosts .= ':' . $component['port'];
                }
            } else {
                $hosts[] = $component;
            }
        }

        return implode(',', $hosts);
    }

    /**
     * @return string
     */
    private function buildUri(array $connection)
    {
        return 'mongodb:' . $this->buildConnectionString($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);
    }

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

        return $this->buildUri(
            !is_array($this->args['connection']) ? array($this->args['connection']) : $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['pdo.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($app),
                        $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()) use ($app) {
                if (null === $uri) {
                    $uri = array();
                }

                return new MongoDB\Client(is_array($uri) ? $this->buildUri($uri) : $uri, $uriOptions, $driverOptions);
            }
        );
    }
}