Subversion Repositories SmartDukaan

Rev

Blame | Last modification | View Log | RSS feed

<?php  if ( ! defined('BASEPATH')) exit('No direct script access allowed');
/**
 * CodeIgniter
 *
 * An open source application development framework for PHP 5.1.6 or newer
 *
 * @package             CodeIgniter
 * @author              ExpressionEngine Dev Team
 * @copyright   Copyright (c) 2008 - 2011, EllisLab, Inc.
 * @license             http://codeigniter.com/user_guide/license.html
 * @link                http://codeigniter.com
 * @since               Version 1.0
 * @filesource
 */

// ------------------------------------------------------------------------

/**
 * Postgre Forge Class
 *
 * @category    Database
 * @author              ExpressionEngine Dev Team
 * @link                http://codeigniter.com/user_guide/database/
 */
class CI_DB_postgre_forge extends CI_DB_forge {

        /**
         * Create database
         *
         * @access      private
         * @param       string  the database name
         * @return      bool
         */
        function _create_database($name)
        {
                return "CREATE DATABASE ".$name;
        }

        // --------------------------------------------------------------------

        /**
         * Drop database
         *
         * @access      private
         * @param       string  the database name
         * @return      bool
         */
        function _drop_database($name)
        {
                return "DROP DATABASE ".$name;
        }

        // --------------------------------------------------------------------

        /**
         * Create Table
         *
         * @access      private
         * @param       string  the table name
         * @param       array   the fields
         * @param       mixed   primary key(s)
         * @param       mixed   key(s)
         * @param       boolean should 'IF NOT EXISTS' be added to the SQL
         * @return      bool
         */
        function _create_table($table, $fields, $primary_keys, $keys, $if_not_exists)
        {
                $sql = 'CREATE TABLE ';

                if ($if_not_exists === TRUE)
                {
                        if ($this->db->table_exists($table))
                        {
                                return "SELECT * FROM $table"; // Needs to return innocous but valid SQL statement
                        }
                }

                $sql .= $this->db->_escape_identifiers($table)." (";
                $current_field_count = 0;

                foreach ($fields as $field=>$attributes)
                {
                        // Numeric field names aren't allowed in databases, so if the key is
                        // numeric, we know it was assigned by PHP and the developer manually
                        // entered the field information, so we'll simply add it to the list
                        if (is_numeric($field))
                        {
                                $sql .= "\n\t$attributes";
                        }
                        else
                        {
                                $attributes = array_change_key_case($attributes, CASE_UPPER);

                                $sql .= "\n\t".$this->db->_protect_identifiers($field);

                                $is_unsigned = (array_key_exists('UNSIGNED', $attributes) && $attributes['UNSIGNED'] === TRUE);

                                // Convert datatypes to be PostgreSQL-compatible
                                switch (strtoupper($attributes['TYPE']))
                                {
                                        case 'TINYINT':
                                                $attributes['TYPE'] = 'SMALLINT';
                                                break;
                                        case 'SMALLINT':
                                                $attributes['TYPE'] = ($is_unsigned) ? 'INTEGER' : 'SMALLINT';
                                                break;
                                        case 'MEDIUMINT':
                                                $attributes['TYPE'] = 'INTEGER';
                                                break;
                                        case 'INT':
                                                $attributes['TYPE'] = ($is_unsigned) ? 'BIGINT' : 'INTEGER';
                                                break;
                                        case 'BIGINT':
                                                $attributes['TYPE'] = ($is_unsigned) ? 'NUMERIC' : 'BIGINT';
                                                break;
                                        case 'DOUBLE':
                                                $attributes['TYPE'] = 'DOUBLE PRECISION';
                                                break;
                                        case 'DATETIME':
                                                $attributes['TYPE'] = 'TIMESTAMP';
                                                break;
                                        case 'LONGTEXT':
                                                $attributes['TYPE'] = 'TEXT';
                                                break;
                                        case 'BLOB':
                                                $attributes['TYPE'] = 'BYTEA';
                                                break;
                                }

                                // If this is an auto-incrementing primary key, use the serial data type instead
                                if (in_array($field, $primary_keys) && array_key_exists('AUTO_INCREMENT', $attributes) 
                                        && $attributes['AUTO_INCREMENT'] === TRUE)
                                {
                                        $sql .= ' SERIAL';
                                }
                                else
                                {
                                        $sql .=  ' '.$attributes['TYPE'];
                                }

                                // Modified to prevent constraints with integer data types
                                if (array_key_exists('CONSTRAINT', $attributes) && strpos($attributes['TYPE'], 'INT') === false)
                                {
                                        $sql .= '('.$attributes['CONSTRAINT'].')';
                                }

                                if (array_key_exists('DEFAULT', $attributes))
                                {
                                        $sql .= ' DEFAULT \''.$attributes['DEFAULT'].'\'';
                                }

                                if (array_key_exists('NULL', $attributes) && $attributes['NULL'] === TRUE)
                                {
                                        $sql .= ' NULL';
                                }
                                else
                                {
                                        $sql .= ' NOT NULL';
                                }

                                // Added new attribute to create unqite fields. Also works with MySQL
                                if (array_key_exists('UNIQUE', $attributes) && $attributes['UNIQUE'] === TRUE)
                                {
                                        $sql .= ' UNIQUE';
                                }
                        }

                        // don't add a comma on the end of the last field
                        if (++$current_field_count < count($fields))
                        {
                                $sql .= ',';
                        }
                }

                if (count($primary_keys) > 0)
                {
                        // Something seems to break when passing an array to _protect_identifiers()
                        foreach ($primary_keys as $index => $key)
                        {
                                $primary_keys[$index] = $this->db->_protect_identifiers($key);
                        }

                        $sql .= ",\n\tPRIMARY KEY (" . implode(', ', $primary_keys) . ")";
                }

                $sql .= "\n);";

                if (is_array($keys) && count($keys) > 0)
                {
                        foreach ($keys as $key)
                        {
                                if (is_array($key))
                                {
                                        $key = $this->db->_protect_identifiers($key);
                                }
                                else
                                {
                                        $key = array($this->db->_protect_identifiers($key));
                                }

                                foreach ($key as $field)
                                {
                                        $sql .= "CREATE INDEX " . $table . "_" . str_replace(array('"', "'"), '', $field) . "_index ON $table ($field); ";
                                }
                        }
                }

                return $sql;
        }

        // --------------------------------------------------------------------

        /**
         * Drop Table
         *
         * @access    private
         * @return    bool
         */
        function _drop_table($table)
        {
                return "DROP TABLE IF EXISTS ".$this->db->_escape_identifiers($table)." CASCADE";
        }

        // --------------------------------------------------------------------

        /**
         * Alter table query
         *
         * Generates a platform-specific query so that a table can be altered
         * Called by add_column(), drop_column(), and column_alter(),
         *
         * @access      private
         * @param       string  the ALTER type (ADD, DROP, CHANGE)
         * @param       string  the column name
         * @param       string  the table name
         * @param       string  the column definition
         * @param       string  the default value
         * @param       boolean should 'NOT NULL' be added
         * @param       string  the field after which we should add the new field
         * @return      object
         */
        function _alter_table($alter_type, $table, $column_name, $column_definition = '', $default_value = '', $null = '', $after_field = '')
        {
                $sql = 'ALTER TABLE '.$this->db->_protect_identifiers($table)." $alter_type ".$this->db->_protect_identifiers($column_name);

                // DROP has everything it needs now.
                if ($alter_type == 'DROP')
                {
                        return $sql;
                }

                $sql .= " $column_definition";

                if ($default_value != '')
                {
                        $sql .= " DEFAULT \"$default_value\"";
                }

                if ($null === NULL)
                {
                        $sql .= ' NULL';
                }
                else
                {
                        $sql .= ' NOT NULL';
                }

                if ($after_field != '')
                {
                        $sql .= ' AFTER ' . $this->db->_protect_identifiers($after_field);
                }

                return $sql;

        }

        // --------------------------------------------------------------------

        /**
         * Rename a table
         *
         * Generates a platform-specific query so that a table can be renamed
         *
         * @access      private
         * @param       string  the old table name
         * @param       string  the new table name
         * @return      string
         */
        function _rename_table($table_name, $new_table_name)
        {
                $sql = 'ALTER TABLE '.$this->db->_protect_identifiers($table_name)." RENAME TO ".$this->db->_protect_identifiers($new_table_name);
                return $sql;
        }


}

/* End of file postgre_forge.php */
/* Location: ./system/database/drivers/postgre/postgre_forge.php */