WordPress/wp-includes/wp-db.php

<?php
	// ==================================================================
	//  Author: Justin Vincent (justin@visunet.ie)
	//	Web: 	http://php.justinvincent.com
	//	Name: 	ezSQL
	// 	Desc: 	Class to make it very easy to deal with mySQL database connections.
	//  WordPress is using this class to make the code cleaner and faster.
	//  We highly recommend it.
	//  We have modified the HTML it returns slightly.

	define('EZSQL_VERSION', '1.21');
	define('OBJECT', 'OBJECT', true);
	define('ARRAY_A', 'ARRAY_A', true);
	define('ARRAY_N', 'ARRAY_N', true);
	define('SAVEQUERIES', true);

	//	The Main Class, renamed to avoid conflicts.

	class wpdb {

		var $debug_called;
		var $vardump_called;
		var $show_errors = true;
		var $querycount;

		// ==================================================================
		//	DB Constructor - connects to the server and selects a database

		function wpdb($dbuser, $dbpassword, $dbname, $dbhost) {
			$this->dbh = @mysql_connect($dbhost,$dbuser,$dbpassword);
			if ( ! $this->dbh )	{
				die("<div>
				<p><strong>Error establishing a database connection!</strong> This probably means that the connection information in youn <code>wp-config.php</code> file is incorrect. Double check it and try again.</p>
				<ul>
				<li>Are you sure you have the correct user/password?</li>
				<li>Are you sure that you have typed the correct hostname?</li>
				<li>Are you sure that the database server is running?</li>
				</ul>
				<p><a href='http://wordpress.org/support/'>WordPress Support Forums</a></p>
				</div>");
			}

			$this->select($dbname);
			$this->querycount = 0;
		}

		// ==================================================================
		//	Select a DB (if another one needs to be selected)

		function select($db) {
			if ( !@mysql_select_db($db,$this->dbh)) {
				die("
				<p>We're having a little trouble selecting the proper database for WordPress.</p>
				<ul>
				<li>Are you sure it exists?</li>
				<li>Your database name is currently specified as <code>" . DB_NAME ."</code>. Is this correct?</li>
				<li>On some systems the name of your database is prefixed with your username, so it would be like username_wordpress. Could that be the problem?</li>
				</ul>
				<p><a href='http://wordpress.org/support/'>WordPress Support Forums</a></p>");
			}
		}

		// ====================================================================
		//	Format a string correctly for safe insert under all PHP conditions
		
		function escape($str) {
			return mysql_escape_string(stripslashes($str));				
		}

		// ==================================================================
		//	Print SQL/DB error.

		function print_error($str = '') {
			// All errors go to the global error array $EZSQL_ERROR..
			global $EZSQL_ERROR;

			// If no special error string then use mysql default..
			if ( !$str ) $str = mysql_error();
			
			// Log this error to the global array..
			$EZSQL_ERROR[] = array 
							(
								'query' => $this->last_query,
								'error_str'  => $str
							);

			// Is error output turned on or not..
			if ( $this->show_errors ) {
				// If there is an error then take note of it
				print "<div id='error'>
				<p><strong>SQL/DB Error:</strong><br />
				[<span style='color: #007;'>$str</span>]<br />
				<code>$this->last_query</code></p>
				</div>";
			} else {
				return false;	
			}

		}

		// ==================================================================
		//	Turn error handling on or off..

		function show_errors() {
			$this->show_errors = true;
		}
		
		function hide_errors() {
			$this->show_errors = false;
		}

		// ==================================================================
		//	Kill cached query results

		function flush() {
			// Get rid of these
			$this->last_result = null;
			$this->col_info = null;
			$this->last_query = null;

		}

		// ==================================================================
		//	Basic Query	- see docs for more detail

		function query($query) {

			// Flush cached values..
			$this->flush();

			// Log how the function was called
			$this->func_call = "\$db->query(\"$query\")";

			// Keep track of the last query for debug..
			$this->last_query = $query;

			// Perform the query via std mysql_query function..
			$this->result = mysql_query($query, $this->dbh);
			++$this->querycount;
			if (SAVEQUERIES) {
				$this->savedqueries[] = $query;
			}

			// If there was an insert, delete or update see how many rows were affected
			// (Also, If there there was an insert take note of the insert_id
			$query_type = array('insert','delete','update','replace');

			// loop through the above array
			foreach ( $query_type as $word ) {
				// This is true if the query starts with insert, delete or update
				if ( preg_match("/^\\s*$word /i",$query) ) {
					$this->rows_affected = mysql_affected_rows();
					// This gets the insert ID
					if ( $word == 'insert' || $word == 'replace' ) {
						$this->insert_id = mysql_insert_id($this->dbh);
					}
					$this->result = false;
				}
			}
   
			if ( mysql_error() ) { // If there is an error then take note of it..
				$this->print_error();
			} else {
				// In other words if this was a select statement..
				if ( $this->result ) {
					// =======================================================
					// Take note of column info

					$i=0;
					while ($i < @mysql_num_fields($this->result)) {
						$this->col_info[$i] = @mysql_fetch_field($this->result);
						$i++;
					}

					// =======================================================
					// Store Query Results

					$i=0;
					while ( $row = @mysql_fetch_object($this->result) ) {
						// Store relults as an objects within main array
						$this->last_result[$i] = $row;
						$i++;
					}

					// Log number of rows the query returned
					$this->num_rows = $i;

					@mysql_free_result($this->result);


					// If there were results then return true for $db->query
					if ( $i ) {
						return true;
					} else {
						return false;
					}

				} else { // Update insert etc. was good..
					return true;
				}
			}
		}

		// ==================================================================
		//	Get one variable from the DB - see docs for more detail

		function get_var($query=null, $x=0, $y=0) {

			// Log how the function was called
			$this->func_call = "\$db->get_var(\"$query\",$x,$y)";

			// If there is a query then perform it if not then use cached results..
			if ( $query ) {
				$this->query($query);
			}

			// Extract var out of cached results based x,y vals
			if ( $this->last_result[$y] ) {
				$values = array_values(get_object_vars($this->last_result[$y]));
			}

			// If there is a value return it else return null
			return (isset($values[$x]) && $values[$x]!=='') ? $values[$x] : null;
		}

		// ==================================================================
		//	Get one row from the DB - see docs for more detail

		function get_row($query=null, $output=OBJECT, $y=0) {

			// Log how the function was called
			$this->func_call = "\$db->get_row(\"$query\",$output,$y)";

			// If there is a query then perform it if not then use cached results..
			if ( $query ) {
				$this->query($query);
			}

			// If the output is an object then return object using the row offset..
			if ( $output == OBJECT ) {
				return $this->last_result[$y]?$this->last_result[$y]:null;
			}
			// If the output is an associative array then return row as such..
			elseif ( $output == ARRAY_A ) {
				return $this->last_result[$y]?get_object_vars($this->last_result[$y]):null;
			}
			// If the output is an numerical array then return row as such..
			elseif ( $output == ARRAY_N ) {
				return $this->last_result[$y]?array_values(get_object_vars($this->last_result[$y])):null;
			}
			// If invalid output type was specified..
			else {
				$this->print_error(" \$db->get_row(string query, output type, int offset) -- Output type must be one of: OBJECT, ARRAY_A, ARRAY_N");
			}

		}

		// ==================================================================
		//	Function to get 1 column from the cached result set based in X index
		// se docs for usage and info

		function get_col($query=null,$x=0) {
			// If there is a query then perform it if not then use cached results..
			if ( $query ) {
				$this->query($query);
			}
			// Extract the column values
			for ( $i=0; $i < count($this->last_result); $i++ ) {
				$new_array[$i] = $this->get_var(null,$x,$i);
			}
			return $new_array;
		}

		// ==================================================================
		// Return the the query as a result set - see docs for more details

		function get_results($query=null, $output = OBJECT){

			// Log how the function was called
			$this->func_call = "\$db->get_results(\"$query\", $output)";

			// If there is a query then perform it if not then use cached results..
			if ( $query ) {
				$this->query($query);
			}

			// Send back array of objects. Each row is an object
			if ( $output == OBJECT ) {
				return $this->last_result;
			} elseif ( $output == ARRAY_A || $output == ARRAY_N ) {
				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]);
						}

						$i++;
					}

					return $new_array;
				} else {
					return null;
				}
			}
		}


		// ==================================================================
		// Function to get column meta data info pertaining to the last query
		// see docs for more info and usage

		function get_col_info($info_type='name', $col_offset=-1) {

			if ( $this->col_info ) {
				if ( $col_offset == -1 ) {
					$i=0;
					foreach($this->col_info as $col ) {
						$new_array[$i] = $col->{$info_type};
						$i++;
					}
					return $new_array;
				} else {
					return $this->col_info[$col_offset]->{$info_type};
				}

			}

		}

	}

$wpdb = new wpdb(DB_USER, DB_PASSWORD, DB_NAME, DB_HOST);

?>