<?php
/*
UserSpice 4
An Open Source PHP User Management System
by the UserSpice Team at http://UserSpice.com

This program 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.

This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
*/
class DB2 {
    private static $_instance = null;
    private $_pdo, $_query, $_error = false, $_results, $_resultsArray, $_count = 0, $_lastId, $_queryCount=0;

    private function __construct(){
        try{
            $this->_pdo = new PDO('mysql:host=' .
                Config::get('mysql2/host2') .';dbname='.
                Config::get('mysql2/db2'),
                Config::get('mysql2/username2'),
                Config::get('mysql2/password2'),
                array(PDO::MYSQL_ATTR_INIT_COMMAND => "SET SESSION sql_mode = ''"));
        } catch(PDOException $e){
            die($e->getMessage());
        }
    }

    public static function getInstance(){
        if (!isset(self::$_instance)) {
            self::$_instance = new DB2();
        }
        return self::$_instance;
    }

    public function query($sql, $params = array()){
        $this->_queryCount++;
        $this->_error = false;
        if ($this->_query = $this->_pdo->prepare($sql)) {
            $x = 1;
            if (count($params)) {
                foreach ($params as $param) {
                    $this->_query->bindValue($x, $param);
                    $x++;
                }
            }

            if ($this->_query->execute()) {
                $this->_results = $this->_query->fetchALL(PDO::FETCH_OBJ);
                $this->_resultsArray = json_decode(json_encode($this->_results),true);
                $this->_count = $this->_query->rowCount();
                $this->_lastId = $this->_pdo->lastInsertId();
            } else{
                $this->_error = true;
            }
        }
        return $this;
    }

    public function findAll($table){
        return $this->action('SELECT *',$table);
    }

    public function findById($id,$table){
        return $this->action('SELECT *',$table,array('id','=',$id));
    }

    public function action($action, $table, $where = array()){
        $sql = "{$action} FROM {$table}";
        $value = '';
        if (count($where) === 3) {
            $operators = array('=', '>', '<', '>=', '<=');

            $field = $where[0];
            $operator = $where[1];
            $value = $where[2];

            if(in_array($operator, $operators)){
                $sql .= " WHERE {$field} {$operator} ?";
            }
        }
        if (!$this->query($sql, array($value))->error()) {
            return $this;
        }
        return false;
    }

    public function get($table, $where){
        return $this->action('SELECT *', $table, $where);
    }

    public function delete($table, $where){
        return $this->action('DELETE', $table, $where);
    }

    public function deleteById($table,$id){
        return $this->action('DELETE',$table,array('id','=',$id));
    }

    public function insert($table, $fields = array()){
        $keys = array_keys($fields);
        $values = null;
        $x = 1;

        foreach ($fields as $field) {
            $values .= "?";
            if ($x < count($fields)) {
                $values .= ', ';
            }
            $x++;
        }

        $sql = "INSERT INTO {$table} (`". implode('`,`', $keys)."`) VALUES ({$values})";

        if (!$this->query($sql, $fields)->error()) {
            return true;
        }
        return false;
    }

    public function update($table, $id, $fields){
        $set = '';
        $x = 1;

        foreach ($fields as $name => $value) {
            $set .= "{$name} = ?";
            if ($x < count($fields)) {
                $set .= ', ';
            }
            $x++;
        }

        $sql = "UPDATE {$table} SET {$set} WHERE id = {$id}";

        if (!$this->query($sql, $fields)->error()) {
            return true;
        }
        return false;
    }

    public function results($assoc = false){
        if($assoc) return $this->_resultsArray;
        return $this->_results;
    }

    public function first(){
        return $this->results()[0];
    }

    public function count(){
        return $this->_count;
    }

    public function error(){
        return $this->_error;
    }

    public function lastId(){
        return $this->_lastId;
    }

    public function getQueryCount(){
        return $this->_queryCount;
    }

}