system/database/DB_forge.php

   1 <?php  if ( ! defined('BASEPATH')) exit('No direct script access allowed');
   2 /**
   3  * Code Igniter
   4  *
   5  * An open source application development framework for PHP 5.1.6 or newer
   6  *
   7  * @package             CodeIgniter
   8  * @author              ExpressionEngine Dev Team
   9  * @copyright   Copyright (c) 2008 - 2011, EllisLab, Inc.
  10  * @license             http://codeigniter.com/user_guide/license.html
  11  * @link                http://codeigniter.com
  12  * @since               Version 1.0
  13  * @filesource
  14  */
  15
  16 // ------------------------------------------------------------------------
  17
  18 /**
  19  * Database Utility Class
  20  *
  21  * @category    Database
  22  * @author              ExpressionEngine Dev Team
  23  * @link                http://codeigniter.com/user_guide/database/
  24  */
  25 class CI_DB_forge {
  26
  27         var $fields                     = array();
  28         var $keys                       = array();
  29         var $primary_keys       = array();
  30         var $db_char_set        =       '';
  31
  32         /**
  33          * Constructor
  34          *
  35          * Grabs the CI super object instance so we can access it.
  36          *
  37          */
  38         function __construct()
  39         {
  40                 // Assign the main database object to $this->db
  41                 $CI =& get_instance();
  42                 $this->db =& $CI->db;
  43                 log_message('debug', "Database Forge Class Initialized");
  44         }
  45
  46         // --------------------------------------------------------------------
  47
  48         /**
  49          * Create database
  50          *
  51          * @access      public
  52          * @param       string  the database name
  53          * @return      bool
  54          */
  55         function create_database($db_name)
  56         {
  57                 $sql = $this->_create_database($db_name);
  58
  59                 if (is_bool($sql))
  60                 {
  61                         return $sql;
  62                 }
  63
  64                 return $this->db->query($sql);
  65         }
  66
  67         // --------------------------------------------------------------------
  68
  69         /**
  70          * Drop database
  71          *
  72          * @access      public
  73          * @param       string  the database name
  74          * @return      bool
  75          */
  76         function drop_database($db_name)
  77         {
  78                 $sql = $this->_drop_database($db_name);
  79
  80                 if (is_bool($sql))
  81                 {
  82                         return $sql;
  83                 }
  84
  85                 return $this->db->query($sql);
  86         }
  87
  88         // --------------------------------------------------------------------
  89
  90         /**
  91          * Add Key
  92          *
  93          * @access      public
  94          * @param       string  key
  95          * @param       string  type
  96          * @return      void
  97          */
  98         function add_key($key = '', $primary = FALSE)
  99         {
 100                 if (is_array($key))
 101                 {
 102                         foreach ($key as $one)
 103                         {
 104                                 $this->add_key($one, $primary);
 105                         }
 106
 107                         return;
 108                 }
 109
 110                 if ($key == '')
 111                 {
 112                         show_error('Key information is required for that operation.');
 113                 }
 114
 115                 if ($primary === TRUE)
 116                 {
 117                         $this->primary_keys[] = $key;
 118                 }
 119                 else
 120                 {
 121                         $this->keys[] = $key;
 122                 }
 123         }
 124
 125         // --------------------------------------------------------------------
 126
 127         /**
 128          * Add Field
 129          *
 130          * @access      public
 131          * @param       string  collation
 132          * @return      void
 133          */
 134         function add_field($field = '')
 135         {
 136                 if ($field == '')
 137                 {
 138                         show_error('Field information is required.');
 139                 }
 140
 141                 if (is_string($field))
 142                 {
 143                         if ($field == 'id')
 144                         {
 145                                 $this->add_field(array(
 146                                                                                 'id' => array(
 147                                                                                                         'type' => 'INT',
 148                                                                                                         'constraint' => 9,
 149                                                                                                         'auto_increment' => TRUE
 150                                                                                                         )
 151                                                                 ));
 152                                 $this->add_key('id', TRUE);
 153                         }
 154                         else
 155                         {
 156                                 if (strpos($field, ' ') === FALSE)
 157                                 {
 158                                         show_error('Field information is required for that operation.');
 159                                 }
 160
 161                                 $this->fields[] = $field;
 162                         }
 163                 }
 164
 165                 if (is_array($field))
 166                 {
 167                         $this->fields = array_merge($this->fields, $field);
 168                 }
 169
 170         }
 171
 172         // --------------------------------------------------------------------
 173
 174         /**
 175          * Create Table
 176          *
 177          * @access      public
 178          * @param       string  the table name
 179          * @return      bool
 180          */
 181         function create_table($table = '', $if_not_exists = FALSE)
 182         {
 183                 if ($table == '')
 184                 {
 185                         show_error('A table name is required for that operation.');
 186                 }
 187
 188                 if (count($this->fields) == 0)
 189                 {
 190                         show_error('Field information is required.');
 191                 }
 192
 193                 $sql = $this->_create_table($this->db->dbprefix.$table, $this->fields, $this->primary_keys, $this->keys, $if_not_exists);
 194
 195                 $this->_reset();
 196                 return $this->db->query($sql);
 197         }
 198
 199         // --------------------------------------------------------------------
 200
 201         /**
 202          * Drop Table
 203          *
 204          * @access      public
 205          * @param       string  the table name
 206          * @return      bool
 207          */
 208         function drop_table($table_name)
 209         {
 210                 $sql = $this->_drop_table($this->db->dbprefix.$table_name);
 211
 212                 if (is_bool($sql))
 213                 {
 214                         return $sql;
 215                 }
 216
 217                 return $this->db->query($sql);
 218         }
 219
 220         // --------------------------------------------------------------------
 221
 222         /**
 223          * Rename Table
 224          *
 225          * @access      public
 226          * @param       string  the old table name
 227          * @param       string  the new table name
 228          * @return      bool
 229          */
 230         function rename_table($table_name, $new_table_name)
 231         {
 232                 if ($table_name == '' OR $new_table_name == '')
 233                 {
 234                         show_error('A table name is required for that operation.');
 235                 }
 236
 237                 $sql = $this->_rename_table($this->db->dbprefix.$table_name, $this->db->dbprefix.$new_table_name);
 238                 return $this->db->query($sql);
 239         }
 240
 241         // --------------------------------------------------------------------
 242
 243         /**
 244          * Column Add
 245          *
 246          * @access      public
 247          * @param       string  the table name
 248          * @param       string  the column name
 249          * @param       string  the column definition
 250          * @return      bool
 251          */
 252         function add_column($table = '', $field = array(), $after_field = '')
 253         {
 254                 if ($table == '')
 255                 {
 256                         show_error('A table name is required for that operation.');
 257                 }
 258
 259                 // add field info into field array, but we can only do one at a time
 260                 // so we cycle through
 261
 262                 foreach ($field as $k => $v)
 263                 {
 264                         $this->add_field(array($k => $field[$k]));
 265
 266                         if (count($this->fields) == 0)
 267                         {
 268                                 show_error('Field information is required.');
 269                         }
 270
 271                         $sql = $this->_alter_table('ADD', $this->db->dbprefix.$table, $this->fields, $after_field);
 272
 273                         $this->_reset();
 274
 275                         if ($this->db->query($sql) === FALSE)
 276                         {
 277                                 return FALSE;
 278                         }
 279                 }
 280
 281                 return TRUE;
 282
 283         }
 284
 285         // --------------------------------------------------------------------
 286
 287         /**
 288          * Column Drop
 289          *
 290          * @access      public
 291          * @param       string  the table name
 292          * @param       string  the column name
 293          * @return      bool
 294          */
 295         function drop_column($table = '', $column_name = '')
 296         {
 297
 298                 if ($table == '')
 299                 {
 300                         show_error('A table name is required for that operation.');
 301                 }
 302
 303                 if ($column_name == '')
 304                 {
 305                         show_error('A column name is required for that operation.');
 306                 }
 307
 308                 $sql = $this->_alter_table('DROP', $this->db->dbprefix.$table, $column_name);
 309
 310                 return $this->db->query($sql);
 311         }
 312
 313         // --------------------------------------------------------------------
 314
 315         /**
 316          * Column Modify
 317          *
 318          * @access      public
 319          * @param       string  the table name
 320          * @param       string  the column name
 321          * @param       string  the column definition
 322          * @return      bool
 323          */
 324         function modify_column($table = '', $field = array())
 325         {
 326                 if ($table == '')
 327                 {
 328                         show_error('A table name is required for that operation.');
 329                 }
 330
 331                 // add field info into field array, but we can only do one at a time
 332                 // so we cycle through
 333
 334                 foreach ($field as $k => $v)
 335                 {
 336                         // If no name provided, use the current name
 337                         if ( ! isset($field[$k]['name']))
 338                         {
 339                                 $field[$k]['name'] = $k;
 340                         }
 341
 342                         $this->add_field(array($k => $field[$k]));
 343
 344                         if (count($this->fields) == 0)
 345                         {
 346                                 show_error('Field information is required.');
 347                         }
 348
 349                         $sql = $this->_alter_table('CHANGE', $this->db->dbprefix.$table, $this->fields);
 350
 351                         $this->_reset();
 352
 353                         if ($this->db->query($sql) === FALSE)
 354                         {
 355                                 return FALSE;
 356                         }
 357                 }
 358
 359                 return TRUE;
 360         }
 361
 362         // --------------------------------------------------------------------
 363
 364         /**
 365          * Reset
 366          *
 367          * Resets table creation vars
 368          *
 369          * @access      private
 370          * @return      void
 371          */
 372         function _reset()
 373         {
 374                 $this->fields           = array();
 375                 $this->keys                     = array();
 376                 $this->primary_keys     = array();
 377         }
 378
 379 }
 380
 381 /* End of file DB_forge.php */
 382 /* Location: ./system/database/DB_forge.php */