wp-admin/install-helper.php

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