<?php

namespace bff\db\illuminate;

use bff;
use Closure;

/**
 * Illuminate Connection Extension
 * @copyright Tamaranga
 */
trait Connection
{
    /**
     * Set the fetch mode for the database connections.
     *
     * @param int $fetchMode
     * @return static
     */
    public function setFetchMode($fetchMode)
    {
        $this->fetchMode = $fetchMode;
    }

    /**
     * Get a new query builder instance.
     * > Use extended QueryBuilder
     *
     * @return \Illuminate\Database\Query\Builder|QueryBuilder
     */
    public function query()
    {
        return new QueryBuilder(
            $this,
            $this->getQueryGrammar(),
            $this->getPostProcessor()
        );
    }

    protected function run($query, $bindings, Closure $callback)
    {
        $result = null;
        $return = bff::filter('db.query.illuminate', false, [
            'result' => &$result,
            'query' => $query,
            'bindings' => &$bindings,
            'callback' => $callback,
            'connection' => $this,
        ]);
        if ($return) {
            return $result;
        }

        return parent::run($query, $bindings, $callback);
    }
}