<?php
/**
 * sysPass
 *
 * @author    nuxsmin
 * @link      http://syspass.org
 * @copyright 2012-2015 Rubén Domínguez nuxsmin@syspass.org
 *
 * This file is part of sysPass.
 *
 * sysPass is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * sysPass is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with sysPass.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

namespace SP\Storage;

use PDO;
use SP\Log\Log;
use SP\Core\SPException;

defined('APP_ROOT') || die(_('No es posible acceder directamente a este archivo'));

/**
 * Esta clase es la encargada de realizar las operaciones con la BBDD de sysPass.
 */
class DB
{
    /**
     * @var string
     */
    public static $txtError = '';
    /**
     * @var int
     */
    public static $numError = 0;
    /**
     * @var int
     */
    public static $lastNumRows = 0;
    /**
     * @var int
     */
    public static $lastId = null;
    /**
     * @var bool Resultado como array
     */
    private static $_retArray = false;
    /**
     * @var bool Resultado como un objeto PDO
     */
    private static $_returnRawData = false;
    /**
     * @var bool Contar el número de filas totales
     */
    private static $_fullRowCount = false;
    /**
     * @var int Número de registros obtenidos
     */
    private $_numRows = 0;
    /**
     * @var int Número de campos de la consulta
     */
    private $_numFields = 0;
    /**
     * @var array Resultados de la consulta
     */
    private $_lastResult = null;
    /**
     * @var string Nombre de la función que realiza la consulta
     */
    private $_querySource;
    /**
     * Datos para el objeto PDOStatement
     *
     * @var array
     */
    private $_stData;

    /**
     * @return int
     */
    public static function getLastId()
    {
        return self::$lastId;
    }

    /**
     * Establecer si se devuelve un array de objetos siempre
     */
    public static function setReturnArray()
    {
        self::$_retArray = true;
    }

    /**
     * Obtener los resultados de una consulta.
     *
     * @param  $query       string    con la consulta a realizar
     * @param  $querySource string    con el nombre de la función que realiza la consulta
     * @param  $data        array     con los datos de la consulta
     * @return bool|array devuelve bool si hay un error. Devuelve array con el array de registros devueltos
     */
    public static function getResults($query, $querySource, &$data = null)
    {
        if (empty($query)) {
            self::resetVars();
            return false;
        }

        try {
            $db = new DB();
            $db->_querySource = $querySource;
            $db->_stData = $data;
            $doQuery = $db->doQuery($query, $querySource, self::$_returnRawData);
            self::$lastNumRows = (self::$_fullRowCount === false) ? $db->_numRows : $db->getFullRowCount($query);
        } catch (SPException $e) {
            self::logDBException($query, $e->getMessage(), $e->getCode(), $querySource);
            return false;
        }

        if (self::$_returnRawData && is_object($doQuery) && get_class($doQuery) == "PDOStatement") {
            return $doQuery;
        }

        if ($db->_numRows == 0) {
            self::resetVars();
            return false;
        }

        if ($db->_numRows == 1 && self::$_retArray === false) {
            self::resetVars();
            return $db->_lastResult[0];
        }

        self::resetVars();
        return $db->_lastResult;
    }

    /**
     * Restablecer los atributos estáticos
     */
    private static function resetVars()
    {
        self::$_returnRawData = false;
        self::$_fullRowCount = false;
        self::$_retArray = false;
    }

    /**
     * Realizar una consulta a la BBDD.
     *
     * @param $query       string  con la consulta a realizar
     * @param $querySource string  con el nombre de la función que realiza la consulta
     * @param $getRawData  bool    realizar la consulta para obtener registro a registro
     * @return false|int devuelve bool si hay un error. Devuelve int con el número de registros
     * @throws SPException
     */
    public function doQuery(&$query, $querySource, $getRawData = false)
    {
        $isSelect = preg_match("/^(select|show)\s/i", $query);

        // Limpiar valores de caché y errores
        $this->_lastResult = array();

        try {
            $queryRes = $this->prepareQueryData($query);
        } catch (SPException $e) {
            throw $e;
        }

        if ($isSelect) {
            if (!$getRawData) {
                $this->_numFields = $queryRes->columnCount();
                $this->_lastResult = $queryRes->fetchAll(PDO::FETCH_OBJ);
            } else {
                return $queryRes;
            }

//            $queryRes->closeCursor();

            $this->_numRows = count($this->_lastResult);
        }
    }

    /**
     * Asociar los parámetros de la consulta utilizando el tipo adecuado
     *
     * @param &$query  string La consulta a realizar
     * @param $isCount bool   Indica si es una consulta de contador de registros
     * @return bool|\PDOStatement
     * @throws SPException
     */
    private function prepareQueryData(&$query, $isCount = false)
    {
        if ($isCount === true) {
            // No incluimos en el array de parámetros de posición los valores
            // utilizados para LIMIT
            preg_match_all('/(\?|:)/', $query, $count);

            // Indice a partir del cual no se incluyen valores
            $paramMaxIndex = (count($count[1]) > 0) ? count($count[1]) : 0;
        }

        try {
            $db = DBConnectionFactory::getFactory()->getConnection();

            if (is_array($this->_stData)) {
                $sth = $db->prepare($query);
                $paramIndex = 0;

                foreach ($this->_stData as $param => $value) {
                    // Si la clave es un número utilizamos marcadores de posición "?" en
                    // la consulta. En caso contrario marcadores de nombre
                    $param = (is_int($param)) ? $param + 1 : ':' . $param;

                    if ($isCount === true && count($count) > 0 && $paramIndex >= $paramMaxIndex) {
                        continue;
                    }

                    if ($param === 'blobcontent') {
                        $sth->bindValue($param, $value, PDO::PARAM_LOB);
                    } elseif (is_int($value)) {
//                        error_log("INT: " . $param . " -> " . $value);
                        $sth->bindValue($param, $value, PDO::PARAM_INT);
                    } else {
//                        error_log("STR: " . $param . " -> " . $value);
                        $sth->bindValue($param, $value, PDO::PARAM_STR);
                    }

                    $paramIndex++;
                }

                $sth->execute();
            } else {
                $sth = $db->query($query);
            }

            DB::$lastId = $db->lastInsertId();

            return $sth;
        } catch (\Exception $e) {
            error_log("Exception: " . $e->getMessage());
            throw new SPException(SPException::SP_CRITICAL, $e->getMessage(), $e->getCode());
        }
    }

    /**
     * Obtener el número de filas de una consulta realizada
     *
     * @param &$query string La consulta para contar los registros
     * @return int Número de files de la consulta
     * @throws SPException
     */
    private function getFullRowCount(&$query)
    {
        if (empty($query)) {
            return 0;
        }

        $num = 0;
        $patterns = array(
            '/(LIMIT|ORDER BY|GROUP BY).*/i',
            '/SELECT DISTINCT\s([\w_]+),.* FROM/iU',
            '/SELECT [\w_]+,.* FROM/iU',
        );
        $replace = array('', 'SELECT COUNT(DISTINCT \1) FROM', 'SELECT COUNT(*) FROM', '');

        preg_match('/SELECT DISTINCT\s([\w_]*),.*\sFROM\s([\w_]*)\s(LEFT|RIGHT|WHERE).*/iU', $query, $match);

        $query = preg_replace($patterns, $replace, $query);

        try {
            $db = DBConnectionFactory::getFactory()->getConnection();

            if (!is_array($this->_stData)) {
                $queryRes = $db->query($query);
                $num = intval($queryRes->fetchColumn());
            } else {
                if ($queryRes = $this->prepareQueryData($query, true)) {
                    $num = intval($queryRes->fetchColumn());
                }
            }

            $queryRes->closeCursor();

            return $num;
        } catch (SPException $e) {
            error_log("Exception: " . $e->getMessage());
            throw new SPException(SPException::SP_CRITICAL, $e->getMessage(), $e->getCode());
        }
    }

    /**
     * Método para registar los eventos de BD en el log
     *
     * @param $query     string  La consulta que genera el error
     * @param $errorMsg  string  El mensaje de error
     * @param $errorCode int     El código de error
     */
    private static function logDBException($query, $errorMsg, $errorCode, $querySource)
    {
        $Log = new Log($querySource, Log::ERROR);
        $Log->addDescription($errorMsg . '(' . $errorCode . ')');
        $Log->addDetails('SQL', DBUtil::escape($query));
        $Log->writeLog();

        error_log($query);
        error_log($errorMsg);
    }

    /**
     * Realizar una consulta y devolver el resultado sin datos
     *
     * @param       $query       string La consulta a realizar
     * @param       $querySource string La función orígen de la consulta
     * @param array $data        Los valores de los parámetros de la consulta
     * @param       $getRawData  bool   Si se deben de obtener los datos como PDOStatement
     * @return bool
     */
    public static function getQuery($query, $querySource, array &$data = null, $getRawData = false)
    {
        if (empty($query)) {
            return false;
        }

        try {
            $db = new DB();
            $db->_querySource = $querySource;
            $db->_stData = $data;
            $db->doQuery($query, $querySource, $getRawData);
            DB::$lastNumRows = $db->_numRows;
        } catch (SPException $e) {
            self::logDBException($query, $e->getMessage(), $e->getCode(), $querySource);
            self::$txtError = $e->getMessage();
            self::$numError = $e->getCode();

            return false;
        }

        return true;
    }

    /**
     * Establecer si se devuelven los datos obtenidos como PDOStatement
     *
     * @param bool $on
     */
    public static function setReturnRawData($on = true)
    {
        self::$_returnRawData = (bool)$on;
    }

    /**
     * Establecer si es necesario contar el número total de resultados devueltos
     */
    public static function setFullRowCount()
    {
        self::$_fullRowCount = true;
    }


    /**
     * Establecer los parámetos de la consulta preparada
     *
     * @param &$data array Con los datos de los parámetros de la consulta
     */
    public function setParamData(&$data)
    {
        $this->_stData = $data;
    }
}