misuzu/src/Database.php

<?php
namespace Misuzu;

use Illuminate\Database\Capsule\Manager as LaravelDatabaseManager;
use Misuzu\Config\ConfigManager;

class Database extends LaravelDatabaseManager
{
    private $configManager;

    private const SUPPORTED_DB_ALS = [
        'mysql',
        'sqlite',
        'pgsql',
        'sqlsrv',
    ];

    private const DEFAULT_PORT_MYSQL = 3306;
    private const DEFAULT_PORT_PGSQL = 5432;
    private const DEFAULT_PORT_MSSQL = 1433;

    private const DEFAULT_HOST = '127.0.0.1';

    public function __construct(
        ConfigManager $config,
        string $default = 'default',
        bool $startEloquent = true,
        bool $setAsGlobal = true
    ) {
        $this->configManager = $config;
        parent::__construct();

        $this->container['config']['database.default'] = $default;

        if ($startEloquent) {
            $this->bootEloquent();
        }

        if ($setAsGlobal) {
            $this->setAsGlobal();
        }
    }

    public function addConnectionFromConfig(string $section, string $name = 'default'): void
    {
        if (!$this->configManager->contains($section, 'driver')) {
            throw new \Exception('Config section not found!');
        }

        $driver = $this->configManager->get($section, 'driver');

        if (!in_array($driver, self::SUPPORTED_DB_ALS)) {
            throw new \Exception('Unsupported driver.');
        }

        $args = [
            'driver' => $driver,
            'database' => $this->configManager->get($section, 'database', 'string', 'misuzu'),
            'prefix' => $this->configManager->contains($section, 'prefix')
                ? $this->configManager->get($section, 'prefix', 'string')
                : '',
        ];

        switch ($driver) {
            case 'mysql':
                $is_unix_socket = $this->configManager->contains($section, 'unix_socket');

                $args['host'] = $is_unix_socket
                    ? ''
                    : $this->configManager->get($section, 'host', 'string', self::DEFAULT_HOST);

                $args['port'] = $is_unix_socket
                    ? self::DEFAULT_PORT_MYSQL
                    : $this->configManager->get($section, 'port', 'int', self::DEFAULT_PORT_MYSQL);

                $args['username'] = $this->configManager->get($section, 'username', 'string');
                $args['password'] = $this->configManager->get($section, 'password', 'string');
                $args['unix_socket'] = $is_unix_socket
                    ? $this->configManager->get($section, 'unix_socket', 'string')
                    : '';

                // these should probably be locked to these types
                $args['charset'] = $this->configManager->contains($section, 'charset')
                    ? $this->configManager->get($section, 'charset', 'string')
                    : 'utf8mb4';

                $args['collation'] = $this->configManager->contains($section, 'collation')
                    ? $this->configManager->get($section, 'collation', 'string')
                    : 'utf8mb4_bin';

                $args['strict'] = true;
                $args['engine'] = null;
                break;

            case 'pgsql':
                $is_unix_socket = $this->configManager->contains($section, 'unix_socket');

                $args['host'] = $is_unix_socket
                    ? ''
                    : $this->configManager->get($section, 'host', 'string', self::DEFAULT_HOST);

                $args['port'] = $is_unix_socket
                    ? self::DEFAULT_PORT_PGSQL
                    : $this->configManager->get($section, 'port', 'int', self::DEFAULT_PORT_PGSQL);

                $args['username'] = $this->configManager->get($section, 'username', 'string');
                $args['password'] = $this->configManager->get($section, 'password', 'string');

                $args['unix_socket'] = $is_unix_socket
                    ? $this->configManager->get($section, 'unix_socket', 'string')
                    : '';

                // these should probably be locked to these types
                $args['charset'] = $this->configManager->contains($section, 'charset')
                    ? $this->configManager->get($section, 'charset', 'string')
                    : 'utf8';

                $args['schema'] = 'public';
                $args['sslmode'] = 'prefer';
                break;

            case 'sqlsrv':
                $args['host'] = $this->configManager->get($section, 'host', 'string', self::DEFAULT_HOST);
                $args['port'] = $this->configManager->get($section, 'port', 'int', self::DEFAULT_PORT_MSSQL);
                $args['username'] = $this->configManager->get($section, 'username', 'string');
                $args['password'] = $this->configManager->get($section, 'password', 'string');

                // these should probably be locked to these types
                $args['charset'] = $this->configManager->contains($section, 'charset')
                    ? $this->configManager->get($section, 'charset', 'string')
                    : 'utf8';
                break;
        }

        $this->addConnection($args, $name);
    }
}