| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | <?php | 
					
						
							|  |  |  | /** | 
					
						
							|  |  |  |  * Plugins may load this file to gain access to special helper functions for | 
					
						
							|  |  |  |  * plugin installation. This file is not included by WordPress and it is | 
					
						
							|  |  |  |  * recommended, to prevent fatal errors, that this file is included using | 
					
						
							|  |  |  |  * require_once(). | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  * These functions are not optimized for speed, but they should only be used | 
					
						
							|  |  |  |  * once in a while, so speed shouldn't be a concern. If it is and you are | 
					
						
							|  |  |  |  * needing to use these functions a lot, you might experience time outs. If you | 
					
						
							|  |  |  |  * do, then it is advised to just write the SQL code yourself. | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  *     check_column( 'wp_links', 'link_description', 'mediumtext' ); | 
					
						
							|  |  |  |  *     if ( check_column( $wpdb->comments, 'comment_author', 'tinytext' ) ) { | 
					
						
							|  |  |  |  *         echo "ok\n"; | 
					
						
							|  |  |  |  *     } | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  *     $error_count = 0; | 
					
						
							|  |  |  |  *     $tablename = $wpdb->links; | 
					
						
							|  |  |  |  *     // Check the column.
 | 
					
						
							|  |  |  |  *     if ( ! check_column($wpdb->links, 'link_description', 'varchar( 255 )' ) ) { | 
					
						
							|  |  |  |  *         $ddl = "ALTER TABLE $wpdb->links MODIFY COLUMN link_description varchar(255) NOT NULL DEFAULT '' "; | 
					
						
							|  |  |  |  *         $q = $wpdb->query( $ddl ); | 
					
						
							|  |  |  |  *     } | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  *     if ( check_column( $wpdb->links, 'link_description', 'varchar( 255 )' ) ) { | 
					
						
							|  |  |  |  *         $res .= $tablename . ' - ok <br />'; | 
					
						
							|  |  |  |  *     } else { | 
					
						
							|  |  |  |  *         $res .= 'There was a problem with ' . $tablename . '<br />'; | 
					
						
							|  |  |  |  *         ++$error_count; | 
					
						
							|  |  |  |  *     } | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  * @package WordPress | 
					
						
							|  |  |  |  * @subpackage Plugin | 
					
						
							|  |  |  |  */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /** Load WordPress Bootstrap */ | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | require_once( dirname( dirname( __FILE__ ) ) . '/wp-load.php' ); | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | if ( ! function_exists( 'maybe_create_table' ) ) : | 
					
						
							|  |  |  | 	/** | 
					
						
							|  |  |  | 	 * Create database table, if it doesn't already exist. | 
					
						
							|  |  |  | 	 * | 
					
						
							|  |  |  | 	 * @since 1.0.0 | 
					
						
							|  |  |  | 	 * | 
					
						
							|  |  |  | 	 * @global wpdb $wpdb WordPress database abstraction object. | 
					
						
							|  |  |  | 	 * | 
					
						
							|  |  |  | 	 * @param string $table_name Database table name. | 
					
						
							|  |  |  | 	 * @param string $create_ddl Create database table SQL. | 
					
						
							|  |  |  | 	 * @return bool False on error, true if already exists or success. | 
					
						
							|  |  |  | 	 */ | 
					
						
							|  |  |  | 	function maybe_create_table( $table_name, $create_ddl ) { | 
					
						
							|  |  |  | 		global $wpdb; | 
					
						
							|  |  |  | 		foreach ( $wpdb->get_col( 'SHOW TABLES', 0 ) as $table ) { | 
					
						
							|  |  |  | 			if ( $table == $table_name ) { | 
					
						
							|  |  |  | 				return true; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 		} | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 		// Didn't find it, so try to create it.
 | 
					
						
							|  |  |  | 		$wpdb->query( $create_ddl ); | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 		// We cannot directly tell that whether this succeeded!
 | 
					
						
							|  |  |  | 		foreach ( $wpdb->get_col( 'SHOW TABLES', 0 ) as $table ) { | 
					
						
							|  |  |  | 			if ( $table == $table_name ) { | 
					
						
							|  |  |  | 				return true; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 		} | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 		return false; | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 	} | 
					
						
							|  |  |  | endif; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | if ( ! function_exists( 'maybe_add_column' ) ) : | 
					
						
							|  |  |  | 	/** | 
					
						
							|  |  |  | 	 * Add column to database table, if column doesn't already exist in table. | 
					
						
							|  |  |  | 	 * | 
					
						
							|  |  |  | 	 * @since 1.0.0 | 
					
						
							|  |  |  | 	 * | 
					
						
							|  |  |  | 	 * @global wpdb $wpdb WordPress database abstraction object. | 
					
						
							|  |  |  | 	 * | 
					
						
							|  |  |  | 	 * @param string $table_name Database table name | 
					
						
							|  |  |  | 	 * @param string $column_name Table column name | 
					
						
							|  |  |  | 	 * @param string $create_ddl SQL to add column to table. | 
					
						
							|  |  |  | 	 * @return bool False on failure. True, if already exists or was successful. | 
					
						
							|  |  |  | 	 */ | 
					
						
							|  |  |  | 	function maybe_add_column( $table_name, $column_name, $create_ddl ) { | 
					
						
							|  |  |  | 		global $wpdb; | 
					
						
							|  |  |  | 		foreach ( $wpdb->get_col( "DESC $table_name", 0 ) as $column ) { | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 			if ( $column == $column_name ) { | 
					
						
							|  |  |  | 				return true; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 		} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 		// Didn't find it, so try to create it.
 | 
					
						
							|  |  |  | 		$wpdb->query( $create_ddl ); | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 		// We cannot directly tell that whether this succeeded!
 | 
					
						
							|  |  |  | 		foreach ( $wpdb->get_col( "DESC $table_name", 0 ) as $column ) { | 
					
						
							|  |  |  | 			if ( $column == $column_name ) { | 
					
						
							|  |  |  | 				return true; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 		} | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 		return false; | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 	} | 
					
						
							|  |  |  | endif; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /** | 
					
						
							|  |  |  |  * Drop column from database table, if it exists. | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  * @since 1.0.0 | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  * @global wpdb $wpdb WordPress database abstraction object. | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  * @param string $table_name Table name | 
					
						
							|  |  |  |  * @param string $column_name Column name | 
					
						
							|  |  |  |  * @param string $drop_ddl SQL statement to drop column. | 
					
						
							|  |  |  |  * @return bool False on failure, true on success or doesn't exist. | 
					
						
							|  |  |  |  */ | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | function maybe_drop_column( $table_name, $column_name, $drop_ddl ) { | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 	global $wpdb; | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 	foreach ( $wpdb->get_col( "DESC $table_name", 0 ) as $column ) { | 
					
						
							|  |  |  | 		if ( $column == $column_name ) { | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 			// Found it, so try to drop it.
 | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 			$wpdb->query( $drop_ddl ); | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 			// We cannot directly tell that whether this succeeded!
 | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 			foreach ( $wpdb->get_col( "DESC $table_name", 0 ) as $column ) { | 
					
						
							|  |  |  | 				if ( $column == $column_name ) { | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 					return false; | 
					
						
							|  |  |  | 				} | 
					
						
							|  |  |  | 			} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	// Else didn't find it.
 | 
					
						
							|  |  |  | 	return true; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /** | 
					
						
							|  |  |  |  * Check column matches criteria. | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  * Uses the SQL DESC for retrieving the table info for the column. It will help | 
					
						
							|  |  |  |  * understand the parameters, if you do more research on what column information | 
					
						
							|  |  |  |  * is returned by the SQL statement. Pass in null to skip checking that | 
					
						
							|  |  |  |  * criteria. | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  * Column names returned from DESC table are case sensitive and are listed: | 
					
						
							|  |  |  |  *      Field | 
					
						
							|  |  |  |  *      Type | 
					
						
							|  |  |  |  *      Null | 
					
						
							|  |  |  |  *      Key | 
					
						
							|  |  |  |  *      Default | 
					
						
							|  |  |  |  *      Extra | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  * @since 1.0.0 | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  * @global wpdb $wpdb WordPress database abstraction object. | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  * @param string $table_name Table name | 
					
						
							|  |  |  |  * @param string $col_name   Column name | 
					
						
							|  |  |  |  * @param string $col_type   Column type | 
					
						
							|  |  |  |  * @param bool   $is_null    Optional. Check is null. | 
					
						
							|  |  |  |  * @param mixed  $key        Optional. Key info. | 
					
						
							|  |  |  |  * @param mixed  $default    Optional. Default value. | 
					
						
							|  |  |  |  * @param mixed  $extra      Optional. Extra value. | 
					
						
							|  |  |  |  * @return bool True, if matches. False, if not matching. | 
					
						
							|  |  |  |  */ | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | function check_column( $table_name, $col_name, $col_type, $is_null = null, $key = null, $default = null, $extra = null ) { | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 	global $wpdb; | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 	$diffs   = 0; | 
					
						
							|  |  |  | 	$results = $wpdb->get_results( "DESC $table_name" ); | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 	foreach ( $results as $row ) { | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 		if ( $row->Field == $col_name ) { | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 			// Got our column, check the params.
 | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 			if ( ( $col_type != null ) && ( $row->Type != $col_type ) ) { | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 				++$diffs; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 			if ( ( $is_null != null ) && ( $row->Null != $is_null ) ) { | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 				++$diffs; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 			if ( ( $key != null ) && ( $row->Key != $key ) ) { | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 				++$diffs; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 			if ( ( $default != null ) && ( $row->Default != $default ) ) { | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 				++$diffs; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 			if ( ( $extra != null ) && ( $row->Extra != $extra ) ) { | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 				++$diffs; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2019-04-16 20:56:22 +01:00
										 |  |  | 			if ( $diffs > 0 ) { | 
					
						
							| 
									
										
										
										
											2019-03-12 09:27:46 +00:00
										 |  |  | 				return false; | 
					
						
							|  |  |  | 			} | 
					
						
							|  |  |  | 			return true; | 
					
						
							|  |  |  | 		} // end if found our column
 | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return false; | 
					
						
							|  |  |  | } |