<?php
/* SVN FILE: $Id$ */

/**
 * PostgreSQL layer for DBO.
 *
 * Long description for file
 *
 * PHP versions 4 and 5
 *
 * CakePHP :  Rapid Development Framework <http://www.cakephp.org/>
 * Copyright (c) 2006, Cake Software Foundation, Inc.
 *                     1785 E. Sahara Avenue, Suite 490-204
 *                     Las Vegas, Nevada 89104
 *
 * Licensed under The MIT License
 * Redistributions of files must retain the above copyright notice.
 *
 * @filesource
 * @copyright    Copyright (c) 2006, Cake Software Foundation, Inc.
 * @link         http://www.cakefoundation.org/projects/info/cakephp CakePHP Project
 * @package      cake
 * @subpackage   cake.cake.libs.model.dbo
 * @since        CakePHP v 0.9.1.114
 * @version      $Revision$
 * @modifiedby   $LastChangedBy$
 * @lastmodified $Date$
 * @license      http://www.opensource.org/licenses/mit-license.php The MIT License
 */

/**
 * Include DBO.
 */
uses('model'.DS.'datasources'.DS.'dbo_source');

/**
 * PostgreSQL layer for DBO.
 *
 * Long description for class
 *
 * @package    cake
 * @subpackage cake.cake.libs.model.dbo
 * @since      CakePHP v 0.9.1.114
 */
class  DboPostgres extends DboSource
{

    var $description = "PostgreSQL DBO Driver";

    var $_baseConfig = array('persistent' => true,
                            'host'        => 'localhost',
                            'login'      => 'root',
                            'password'    => '',
                            'database'    => 'cake',
                            'port'        => 5432);

    var $columns = array(
        'primary_key' => array('name' => 'serial primary key'),
        'string'      => array('name' => 'varchar', 'limit' => '255'),
        'text'        => array('name' => 'text'),
        'integer'     => array('name' => 'integer'),
        'float'        => array('name' => 'float'),
        'datetime'    => array('name' => 'timestamp'),
        'timestamp'    => array('name' => 'timestamp'),
        'time'        => array('name' => 'time'),
        'date'        => array('name' => 'date'),
        'binary'      => array('name' => 'bytea'),
        'boolean'     => array('name' => 'boolean'),
	    'number'      => array('name' => 'numeric'));

/**
 * Connects to the database using options in the given configuration array.
 *
 * @return True if successfully connected.
 */
    function connect ()
    {
      $config = $this->config;
      $connect = $config['connect'];

      $this->connection = $connect("dbname={$config['database']} user={$config['login']} password={$config['password']}");
      if ($this->connection)
      {
          $this->connected = true;
      }
      else
      {
          $this->connected = false;
      }

      return $this->connected;
    }

/**
 * Disconnects from database.
 *
 * @return boolean True if the database could be disconnected, else false
 */
    function disconnect ()
    {
      return pg_close($this->connection);
    }

/**
 * Executes given SQL statement.
 *
 * @param string $sql SQL statement
 * @return resource Result resource identifier
 */
    function _execute ($sql)
    {
      return pg_query($this->connection, $sql);
    }

    function query ()
    {
      $args = func_get_args();
      if (count($args) == 1)
      {
         return $this->fetchAll($args[0]);
      }
      elseif (count($args) > 1 && strpos($args[0], 'findBy') === 0)
      {
         $field = Inflector::underscore(str_replace('findBy', '', $args[0]));
         $query = '"' . $args[2]->name . '"."' . $field . '" = ' . $this->value($args[1][0]);
         return $args[2]->find($query);
      }
      elseif (count($args) > 1 && strpos($args[0], 'findAllBy') === 0)
      {
         $field = Inflector::underscore(str_replace('findAllBy', '', $args[0]));
         $query = '"' . $args[2]->name . '"."' . $field . '" = ' . $this->value($args[1][0]);
         return $args[2]->findAll($query);
      }
    }

/**
 * Returns a row from given resultset as an array .
 *
 * @return array The fetched row as an array
 */
    function fetchRow ($assoc = false)
    {
	    if(is_resource($this->_result))
        {
            $this->resultSet($this->_result);
            $resultRow = $this->fetchResult();
            return $resultRow[0];
        }
        else
        {
            return null;
        }
    }

/**
 * Returns an array of tables in the database. If there are no tables, an error is raised and the application exits.
 *
 * @return array Array of tablenames in the database
 */
    function listSources ()
    {
	 $sql = "SELECT table_name as name FROM information_schema.tables WHERE table_schema = 'public';";

	$result = $this->query($sql);

      if (!$result)
      {
         return null;
      }
      else
      {
         $tables = array();
         foreach ($result as $item)
         {
             $tables[] = $item['name'];
         }
         return $tables;
      }
    }

/**
 * Generates the fields list of an SQL query.
 *
 * @param Model $model
 * @param string $alias Alias tablename
 * @param mixed $fields
 * @return array
 */
    function fields (&$model, $alias, $fields)
    {
      if (is_array($fields))
        {
            $fields = $fields;
        }
        else
        {
            if ($fields != null)
            {
                if (strpos($fields, ','))
                {
                    $fields = explode(',', $fields);
                }
                else
                {
                    $fields = array($fields);
                }
                $fields = array_map('trim', $fields);
            }
            else
            {
                 foreach ($model->_tableInfo->value as $field)
                 {
                     $fields[]= $field[0]['name'];
                 }

            }
        }

        $count = count($fields);
        if ($count >= 1 && $fields[0] != '*' && strpos($fields[0], 'COUNT(*)') === false)
        {
            for ($i = 0; $i < $count; $i++)
            {
                $dot = strrpos($fields[$i], '.');
                if ($dot === false)
                {
                    $fields[$i] = $this->name($alias).'.'.$this->name($fields[$i]) . ' AS ' . $this->name($alias . '__' . $fields[$i]);
                }
                else
                {
                    $build = explode('.',$fields[$i]);
                    $fields[$i] = $this->name($build[0]).'.'.$this->name($build[1]) . ' AS ' . $this->name($build[0] . '__' . $build[1]);
                }
            }
        }

        return $fields;
    }

/**
 * Returns an array of the fields in given table name.
 *
 * @param string $tableName Name of database table to inspect
 * @return array Fields in table. Keys are name and type
 */
    function &describe (&$model)
    {
        $cache = parent::describe($model);
        if ($cache != null)
        {
            return $cache;
        }

        $fields = false;

	$fields = $this->query("SELECT column_name as name, data_type as type FROM information_schema.columns WHERE table_name =".$this->value($model->table));

        $this->__cacheDescription($model->table, $fields);
        return $fields;
    }

/**
 * Returns a quoted and escaped string of $data for use in an SQL statement.
 *
 * @param string $data String to be prepared for use in an SQL statement
 * @return string Quoted and escaped
 */
    function name ($data)
    {
      if ($data == '*')
      {
      		return '*';
      }
      return '"'. ereg_replace('\.', '"."', $data) .'"';
    }

/**
 * Returns a quoted and escaped string of $data for use in an SQL statement.
 *
 * @param string $data String to be prepared for use in an SQL statement
 * @param string $column The column into which this data will be inserted
 * @return string Quoted and escaped
 * @todo Add logic that formats/escapes data based on column type
 */
    function value ($data, $column = null)
    {
      $parent = parent::value($data, $column);
      if ($parent != null)
        {
            return $parent;
        }
        if ($data === null)
        {
            return 'NULL';
        }
        if (ini_get('magic_quotes_gpc') == 1)
        {
            $data = stripslashes($data);
        }

        $data = pg_escape_string($data);

        $return = "'" . $data . "'";
        return $return;
    }

/**
 * Returns a formatted error message from previous database operation.
 *
 * @return string Error message
 */
    function lastError ()
    {
      $last_error = pg_last_error($this->connection);
      if ($last_error)
      {
          return $last_error;
      }
      return null;
    }

/**
 * Returns number of affected rows in previous database operation. If no previous operation exists, this returns false.
 *
 * @return int Number of affected rows
 */
    function lastAffected ()
    {
      if ($this->_result)
      {
          return pg_affected_rows($this->_result);
      }
      return false;
    }

/**
 * Returns number of rows in previous resultset. If no previous resultset exists,
 * this returns false.
 *
 * @return int Number of rows in resultset
 */
    function lastNumRows ()
    {
      if ($this->_result)
      {
          return pg_num_rows($this->_result);
      }
      return false;
    }

/**
 * Returns the ID generated from the previous INSERT operation.
 *
 * @param string $source Name of the database table
 * @param string $field Name of the ID database field. Defaults to "id"
 * @return int
 */
    function lastInsertId ($source, $field='id')
    {
      $sql = "SELECT CURRVAL('{$source}_{$field}_seq') AS max";
      $res = $this->rawQuery($sql);
      $data = $this->fetchRow($res);
      return $data['max'];
    }

/**
 * Returns a limit statement in the correct format for the particular database.
 *
 * @param int $limit Limit of results returned
 * @param int $offset Offset from which to start results
 * @return string SQL limit/offset statement
 */
    function limit ($limit, $offset = null)
    {
      if ($limit)
        {
            $rt = '';
            if (!strpos(strtolower($limit), 'limit') || strpos(strtolower($limit), 'limit') === 0)
            {
                $rt = ' LIMIT';
            }
            if ($offset)
            {
                $rt .= ' ' . $offset. ',';
            }
            $rt .= ' ' . $limit;
            return $rt;
        }
        return null;
    }

/**
 * Converts database-layer column types to basic types
 *
 * @param string $real Real database-layer column type (i.e. "varchar(255)")
 * @return string Abstract column type (i.e. "string")
 */
    function column($real)
    {
        $col = r(')', '', $real);
        $limit = null;
        @list($col, $limit) = explode('(', $col);

        if (in_array($col, array('date', 'time', 'timestamp')))
        {
            return $col;
        }
        if ($col == 'boolean')
        {
            return 'boolean';
        }
        if (strpos($col, 'integer') !== false)
        {
            return 'integer';
        }
        if (strpos($col, 'char') !== false)
        {
            return 'string';
        }
        if (strpos($col, 'text') !== false)
        {
            return 'text';
        }
        if (strpos($col, 'bytea') !== false)
        {
            return 'binary';
        }
        if (in_array($col, array('float', 'double', 'decimal')))
        {
            return 'float';
        }
        return 'text';
    }

/**
 * Enter description here...
 *
 * @param unknown_type $results
 */
    function resultSet(&$results)
    {
        $this->results =& $results;
        $this->map = array();
        $num_fields = pg_num_fields($results);
        $index = 0;
        $j = 0;

        while ($j < $num_fields)
        {
            $columnName = pg_field_name($results, $j);

            if (strpos($columnName, '__'))
            {
                $parts = explode('__', $columnName);
                $this->map[$index++] = array($parts[0], $parts[1]);
            }
            else
            {
                $this->map[$index++] = array(0, $columnName);
            }
            $j++;
        }
    }

/**
 * Fetches the next row from the current result set
 *
 * @return unknown
 */
    function fetchResult()
    {
        if ($row = pg_fetch_row($this->results))
        {
            $resultRow = array();
            $i =0;
            foreach ($row as $index => $field)
            {
                list($table, $column) = $this->map[$index];
                $resultRow[$table][$column] = $row[$index];
                $i++;
            }
            return $resultRow;
        }
        else
        {
            return false;
        }
    }
}

?>