<?php

namespace bff\db\illuminate;

use PDO;
use Illuminate\Container\Container;
use Illuminate\Events\Dispatcher;
use Illuminate\Database\Connectors\ConnectionFactory;
use Illuminate\Database\Capsule\Manager as CapsuleManager;

/**
 * Illuminate Manager Extension
 * @copyright Tamaranga
 */
class Manager extends CapsuleManager
{
    /**
     * @var PDO connection
     */
    protected $_connection;

    /**
     * @var string connection fetch mode
     */
    protected $_connection_fetchmode = PDO::FETCH_OBJ;

    /**
     * Manager constructor.
     *
     * @param PDO $connection
     * @param int $fetchMode
     * @param Container|null $container
     */
    public function __construct(PDO $connection, $fetchMode = PDO::FETCH_OBJ, Container $container = null)
    {
        $this->_connection = $connection;
        $this->_connection_fetchmode = $fetchMode;
        $this->setFetchMode($fetchMode);

        parent::__construct($container);
    }

    /**
     * Build the database manager instance.
     *
     * @return void
     */
    protected function setupManager()
    {
        $factory = new ConnectionFactory($this->container);
        $this->manager = new DatabaseManager($this->container, $factory);
        $this->manager->addDefaultConnection($this->_connection, $this->_connection_fetchmode);
    }

    /**
     * Get a query builder instance.
     * > Use extended QueryBuilder
     *
     * @param string $table
     * @param string $as
     * @param string $connection
     * @return QueryBuilder
     */
    public static function table($table, $as = null, $connection = null)
    {
        return static::$instance->connection($connection)->table($table, $as);
    }

    /**
     * Static init with existing connection
     *
     * @param PDO $connection
     * @param integer $fetchMode
     * @param Container|null $container
     * @return static
     */
    public static function init(PDO $connection, $fetchMode = PDO::FETCH_ASSOC, Container $container = null)
    {
        $capsule = new static($connection, $fetchMode, $container);
        $dispatcher = new Dispatcher(new Container());
        $capsule->setEventDispatcher($dispatcher);
        $capsule->setAsGlobal();
        $capsule->bootEloquent();

        return $capsule;
    }

    /**
     * Get the connection query log.
     *
     * @param string $connectionName
     * @return array
     */
    public function getQueryLog($connectionName = null)
    {
        return $this->getConnection($connectionName)->getQueryLog();
    }

    /**
     * Clear the query log.
     *
     * @param string $connectionName
     * @return void
     */
    public function flushQueryLog($connectionName = null)
    {
        $this->getConnection($connectionName)->flushQueryLog();
    }

    /**
     * Close PDO connection
     */
    public function disconnect()
    {
        $this->manager->disconnect();
        $this->_connection = null;
    }

    /**
     * Enable the query log on the connection.
     *
     * @return void
     */
    public function enableQueryLog()
    {
        $this->manager->enableQueryLog();
    }

    /**
     * Disable the query log on the connection.
     *
     * @return void
     */
    public function disableQueryLog()
    {
        $this->manager->disableQueryLog();
    }
}