wp-admin/install-helper.php

   1 <?php
   2 $wp_only_load_config = true;
   3 require_once(dirname(dirname(__FILE__)).'/wp-load.php');
   4 $debug = 0;
   5
   6 /**
   7  ** maybe_create_table()
   8  ** Create db table if it doesn't exist.
   9  ** Returns:  true if already exists or on successful completion
  10  **           false on error
  11  */
  12 if ( ! function_exists('maybe_create_table') ) :
  13 function maybe_create_table($table_name, $create_ddl) {
  14         global $wpdb;
  15         foreach ($wpdb->get_col("SHOW TABLES",0) as $table ) {
  16                 if ($table == $table_name) {
  17                         return true;
  18                 }
  19         }
  20         //didn't find it try to create it.
  21         $wpdb->query($create_ddl);
  22         // we cannot directly tell that whether this succeeded!
  23         foreach ($wpdb->get_col("SHOW TABLES",0) as $table ) {
  24                 if ($table == $table_name) {
  25                         return true;
  26                 }
  27         }
  28         return false;
  29 }
  30 endif;
  31
  32 /**
  33  ** maybe_add_column()
  34  ** Add column to db table if it doesn't exist.
  35  ** Returns:  true if already exists or on successful completion
  36  **           false on error
  37  */
  38 if ( ! function_exists('maybe_add_column') ) :
  39 function maybe_add_column($table_name, $column_name, $create_ddl) {
  40         global $wpdb, $debug;
  41         foreach ($wpdb->get_col("DESC $table_name",0) as $column ) {
  42                 if ($debug) echo("checking $column == $column_name<br />");
  43                         if ($column == $column_name) {
  44                                 return true;
  45                         }
  46         }
  47         //didn't find it try to create it.
  48         $wpdb->query($create_ddl);
  49         // we cannot directly tell that whether this succeeded!
  50         foreach ($wpdb->get_col("DESC $table_name",0) as $column ) {
  51                 if ($column == $column_name) {
  52                         return true;
  53                 }
  54         }
  55         return false;
  56 }
  57 endif;
  58
  59 /**
  60  ** maybe_drop_column()
  61  ** Drop column from db table if it exists.
  62  ** Returns:  true if it doesn't already exist or on successful drop
  63  **           false on error
  64  */
  65 function maybe_drop_column($table_name, $column_name, $drop_ddl) {
  66         global $wpdb;
  67         foreach ($wpdb->get_col("DESC $table_name",0) as $column ) {
  68                 if ($column == $column_name) {
  69                         //found it try to drop it.
  70                         $wpdb->query($drop_ddl);
  71                         // we cannot directly tell that whether this succeeded!
  72                         foreach ($wpdb->get_col("DESC $table_name",0) as $column ) {
  73                                 if ($column == $column_name) {
  74                                         return false;
  75                                 }
  76                         }
  77                 }
  78         }
  79         // else didn't find it
  80         return true;
  81 }
  82
  83
  84 /**
  85  ** check_column()
  86  ** Check column matches passed in criteria.
  87  ** Pass in null to skip checking that criteria
  88  ** Returns:  true if it matches
  89  **           false otherwise
  90  ** (case sensitive) Column names returned from DESC table are:
  91  **      Field
  92  **      Type
  93  **      Null
  94  **      Key
  95  **      Default
  96  **      Extra
  97  */
  98 function check_column($table_name, $col_name, $col_type, $is_null = null, $key = null, $default = null, $extra = null) {
  99         global $wpdb, $debug;
 100         $diffs = 0;
 101         $results = $wpdb->get_results("DESC $table_name");
 102
 103         foreach ($results as $row ) {
 104                 if ($debug > 1) print_r($row);
 105                         if ($row->Field == $col_name) {
 106                                 // got our column, check the params
 107                                 if ($debug) echo ("checking $row->Type against $col_type\n");
 108                                 if (($col_type != null) && ($row->Type != $col_type)) {
 109                                         ++$diffs;
 110                                 }
 111                                 if (($is_null != null) && ($row->Null != $is_null)) {
 112                                         ++$diffs;
 113                                 }
 114                                 if (($key != null) && ($row->Key  != $key)) {
 115                                         ++$diffs;
 116                                 }
 117                                 if (($default != null) && ($row->Default != $default)) {
 118                                         ++$diffs;
 119                                 }
 120                                 if (($extra != null) && ($row->Extra != $extra)) {
 121                                         ++$diffs;
 122                                 }
 123                                 if ($diffs > 0) {
 124                                         if ($debug) echo ("diffs = $diffs returning false\n");
 125                                         return false;
 126                                 }
 127                                 return true;
 128                         } // end if found our column
 129         }
 130         return false;
 131 }
 132
 133 /*
 134 echo "<p>testing</p>";
 135 echo "<pre>";
 136
 137 //check_column('wp_links', 'link_description', 'mediumtext');
 138 //if (check_column($wpdb->comments, 'comment_author', 'tinytext'))
 139 //    echo "ok\n";
 140 $error_count = 0;
 141 $tablename = $wpdb->links;
 142 // check the column
 143 if (!check_column($wpdb->links, 'link_description', 'varchar(255)'))
 144 {
 145         $ddl = "ALTER TABLE $wpdb->links MODIFY COLUMN link_description varchar(255) NOT NULL DEFAULT '' ";
 146         $q = $wpdb->query($ddl);
 147 }
 148 if (check_column($wpdb->links, 'link_description', 'varchar(255)')) {
 149         $res .= $tablename . ' - ok <br />';
 150 } else {
 151         $res .= 'There was a problem with ' . $tablename . '<br />';
 152         ++$error_count;
 153 }
 154 echo "</pre>";
 155 */
 156 ?>