<?php

// Extended database class
// Date: 2011-11-25


class DatabaseResult
{
  var $PDOStatement;
  var $num_rows = 0;
  
  function fetch_assoc()
  {
    return($this->PDOStatement->fetch(PDO::FETCH_ASSOC));
  }
  
  function fetch_array()
  {
    return($this->PDOStatement->fetch(PDO::FETCH_BOTH));
  }

  function fetch_row()
  {
    return($this->PDOStatement->fetch(PDO::FETCH_NUM));
  }
}

class Database
{
  var $Prefix;
  var $Functions;
  var $Type;  
  var $PDO;
  var $Error;
  var $insert_id;
  var $LastQuery;
  var $ShowSQLError;
  var $ShowSQLQuery;
  var $LogFile;
  
  function __construct()
  {    
  	$this->Functions = array('NOW()', 'CURDATE()', 'CURTIME()', 'UUID()');
  	$this->Type = 'mysql'; // mysql, pgsql
  	$this->ShowSQLError = false;
  	$this->ShowSQLQuery = false;
  	$this->LogSQLQuery = false;
  	$this->LogFile = dirname(__FILE__).'/../Query.log';
  }
  
  function Connect($Host, $User, $Password, $Database)
  {    
    if($this->Type == 'mysql') $ConnectionString = 'mysql:host='.$Host.';dbname='.$Database;
      else if($this->Type == 'pgsql') $ConnectionString = 'pgsql:dbname='.$Database.';host='.$Host;
      else $ConnectionString = '';
    $this->PDO = new PDO($ConnectionString, $User, $Password);
  }
  
  function select_db($Database)
  {
    $this->query('USE `'.$Database.'`');
  }
  
  function query($Query)
  {
    $this->LastQuery = $Query;
    if($this->ShowSQLQuery == true) 
      echo('<div style="border-bottom-width: 1px; border-bottom-style: solid; padding-bottom: 3px; padding-top: 3px; font-size: 12px; font-family: Arial;">'.$Query.'</div>'."\n");
    if($this->LogSQLQuery == true)
      file_put_contents($this->LogFile, $Query."\n", FILE_APPEND);
    $Result = new DatabaseResult();
    $Result->PDOStatement = $this->PDO->query($Query);
    if($Result->PDOStatement)
    {
      $Result->num_rows = $Result->PDOStatement->rowCount();
      $this->insert_id = $this->PDO->lastInsertId();
    } else
    {
      $this->Error = $this->PDO->errorInfo();
      $this->Error = $this->Error[2];
      if(($this->Error != '') and ($this->ShowSQLError == true)) 
        echo('<div><strong>SQL Error: </strong>'.$this->Error.'<br />'.$Query.'</div>');
        throw new Exception('SQL Error: '.$this->Error);
    }
    return($Result);  
  }

  function select($Table, $What = '*', $Condition = 1)
  {   
    return($this->query('SELECT '.$What.' FROM `'.$this->Prefix.$Table.'` WHERE '.$Condition));  
  }

  function delete($Table, $Condition)
  {
    $this->query('DELETE FROM `'.$this->Prefix.$Table.'` WHERE '.$Condition);  
  }
  
  function insert($Table, $Data)
  {
    $Name = '';
    $Values = '';
    foreach($Data as $Key => $Value)
    {
      $Name .= ',`'.$Key.'`';
      if(!in_array($Value, $this->Functions)) 
      {
        if(is_null($Value)) $Value = 'NULL';
        else $Value = $this->PDO->quote($Value);
      }
      $Values .= ','.$Value;
    }
    $Name = substr($Name, 1);
    $Values = substr($Values, 1);
    $this->query('INSERT INTO `'.$this->Prefix.$Table.'` ('.$Name.') VALUES('.$Values.')'); 
    $this->insert_id = $this->PDO->lastInsertId();
  }
  
  function update($Table, $Condition, $Data)
  {
    $Values = '';
    foreach($Data as $Key => $Value)
    {
      if(!in_array($Value, $this->Functions)) 
      {
        if(is_null($Value)) $Value = 'NULL';
        else $Value = $this->PDO->quote($Value);
      }
      $Values .= ', `'.$Key.'`='.$Value;
    }
    $Values = substr($Values, 2);  
    $this->query('UPDATE `'.$this->Prefix.$Table.'` SET '.$Values.' WHERE ('.$Condition.')');
  }
  
  function replace($Table, $Data)
  {
    $Name = '';
    $Values = '';
    foreach($Data as $Key => $Value)
    {
      if(!in_array($Value, $this->Functions)) 
      {
        if(is_null($Value)) $Value = 'NULL';
        else $Value = $this->PDO->quote($Value);
      }
      $Name .= ',`'.$Key.'`';
      $Values .= ','.$Value;
    }
    $Name = substr($Name, 1);
    $Values = substr($Values, 1);
    //echo('REPLACE INTO `'.$this->Prefix.$Table.'` ('.$Name.') VALUES ('.$Values.')<br />');
    $this->query('REPLACE INTO `'.$this->Prefix.$Table.'` ('.$Name.') VALUES('.$Values.')');
    //echo($this->error().'<br>');
  }
  
  function charset($Charset)
  {
    $this->query('SET NAMES "'.$Charset.'"');
  }
  
  function real_escape_string($Text)
  {
    return(addslashes($Text));
  }  
}

function TimeToMysqlDateTime($Time)
{
  if($Time == NULL) return(NULL);
    else return(date('Y-m-d H:i:s', $Time));  
}

function TimeToMysqlDate($Time)
{
  if($Time == NULL) return(NULL);
    else return(date('Y-m-d', $Time));  
}

function TimeToMysqlTime($Time)
{
  if($Time == NULL) return(NULL);
    else return(date('H:i:s', $Time));  
}

function MysqlDateTimeToTime($DateTime)
{
  if($DateTime == '') return(0);      
  $Parts = explode(' ', $DateTime);
  $DateParts = explode('-', $Parts[0]); 
  $TimeParts = explode(':', $Parts[1]);
  $Result = mktime($TimeParts[0], $TimeParts[1], $TimeParts[2], $DateParts[1], $DateParts[2], $DateParts[0]); 
  return($Result);  
}

function MysqlDateToTime($Date)
{
  if($Date == '') return(0);
  return(MysqlDateTimeToTime($Date.' 0:0:0'));  
}

function MysqlTimeToTime($Time)
{
  if($Time == '') return(0);
  return(MysqlDateTimeToTime('0000-00-00 '.$Time));  
}