wp-admin/includes/class-wp-importer.php

   1 <?php
   2 /**
   3  * WP_Importer base class
   4  */
   5 class WP_Importer {
   6         /**
   7          * Class Constructor
   8          *
   9          * @return void
  10          */
  11         function __construct() {}
  12
  13         function WP_Importer() {
  14                 $this->__construct();
  15         }
  16
  17         /**
  18          * Returns array with imported permalinks from WordPress database
  19          *
  20          * @param string $bid
  21          * @return array
  22          */
  23         function get_imported_posts( $importer_name, $bid ) {
  24                 global $wpdb;
  25
  26                 $hashtable = array();
  27
  28                 $limit = 100;
  29                 $offset = 0;
  30
  31                 // Grab all posts in chunks
  32                 do {
  33                         $meta_key = $importer_name . '_' . $bid . '_permalink';
  34                         $sql = $wpdb->prepare( "SELECT post_id, meta_value FROM $wpdb->postmeta WHERE meta_key = '%s' LIMIT %d,%d", $meta_key, $offset, $limit );
  35                         $results = $wpdb->get_results( $sql );
  36
  37                         // Increment offset
  38                         $offset = ( $limit + $offset );
  39
  40                         if ( !empty( $results ) ) {
  41                                 foreach ( $results as $r ) {
  42                                         // Set permalinks into array
  43                                         $hashtable[$r->meta_value] = intval( $r->post_id );
  44                                 }
  45                         }
  46                 } while ( count( $results ) == $limit );
  47
  48                 // unset to save memory
  49                 unset( $results, $r );
  50
  51                 return $hashtable;
  52         }
  53
  54         /**
  55          * Return count of imported permalinks from WordPress database
  56          *
  57          * @param string $bid
  58          * @return int
  59          */
  60         function count_imported_posts( $importer_name, $bid ) {
  61                 global $wpdb;
  62
  63                 $count = 0;
  64
  65                 // Get count of permalinks
  66                 $meta_key = $importer_name . '_' . $bid . '_permalink';
  67                 $sql = $wpdb->prepare( "SELECT COUNT( post_id ) AS cnt FROM $wpdb->postmeta WHERE meta_key = '%s'", $meta_key );
  68
  69                 $result = $wpdb->get_results( $sql );
  70
  71                 if ( !empty( $result ) )
  72                         $count = intval( $result[0]->cnt );
  73
  74                 // unset to save memory
  75                 unset( $results );
  76
  77                 return $count;
  78         }
  79
  80         /**
  81          * Set array with imported comments from WordPress database
  82          *
  83          * @param string $bid
  84          * @return array
  85          */
  86         function get_imported_comments( $bid ) {
  87                 global $wpdb;
  88
  89                 $hashtable = array();
  90
  91                 $limit = 100;
  92                 $offset = 0;
  93
  94                 // Grab all comments in chunks
  95                 do {
  96                         $sql = $wpdb->prepare( "SELECT comment_ID, comment_agent FROM $wpdb->comments LIMIT %d,%d", $offset, $limit );
  97                         $results = $wpdb->get_results( $sql );
  98
  99                         // Increment offset
 100                         $offset = ( $limit + $offset );
 101
 102                         if ( !empty( $results ) ) {
 103                                 foreach ( $results as $r ) {
 104                                         // Explode comment_agent key
 105                                         list ( $ca_bid, $source_comment_id ) = explode( '-', $r->comment_agent );
 106                                         $source_comment_id = intval( $source_comment_id );
 107
 108                                         // Check if this comment came from this blog
 109                                         if ( $bid == $ca_bid ) {
 110                                                 $hashtable[$source_comment_id] = intval( $r->comment_ID );
 111                                         }
 112                                 }
 113                         }
 114                 } while ( count( $results ) == $limit );
 115
 116                 // unset to save memory
 117                 unset( $results, $r );
 118
 119                 return $hashtable;
 120         }
 121
 122         function set_blog( $blog_id ) {
 123                 if ( is_numeric( $blog_id ) ) {
 124                         $blog_id = (int) $blog_id;
 125                 } else {
 126                         $blog = 'http://' . preg_replace( '#^https?://#', '', $blog_id );
 127                         if ( ( !$parsed = parse_url( $blog ) ) || empty( $parsed['host'] ) ) {
 128                                 fwrite( STDERR, "Error: can not determine blog_id from $blog_id\n" );
 129                                 exit();
 130                         }
 131                         if ( empty( $parsed['path'] ) )
 132                                 $parsed['path'] = '/';
 133                         $blog = get_blog_details( array( 'domain' => $parsed['host'], 'path' => $parsed['path'] ) );
 134                         if ( !$blog ) {
 135                                 fwrite( STDERR, "Error: Could not find blog\n" );
 136                                 exit();
 137                         }
 138                         $blog_id = (int) $blog->blog_id;
 139                         // Restore global $current_blog
 140                         global $current_blog;
 141                         $current_blog = $blog;
 142                 }
 143
 144                 if ( function_exists( 'is_multisite' ) ) {
 145                         if ( is_multisite() )
 146                                 switch_to_blog( $blog_id );
 147                 }
 148
 149                 return $blog_id;
 150         }
 151
 152         function set_user( $user_id ) {
 153                 if ( is_numeric( $user_id ) ) {
 154                         $user_id = (int) $user_id;
 155                 } else {
 156                         $user_id = (int) username_exists( $user_id );
 157                 }
 158
 159                 if ( !$user_id || !wp_set_current_user( $user_id ) ) {
 160                         fwrite( STDERR, "Error: can not find user\n" );
 161                         exit();
 162                 }
 163
 164                 return $user_id;
 165         }
 166
 167         /**
 168          * Sort by strlen, longest string first
 169          *
 170          * @param string $a
 171          * @param string $b
 172          * @return int
 173          */
 174         function cmpr_strlen( $a, $b ) {
 175                 return strlen( $b ) - strlen( $a );
 176         }
 177
 178         /**
 179          * GET URL
 180          *
 181          * @param string $url
 182          * @param string $username
 183          * @param string $password
 184          * @param bool $head
 185          * @return array
 186          */
 187         function get_page( $url, $username = '', $password = '', $head = false ) {
 188                 // Increase the timeout
 189                 add_filter( 'http_request_timeout', array( &$this, 'bump_request_timeout' ) );
 190
 191                 $headers = array();
 192                 $args = array();
 193                 if ( true === $head )
 194                         $args['method'] = 'HEAD';
 195                 if ( !empty( $username ) && !empty( $password ) )
 196                         $headers['Authorization'] = 'Basic ' . base64_encode( "$username:$password" );
 197
 198                 $args['headers'] = $headers;
 199
 200                 return wp_remote_request( $url, $args );
 201         }
 202
 203         /**
 204          * Bump up the request timeout for http requests
 205          *
 206          * @param int $val
 207          * @return int
 208          */
 209         function bump_request_timeout( $val ) {
 210                 return 60;
 211         }
 212
 213         /**
 214          * Check if user has exceeded disk quota
 215          *
 216          * @return bool
 217          */
 218         function is_user_over_quota() {
 219                 global $current_blog;
 220
 221                 if ( function_exists( 'upload_is_user_over_quota' ) ) {
 222                         if ( upload_is_user_over_quota( 1 ) ) {
 223                                 echo "Sorry, you have used your upload quota.\n";
 224                                 return true;
 225                         }
 226                 }
 227
 228                 return false;
 229         }
 230
 231         /**
 232          * Replace newlines, tabs, and multiple spaces with a single space
 233          *
 234          * @param string $string
 235          * @return string
 236          */
 237         function min_whitespace( $string ) {
 238                 return preg_replace( '|[\r\n\t ]+|', ' ', $string );
 239         }
 240
 241         /**
 242          * Reset global variables that grow out of control during imports
 243          *
 244          * @return void
 245          */
 246         function stop_the_insanity() {
 247                 global $wpdb, $wp_actions;
 248                 // Or define( 'WP_IMPORTING', true );
 249                 $wpdb->queries = array();
 250                 // Reset $wp_actions to keep it from growing out of control
 251                 $wp_actions = array();
 252         }
 253 }
 254
 255 /**
 256  * Returns value of command line params.
 257  * Exits when a required param is not set.
 258  *
 259  * @param string $param
 260  * @param bool $required
 261  * @return mixed
 262  */
 263 function get_cli_args( $param, $required = false ) {
 264         $args = $_SERVER['argv'];
 265
 266         $out = array();
 267
 268         $last_arg = null;
 269         $return = null;
 270
 271         $il = sizeof( $args );
 272
 273         for ( $i = 1, $il; $i < $il; $i++ ) {
 274                 if ( (bool) preg_match( "/^--(.+)/", $args[$i], $match ) ) {
 275                         $parts = explode( "=", $match[1] );
 276                         $key = preg_replace( "/[^a-z0-9]+/", "", $parts[0] );
 277
 278                         if ( isset( $parts[1] ) ) {
 279                                 $out[$key] = $parts[1];
 280                         } else {
 281                                 $out[$key] = true;
 282                         }
 283
 284                         $last_arg = $key;
 285                 } else if ( (bool) preg_match( "/^-([a-zA-Z0-9]+)/", $args[$i], $match ) ) {
 286                         for ( $j = 0, $jl = strlen( $match[1] ); $j < $jl; $j++ ) {
 287                                 $key = $match[1]{$j};
 288                                 $out[$key] = true;
 289                         }
 290
 291                         $last_arg = $key;
 292                 } else if ( $last_arg !== null ) {
 293                         $out[$last_arg] = $args[$i];
 294                 }
 295         }
 296
 297         // Check array for specified param
 298         if ( isset( $out[$param] ) ) {
 299                 // Set return value
 300                 $return = $out[$param];
 301         }
 302
 303         // Check for missing required param
 304         if ( !isset( $out[$param] ) && $required ) {
 305                 // Display message and exit
 306                 echo "\"$param\" parameter is required but was not specified\n";
 307                 exit();
 308         }
 309
 310         return $return;
 311 }