<?php

namespace bff\db\illuminate;

use PDO;
use Illuminate\Database\DatabaseManager as Base;

/**
 * Illuminate Database Manager Extension
 * @copyright Tamaranga
 */
class DatabaseManager extends Base
{
    /**
     * Add default connection and it's fetch mode
     *
     * @param PDO $pdo
     * @param int $fetchMode fetch mode for the database connection
     */
    public function addDefaultConnection(PDO $pdo, $fetchMode)
    {
        $connection = null;
        switch (mb_strtolower($pdo->getAttribute(PDO::ATTR_DRIVER_NAME))) {
            case 'mysql': {
                $connection = new MySqlConnection($pdo);
            } break;
            case 'pgsql': {
                $connection = new PostgresConnection($pdo);
            } break;
            case 'sqlite': {
                $connection = new SQLiteConnection($pdo);
            } break;
            case 'sqlsrv': {
                $connection = new SqlServerConnection($pdo);
            } break;
        }
        if (! is_null($connection)) {
            $connection->setFetchMode($fetchMode);
            if (bff('debug')) {
                $connection->enableQueryLog();
            }
            $this->connections['default'] = $connection;
        }
    }
}