[Summary view] [Print] [Text view]

   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  }