Wordpress 2.5.1
[autoinstalls/wordpress.git] / wp-includes / wp-db.php
index b37078834847a4ee18c13cbe76c37153178afa9a..216c9cf027261e75d011e60d6393a1a8203abe8f 100644 (file)
@@ -7,6 +7,7 @@
 
 define('EZSQL_VERSION', 'WP1.25');
 define('OBJECT', 'OBJECT', true);
+define('OBJECT_K', 'OBJECT_K', false);
 define('ARRAY_A', 'ARRAY_A', false);
 define('ARRAY_N', 'ARRAY_N', false);
 
@@ -16,10 +17,13 @@ if (!defined('SAVEQUERIES'))
 class wpdb {
 
        var $show_errors = false;
+       var $suppress_errors = false;
+       var $last_error = '';
        var $num_queries = 0;
        var $last_query;
        var $col_info;
        var $queries;
+       var $prefix = '';
        var $ready = false;
 
        // Our tables
@@ -30,16 +34,13 @@ class wpdb {
        var $comments;
        var $links;
        var $options;
-       var $optiontypes;
-       var $optionvalues;
-       var $optiongroups;
-       var $optiongroup_options;
        var $postmeta;
        var $usermeta;
        var $terms;
        var $term_taxonomy;
        var $term_relationships;
-
+       var $tables = array('users', 'usermeta', 'posts', 'categories', 'post2cat', 'comments', 'links', 'link2cat', 'options',
+                       'postmeta', 'terms', 'term_taxonomy', 'term_relationships');
        var $charset;
        var $collate;
 
@@ -66,7 +67,7 @@ class wpdb {
                if ( defined('DB_COLLATE') )
                        $this->collate = DB_COLLATE;
 
-               $this->dbh = @mysql_connect($dbhost, $dbuser, $dbpassword);
+               $this->dbh = @mysql_connect($dbhost, $dbuser, $dbpassword, true);
                if (!$this->dbh) {
                        $this->bail("
 <h1>Error establishing a database connection</h1>
@@ -83,7 +84,7 @@ class wpdb {
 
                $this->ready = true;
 
-               if ( !empty($this->charset) && version_compare(mysql_get_server_info(), '4.1.0', '>=') )
+               if ( !empty($this->charset) && version_compare(mysql_get_server_info($this->dbh), '4.1.0', '>=') )
                        $this->query("SET NAMES '$this->charset'");
 
                $this->select($dbname);
@@ -93,6 +94,26 @@ class wpdb {
                return true;
        }
 
+       function set_prefix($prefix) {
+
+               if ( preg_match('|[^a-z0-9_]|i', $prefix) )
+                       return new WP_Error('invalid_db_prefix', 'Invalid database prefix'); // No gettext here
+
+               $old_prefix = $this->prefix;
+               $this->prefix = $prefix;
+
+               foreach ( $this->tables as $table )
+                       $this->$table = $this->prefix . $table;
+
+               if ( defined('CUSTOM_USER_TABLE') )
+                       $this->users = CUSTOM_USER_TABLE;
+
+               if ( defined('CUSTOM_USER_META_TABLE') )
+                       $this->usermeta = CUSTOM_USER_META_TABLE;
+
+               return $old_prefix;
+       }
+
        /**
         * Selects a database using the current class's $this->dbh
         * @param string $db name
@@ -120,11 +141,14 @@ class wpdb {
         * @return string query safe string
         */
        function escape($string) {
-               return addslashes( $string ); // Disable rest for now, causing problems
+               return addslashes( $string );
+               // Disable rest for now, causing problems
+               /*
                if( !$this->dbh || version_compare( phpversion(), '4.3.0' ) == '-1' )
                        return mysql_escape_string( $string );
                else
                        return mysql_real_escape_string( $string, $this->dbh );
+               */
        }
 
        /**
@@ -155,12 +179,28 @@ class wpdb {
 
        function print_error($str = '') {
                global $EZSQL_ERROR;
+
                if (!$str) $str = mysql_error($this->dbh);
                $EZSQL_ERROR[] =
                array ('query' => $this->last_query, 'error_str' => $str);
 
+               if ( $this->suppress_errors )
+                       return false;
+
                $error_str = "WordPress database error $str for query $this->last_query";
-               error_log($error_str, 0);
+               if ( $caller = $this->get_caller() )
+                       $error_str .= " made by $caller";
+
+               $log_error = true;
+               if ( ! function_exists('error_log') )
+                       $log_error = false;
+
+               $log_file = @ini_get('error_log');
+               if ( !empty($log_file) && ('syslog' != $log_file) && !is_writable($log_file) )
+                       $log_error = false;
+
+               if ( $log_error )
+                       @error_log($error_str, 0);
 
                // Is error output turned on or not..
                if ( !$this->show_errors )
@@ -191,6 +231,12 @@ class wpdb {
                return $show;
        }
 
+       function suppress_errors( $suppress = true ) {
+               $errors = $this->suppress_errors;
+               $this->suppress_errors = $suppress;
+               return $errors;
+       }
+
        // ==================================================================
        //      Kill cached query results
 
@@ -230,10 +276,10 @@ class wpdb {
                ++$this->num_queries;
 
                if (SAVEQUERIES)
-                       $this->queries[] = array( $query, $this->timer_stop() );
+                       $this->queries[] = array( $query, $this->timer_stop(), $this->get_caller() );
 
                // If there is an error then take note of it..
-               if ( mysql_error($this->dbh) ) {
+               if ( $this->last_error = mysql_error($this->dbh) ) {
                        $this->print_error();
                        return false;
                }
@@ -270,6 +316,39 @@ class wpdb {
                return $return_val;
        }
 
+       /**
+        * Insert an array of data into a table
+        * @param string $table WARNING: not sanitized!
+        * @param array $data should not already be SQL-escaped
+        * @return mixed results of $this->query()
+        */
+       function insert($table, $data) {
+               $data = add_magic_quotes($data);
+               $fields = array_keys($data);
+               return $this->query("INSERT INTO $table (`" . implode('`,`',$fields) . "`) VALUES ('".implode("','",$data)."')");
+       }
+
+       /**
+        * Update a row in the table with an array of data
+        * @param string $table WARNING: not sanitized!
+        * @param array $data should not already be SQL-escaped
+        * @param array $where a named array of WHERE column => value relationships.  Multiple member pairs will be joined with ANDs.  WARNING: the column names are not currently sanitized!
+        * @return mixed results of $this->query()
+        */
+       function update($table, $data, $where){
+               $data = add_magic_quotes($data);
+               $bits = $wheres = array();
+               foreach ( array_keys($data) as $k )
+                       $bits[] = "`$k` = '$data[$k]'";
+
+               if ( is_array( $where ) )
+                       foreach ( $where as $c => $v )
+                               $wheres[] = "$c = '" . $this->escape( $v ) . "'";
+               else
+                       return false;
+               return $this->query( "UPDATE $table SET " . implode( ', ', $bits ) . ' WHERE ' . implode( ' AND ', $wheres ) . ' LIMIT 1' );
+       }
+
        /**
         * Get one variable from the database
         * @param string $query (can be null as well, for caching, see codex)
@@ -283,7 +362,7 @@ class wpdb {
                        $this->query($query);
 
                // Extract var out of cached results based x,y vals
-               if ( $this->last_result[$y] ) {
+               if ( !empty( $this->last_result[$y] ) ) {
                        $values = array_values(get_object_vars($this->last_result[$y]));
                }
 
@@ -340,7 +419,7 @@ class wpdb {
        /**
         * Return an entire result set from the database
         * @param string $query (can also be null to pull from the cache)
-        * @param string $output ARRAY_A | ARRAY_N | OBJECT
+        * @param string $output ARRAY_A | ARRAY_N | OBJECT_K | OBJECT
         * @return mixed results
         */
        function get_results($query = null, $output = OBJECT) {
@@ -351,22 +430,33 @@ class wpdb {
                else
                        return null;
 
-               // Send back array of objects. Each row is an object
                if ( $output == OBJECT ) {
+                       // Return an integer-keyed array of row objects
                        return $this->last_result;
+               } elseif ( $output == OBJECT_K ) {
+                       // Return an array of row objects with keys from column 1
+                       // (Duplicates are discarded)
+                       foreach ( $this->last_result as $row ) {
+                               $key = array_shift( get_object_vars( $row ) );
+                               if ( !isset( $new_array[ $key ] ) )
+                                       $new_array[ $key ] = $row;
+                       }
+                       return $new_array;
                } elseif ( $output == ARRAY_A || $output == ARRAY_N ) {
+                       // Return an integer-keyed array of...
                        if ( $this->last_result ) {
                                $i = 0;
                                foreach( $this->last_result as $row ) {
-                                       $new_array[$i] = (array) $row;
                                        if ( $output == ARRAY_N ) {
-                                               $new_array[$i] = array_values($new_array[$i]);
+                                               // ...integer-keyed row arrays
+                                               $new_array[$i] = array_values( get_object_vars( $row ) );
+                                       } else {
+                                               // ...column name-keyed row arrays
+                                               $new_array[$i] = get_object_vars( $row );
                                        }
-                                       $i++;
+                                       ++$i;
                                }
                                return $new_array;
-                       } else {
-                               return null;
                        }
                }
        }
@@ -428,6 +518,57 @@ class wpdb {
                }
                wp_die($message);
        }
+
+       /**
+        * Checks wether of not the database version is high enough to support the features WordPress uses
+        * @global $wp_version
+        */
+       function check_database_version()
+       {
+               global $wp_version;
+               // Make sure the server has MySQL 4.0
+               $mysql_version = preg_replace('|[^0-9\.]|', '', @mysql_get_server_info($this->dbh));
+               if ( version_compare($mysql_version, '4.0.0', '<') )
+                       return new WP_Error('database_version',sprintf(__('<strong>ERROR</strong>: WordPress %s requires MySQL 4.0.0 or higher'), $wp_version));
+       }
+
+       /**
+        * This function is called when WordPress is generating the table schema to determine wether or not the current database
+        * supports or needs the collation statements.
+        */
+       function supports_collation()
+       {
+               return ( version_compare(mysql_get_server_info($this->dbh), '4.1.0', '>=') );
+       }
+
+       /**
+        * Get the name of the function that called wpdb.
+        * @return string the name of the calling function
+        */
+       function get_caller() {
+               // requires PHP 4.3+
+               if ( !is_callable('debug_backtrace') )
+                       return '';
+
+               $bt = debug_backtrace();
+               $caller = '';
+
+               foreach ( $bt as $trace ) {
+                       if ( @$trace['class'] == __CLASS__ )
+                               continue;
+                       elseif ( strtolower(@$trace['function']) == 'call_user_func_array' )
+                               continue;
+                       elseif ( strtolower(@$trace['function']) == 'apply_filters' )
+                               continue;
+                       elseif ( strtolower(@$trace['function']) == 'do_action' )
+                               continue;
+
+                       $caller = $trace['function'];
+                       break;
+               }
+               return $caller;
+       }
+
 }
 
 if ( ! isset($wpdb) )