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