<?php
/**
 * Created by PhpStorm.
 * User: Zip
 * Date: 14/11/28
 * Time: 上午1:28
 */

namespace Hood\Dao\Db\PostgreSQL;

use Hood\Core\Root;
use Hood\Debug\DebugException;
use Hood\Core\Server;
use Hood\Dao\Db\PDOMySQL\Result;
use \Closure;
use Hood\Cache as hoodCache;

/**
 * 适配器
 * Class Connection
 * @package Hood\Dao\Db\PostgreSQL
 */
class PDOConnection extends Root
{

    /**
     * 关闭缓存
     * @var bool
     */
    protected $_cacheStatus = false;

    /**
     * 数据库连接池
     * @var array
     */
    protected $_instances = array();

    /**
     * 缓存驱动
     * @var string
     */
    protected $_cacheDrive = array(
        'drive' => 'Memcached',
        'node' => null,
        'child_node' => 'hosts'
    );

    /**
     * 属性
     * @var array
     */
    private $_attribute = array(
        \PDO::ATTR_CASE => \PDO::CASE_NATURAL
    );


    /**
     * 设置指定集合字段
     * @var int
     */
    private $_columnNumber = 0;

    /**
     * 数据库读取模式
     * @var array
     */
    private $_modality = null;

    /**
     * FETCH_CLASS 专用 class name
     * @var null
     */
    protected $_className = null;

    /**
     *
     * @var \PDOStatement
     */
    protected $_PDOStatement;


    /**
     *
     * @var \PDO
     */
    protected $_PDOConn;


    /**
     * 数据类型
     * @var array
     */
    protected $paramType = array(
        'bool' => \PDO::PARAM_BOOL,
        'boolean' => \PDO::PARAM_BOOL,
        'null' => \PDO::PARAM_NULL,
        'integer' => \PDO::PARAM_INT,
        'int' => \PDO::PARAM_INT,
        'string' => \PDO::PARAM_STR,
        'stmt' => \PDO::PARAM_STMT,
        'object' => \PDO::PARAM_LOB,
        'float' => \PDO::PARAM_STR,
        'double' => \PDO::PARAM_STR,
        'output' => \PDO::PARAM_INPUT_OUTPUT
    );

    protected $_dbname;

    /**
     * 写模式
     */
    const MODALITY_WRITE = 'write';

    /**
     * 读取模式
     */
    const MODALITY_READ = 'read';


    private $_dbConfig = array();

    /**
     * @var null
     */
    private $_form = null;

    /**
     *
     * @var array
     */
    private $_options = array(
        \PDO::ATTR_PERSISTENT => false,
        \PDO::MYSQL_ATTR_INIT_COMMAND => "SET NAMES UTF8"
    );

    /**
     * PDOConnection constructor.
     * @param array $dbConfig
     * @param array $options
     */
    public function __construct(array $dbConfig, array $options = array())
    {
        if (empty($dbConfig['read'])) {
            throw new DebugException('Error : DB Config Read is Null.');
        }
        if (empty($dbConfig['write'])) {
            throw new DebugException('Error : DB Config Write is Null.');
        }
        if (empty($dbConfig['dbname'])) {
            throw new DebugException('Error : DB Config Database is Null.');
        }
        if (empty($dbConfig['username'])) {
            throw new DebugException('Error : DB Config username is Null.');
        }
        $this->_dbConfig['read'] = $dbConfig['read'];
        $this->_dbConfig['write'] = $dbConfig['write'];
        $this->_dbname = $this->_dbConfig['dbname'] = $dbConfig['dbname'];
        $this->_dbConfig['username'] = $dbConfig['username'];
        if (!empty($dbConfig['passwd'])) {
            $this->_dbConfig['passwd'] = $dbConfig['passwd'];
        }
        if (!empty($options)) {
            $this->_options = $this->_options + $options;
        }
        parent::__construct();
    }

    /**
     * 强制设置数据库连接方式
     * @param $modality
     * @return $this
     */
    public function setModality($modality)
    {
        $this->_modality = $modality;
        return $this;
    }

    /**
     * 强制设置主
     * @return $this
     */
    public function master()
    {
        $this->_modality = 'write';
        return $this;
    }

    /**
     * 强制设置从
     * @return $this
     */
    public function slave()
    {
        $this->_modality = 'read';
        return $this;
    }

    /**
     * 创建PDO数据库对象
     * @param string $modality (write/read)
     * @return \PDO
     * @throws DebugException
     */
    public function connect($modality = 'write')
    {
        if ($this->_modality != null) $modality = $this->_modality;
        $connectKey = $this->_dbname . '_' . $modality;
        if (!isset($this->_instances[$connectKey])) {
            $_server = Server::config()->getServerMap($this->_dbConfig[$modality], Server::SERVER_SELECT_MODEL_RAND);
            #如果设置了 dbname 使用 配置的
            $dsn = 'pgsql:host=' . $_server['host'] . ';port=' . $_server['port'] . ';dbname=' . $this->_dbConfig['dbname'];
            try {
                $_connection = new \PDO($dsn, $this->_dbConfig['username'], $this->_dbConfig['passwd'], $this->_options);
            } catch (\Exception $e) {
                $_connection = null;
                throw new DebugException($dsn . ' error ' . $e->getMessage());
            }
            foreach ($this->_attribute as $key => $val) {
                $_connection->setAttribute($key, $val);
            }
            $this->_instances[$connectKey] = $_connection;
        }
        $this->_modality = null;
        return $this->_instances[$connectKey];
    }

    /**
     * 设置属性
     * @param $attribute
     * @param $value
     * @return $this
     */
    public function setAttribute($attribute, $value)
    {
        $this->_attribute[$attribute] = $value;
        return $this;
    }

    /**
     * 单例事务处理
     * @param callable $callback
     * @return mixed
     * @throws DebugException
     * @throws \Exception
     */
    public function transaction(Closure $callback)
    {
        $this->beginTransaction();
        try {
            $result = $callback($this);
            $this->commit();
        } catch (DebugException $e) {
            $this->rollBack();
        }
        return $result;
    }

    /**
     * 开启事务
     * @throws DebugException
     */
    public function beginTransaction()
    {
        $this->connect()->beginTransaction();
    }

    /**
     * 提交事务
     * @throws DebugException
     */
    public function commit()
    {
        $this->connect()->commit();
    }

    /**
     * 判断是否有事务开启
     * @return bool
     * @throws DebugException
     */
    public function inTransaction()
    {
        return $this->connect()->inTransaction();
    }

    /**
     * 回滚
     * @throws DebugException
     */
    public function rollBack()
    {
        $this->connect()->rollBack();
    }

    /**
     * 禁用的命令
     * @var array
     */
    private $disableCommands = array('select', 'insert', 'delete', 'update');

    /**
     * 检查禁用参数
     * @param $str
     * @return bool
     */
    protected function checkDisableCommands($str)
    {
        foreach ($this->disableCommands as $command) {
            if (stristr($str, $command) !== false) {
                return false;
            }
        }
        return true;
    }

    /**
     * 替换SQL的替换参数
     * @param $sql
     * @param array $replaceMap
     * @return string
     * @throws DebugException
     */
    protected function makeReplaceMapToSql($sql, array $replaceMap)
    {
        $_replaceMap = array();
        foreach ($replaceMap as $key => $value) {
            if (!is_string($value)) {
                throw new DebugException('替换参数值不能为数组.');
            }
            if ($this->checkDisableCommands($value) == false) {
                //throw new DebugException('替换参数包括了禁用sql方法.');
            }
            $_replaceMap["#" . $key . "#"] = $value;
        }
        return strtr($sql, $_replaceMap);
    }

    private function drawFunName($funName)
    {
        $funNameList = explode('_', $funName);
        $funList = array();
        foreach ($funNameList as $fun) {
            $funList[] = ucfirst($fun);
        }
        return implode('', $funList);
    }

    /**
     * 格式化 parameter
     * @param array $parameterMap
     * @return array
     */
    protected function formatParameter(array $parameterMap)
    {
        $_parameterMap = array();
        foreach ($parameterMap as $key => $val) {
            $funName = $this->drawFunName($key);
            $setFunName = 'set' . $funName;
            $getFunName = 'get' . $funName;
            if (property_exists($this->_form, $key) && method_exists($this->_form, $setFunName)) {
                $this->_form->$setFunName($val);
                $_parameterMap[$key] = $this->_form->$getFunName();
            } else {
                $_parameterMap[$key] = $val;
            }
        }
        return $_parameterMap;
    }

    /**
     * @param $sql
     * @param $parameterMap
     * @param $replaceMap
     * @return \PDOStatement
     * @throws DebugException
     */
    protected function _statement($sql, array $parameterMap, array $replaceMap, $modality = 'write')
    {
        if (is_object($this->_form)) {
            $this->formatParameter($parameterMap);
        }
        $sql = $this->makeReplaceMapToSql($sql, $replaceMap);
        $this->_PDOConn = $conn = $this->connect($modality);
        $this->_PDOStatement = $stmt = $conn->prepare($sql);
        $this->bindValues($parameterMap, $stmt);
        $stmt->execute();
        return $stmt;
    }

    /**
     * 执行SQL
     * @param $sql
     * @param $parameterMap
     * @throws DebugException
     */
    public function execute($sql, array $parameterMap = array())
    {
        $this->_PDOConn = $conn = $this->connect();
        $this->_PDOStatement = $stmt = $conn->prepare($sql);
        $this->bindValues($parameterMap, $stmt);
        $stmt->execute();
    }

    /**
     * 绑定一组数据
     * @param array $params
     * @param \PDOStatement $stmt
     * @throws DebugException
     */
    private function bindValues(array $params, \PDOStatement $stmt)
    {
        foreach ($params as $parameter => $value) {
            if (is_array($value) || is_object($value)) {
                throw new DebugException('Sql绑定参数不能为数组或对象.');
            }
            $dataType = $this->paramType [strtolower(gettype($value))];
            $stmt->bindValue($parameter, $value, $dataType);
        }
    }

    /**
     * 插入数据
     * @param $sql
     * @param array $parameterMap
     * @param array $replaceMap
     * @return Result
     */
    public function insert($sql, array $parameterMap = array(), array $replaceMap = array())
    {
        $this->_statement($sql, $parameterMap, $replaceMap);
        return new Result($this->_PDOConn, $this->_PDOStatement);
    }

    /**
     * 更新数据
     * @param string $sql
     * @param array $parameterMap
     * @param array $replaceMap
     * @return Result
     */
    public function update($sql, $parameterMap = array(), $replaceMap = array())
    {
        $this->_statement($sql, $parameterMap, $replaceMap);
        return new Result($this->_PDOConn, $this->_PDOStatement);
    }

    /**
     * 删除数据
     * @param string $sql
     * @param array $parameterMap
     * @param array $replaceMap
     * @return Result
     */
    public function delete($sql, $parameterMap = array(), $replaceMap = array())
    {
        $this->_statement($sql, $parameterMap, $replaceMap);
        return new Result($this->_PDOConn, $this->_PDOStatement);
    }

    /**
     * 原生Query
     * @param $sql
     * @return \PDOStatement
     * @throws DebugException
     */
    public function query($sql)
    {
        return $this->connect('read')->query($sql);
    }

    /**
     * 取回结果集中所有字段的值,作为关联数组返回  第一个字段作为码
     * @param $sql
     * @param array $parameterMap
     * @param array $replaceMap
     * @return array
     */
    public function fetchAssoc($sql, $parameterMap = array(), $replaceMap = array())
    {
        $stmt = $this->_statement($sql, $parameterMap, $replaceMap, 'read');
        $data = array();
        while ($row = $stmt->fetch(\PDO::FETCH_ASSOC)) {
            $tmp = array_values(array_slice($row, 0, 1));
            $data[$tmp[0]] = $row;
        }
        return $data;
    }

    /**
     * 设置结果集的Class
     * @param null $className
     * @return $this
     */
    public function setFetchClass($className = null)
    {
        $this->_className = (string)$className;
        return $this;
    }

    /**
     * 取回结果集中所有字段的值以对象形式输出,按自定义class
     * @param $sql
     * @param array $parameterMap
     * @param array $replaceMap
     * @return array
     */
    public function fetchClass($sql, $parameterMap = array(), $replaceMap = array())
    {
        return $this->_statement($sql, $parameterMap, $replaceMap, 'read')->fetchAll(\PDO::FETCH_CLASS, $this->_className);
    }


    /**
     * 取回结果集中所有字段的值以对象形式输出,按自定义class
     * @param $sql
     * @param array $parameterMap
     * @param array $replaceMap
     * @return array
     */
    public function fetchClassRow($sql, $parameterMap = array(), $replaceMap = array())
    {
        $stmt = $this->_statement($sql, $parameterMap, $replaceMap, 'read');
        $stmt->setFetchMode(\PDO::FETCH_CLASS, $this->_className);
        return $stmt->fetch();
    }

    /**
     * 关闭游标，使语句能再次被执行
     * @return $this
     */
    public function closeCursor()
    {
        $this->_PDOStatement->closeCursor();
        return $this;
    }

    /**
     * 返回结果集中的列数
     * @param $sql
     * @param array $parameterMap
     * @param array $replaceMap
     * @return int
     */
    public function columnCount($sql, $parameterMap = array(), $replaceMap = array())
    {
        return $this->_statement($sql, $parameterMap, $replaceMap, 'read')->columnCount();
    }

    /**
     * 打印一条 SQL 预处理命令
     * @param $sql
     * @param $parameterMap
     * @param $replaceMap
     * @return bool
     */
    public function debugDumpParams($sql, $parameterMap = array(), $replaceMap = array())
    {
        return $this->_statement($sql, $parameterMap, $replaceMap, 'read')->debugDumpParams();
    }


    /**
     * 获取所有数据集合
     * @param $sql
     * @param array $parameterMap
     * @param array $replaceMap
     * @return array
     */
    public function fetchAll($sql, $parameterMap = array(), $replaceMap = array())
    {
        return $this->_statement($sql, $parameterMap, $replaceMap, 'read')->fetchAll(\PDO::FETCH_ASSOC);
    }

    /**
     * 获取一条数据集合
     * @param $sql
     * @param array $parameterMap
     * @param array $replaceMap
     * @return mixed
     */
    public function fetchRow($sql, $parameterMap = array(), $replaceMap = array())
    {
        return $this->_statement($sql, $parameterMap, $replaceMap, 'read')->fetch(\PDO::FETCH_ASSOC);
    }

    /**
     * 取回一个相关数组
     * 第一个字段值为key
     * @param $sql
     * @param array $parameterMap
     * @param array $replaceMap
     * @return array
     */
    public function fetchPairs($sql, $parameterMap = array(), $replaceMap = array())
    {
        $stmt = $this->_statement($sql, $parameterMap, $replaceMap, 'read');
        $data = array();
        while ($row = $stmt->fetch(\PDO::FETCH_NUM)) {
            $data[$row[0]] = $row[1];
        }
        return $data;
    }

    /**
     * 取回第一个字段值
     * @param $sql
     * @param array $parameterMap
     * @param array $replaceMap
     * @return string
     */
    public function fetchOne($sql, $parameterMap = array(), $replaceMap = array())
    {
        return $this->_statement($sql, $parameterMap, $replaceMap, 'read')->fetchColumn(0);
    }

    /**
     * 字符转义
     * @param $value
     * @return string
     */
    public function quoteStr($value)
    {
        if (is_int($value)) {
            return $value;
        } elseif (is_float($value)) {
            return sprintf('%F', $value);
        }
        return "'" . addcslashes($value, "\000\n\r\\'\"\032") . "'";
    }

    /**
     * 字符转换
     * @param $string
     * @param \PDO $parameterType
     * @return string
     * @throws DebugException
     */
    public function quote($string, $parameterType = \PDO::PARAM_STR)
    {
        return $this->connect('read')->quote($string, $parameterType);
    }

    /**
     * 获取指定字段值
     * @param $sql
     * @param array $parameterMap
     * @param array $replaceMap
     * @return string
     */
    public function fetchColumn($sql, $parameterMap = array(), $replaceMap = array())
    {
        return $this->_statement($sql, $parameterMap, $replaceMap, 'read')->fetchColumn($this->_columnNumber);
    }

    /**
     * 取回所有结果行的第一个字段名
     * @param $sql
     * @param array $parameterMap
     * @param array $replaceMap
     * @return string
     */
    public function fetchCol($sql, $parameterMap = array(), $replaceMap = array())
    {
        return $this->_statement($sql, $parameterMap, $replaceMap, 'read')->fetchAll(\PDO::FETCH_COLUMN, 0);
    }

    /**
     * 设置指定集合数据字段
     * @param $columnNumber
     * @return $this
     */
    public function setFetchColumn($columnNumber)
    {
        $this->_columnNumber = (int)$columnNumber;
        return $this;
    }

    /**
     * 取回结果集中所有字段的值 以对象形式输出
     * @param $sql
     * @param array $parameterMap
     * @param array $replaceMap
     * @return mixed
     */
    public function fetchObject($sql, $parameterMap = array(), $replaceMap = array())
    {
        return $this->_statement($sql, $parameterMap, $replaceMap, 'read')->fetchObject($this->_className);
    }

    /**
     * 获取SQL
     * @param $sql
     * @param array $parameterMap
     * @param array $replaceMap
     * @return mixed|string
     * @throws DebugException
     */
    public function getSQL($sql, $parameterMap = array(), $replaceMap = array())
    {
        $sql = $this->makeReplaceMapToSql($sql, $replaceMap);
        if (strstr($sql, ':')) {
            $matches_s = array();
            foreach ($parameterMap as $key => $val) {
                if (is_string($val)) {
                    $val = "'{$val}'";
                }
                $matches_s[':' . $key] = $val;
            }
            $asSql = strtr($sql, $matches_s);
        } else {
            $asSql = $sql;
            foreach ($parameterMap as $val) {
                $strPos = strpos($asSql, '?');
                if (is_string($val)) {
                    $val = "'{$val}'";
                }
                $asSql = substr_replace($asSql, $val, $strPos, 1);
            }
        }
        return $asSql;
    }

    /**
     * 缓存tag
     * @param $tagName
     * @return $this
     */
    public function tag($tagName)
    {
        $this->_cacheStatus = true;
        $this->_cacheTagName = $tagName;
        return $this;
    }


    /**
     * 删除多个tag时用到
     */
    public function delTags()
    {
        $this->_delTags = func_get_args();
    }

    /**
     * 缓存key
     * @param string $key
     * @param null $prefix
     * @return $this
     */
    public function key($key)
    {
        $this->_cacheStatus = true;
        $this->_cacheKey = (string)$key;
        return $this;
    }

    /**
     * 缓存时间
     * @param $expire
     * @return $this
     */
    public function expire($expire)
    {
        $this->_cacheStatus = true;
        $this->_cacheExpire = (int)$expire;
        return $this;
    }

    /**
     * 设置缓存
     * @param bool $status
     * @param string $drive
     * @return $this
     */
    public function cache($status = true, array $drive = array())
    {
        if (isset($drive['drive'])) {
            $this->_cacheDrive['drive'] = $drive['drive'];
        }
        if (isset($drive['drive'])) {
            $this->_cacheDrive['node'] = $drive['node'];
        }
        if (isset($drive['drive'])) {
            $this->_cacheDrive['child_node'] = $drive['child_node'];
        }
        $this->_cacheStatus = (bool)$status;
        return $this;
    }

    /**
     * 重置参数
     */
    private function _resetParameter()
    {

    }

    /**
     * @param $formObj
     * @return $this
     */
    public function setForm($formObj)
    {
        $this->_form = $formObj;
        return $this;
    }
}