331 lines
7.3 KiB
PHP
331 lines
7.3 KiB
PHP
<?php
|
|
/**
|
|
* WP_Importer base class
|
|
*/
|
|
class WP_Importer {
|
|
/**
|
|
* Class Constructor
|
|
*/
|
|
public function __construct() {}
|
|
|
|
/**
|
|
* Returns array with imported permalinks from WordPress database
|
|
*
|
|
* @global wpdb $wpdb WordPress database abstraction object.
|
|
*
|
|
* @param string $importer_name
|
|
* @param string $bid
|
|
* @return array
|
|
*/
|
|
public function get_imported_posts( $importer_name, $bid ) {
|
|
global $wpdb;
|
|
|
|
$hashtable = array();
|
|
|
|
$limit = 100;
|
|
$offset = 0;
|
|
|
|
// Grab all posts in chunks
|
|
do {
|
|
$meta_key = $importer_name . '_' . $bid . '_permalink';
|
|
$sql = $wpdb->prepare( "SELECT post_id, meta_value FROM $wpdb->postmeta WHERE meta_key = %s LIMIT %d,%d", $meta_key, $offset, $limit );
|
|
$results = $wpdb->get_results( $sql );
|
|
|
|
// Increment offset
|
|
$offset = ( $limit + $offset );
|
|
|
|
if ( ! empty( $results ) ) {
|
|
foreach ( $results as $r ) {
|
|
// Set permalinks into array
|
|
$hashtable[ $r->meta_value ] = intval( $r->post_id );
|
|
}
|
|
}
|
|
} while ( count( $results ) == $limit );
|
|
|
|
// Unset to save memory.
|
|
unset( $results, $r );
|
|
|
|
return $hashtable;
|
|
}
|
|
|
|
/**
|
|
* Return count of imported permalinks from WordPress database
|
|
*
|
|
* @global wpdb $wpdb WordPress database abstraction object.
|
|
*
|
|
* @param string $importer_name
|
|
* @param string $bid
|
|
* @return int
|
|
*/
|
|
public function count_imported_posts( $importer_name, $bid ) {
|
|
global $wpdb;
|
|
|
|
$count = 0;
|
|
|
|
// Get count of permalinks
|
|
$meta_key = $importer_name . '_' . $bid . '_permalink';
|
|
$sql = $wpdb->prepare( "SELECT COUNT( post_id ) AS cnt FROM $wpdb->postmeta WHERE meta_key = '%s'", $meta_key );
|
|
|
|
$result = $wpdb->get_results( $sql );
|
|
|
|
if ( ! empty( $result ) ) {
|
|
$count = intval( $result[0]->cnt );
|
|
}
|
|
|
|
// Unset to save memory.
|
|
unset( $results );
|
|
|
|
return $count;
|
|
}
|
|
|
|
/**
|
|
* Set array with imported comments from WordPress database
|
|
*
|
|
* @global wpdb $wpdb WordPress database abstraction object.
|
|
*
|
|
* @param string $bid
|
|
* @return array
|
|
*/
|
|
public function get_imported_comments( $bid ) {
|
|
global $wpdb;
|
|
|
|
$hashtable = array();
|
|
|
|
$limit = 100;
|
|
$offset = 0;
|
|
|
|
// Grab all comments in chunks
|
|
do {
|
|
$sql = $wpdb->prepare( "SELECT comment_ID, comment_agent FROM $wpdb->comments LIMIT %d,%d", $offset, $limit );
|
|
$results = $wpdb->get_results( $sql );
|
|
|
|
// Increment offset
|
|
$offset = ( $limit + $offset );
|
|
|
|
if ( ! empty( $results ) ) {
|
|
foreach ( $results as $r ) {
|
|
// Explode comment_agent key
|
|
list ( $ca_bid, $source_comment_id ) = explode( '-', $r->comment_agent );
|
|
$source_comment_id = intval( $source_comment_id );
|
|
|
|
// Check if this comment came from this blog
|
|
if ( $bid == $ca_bid ) {
|
|
$hashtable[ $source_comment_id ] = intval( $r->comment_ID );
|
|
}
|
|
}
|
|
}
|
|
} while ( count( $results ) == $limit );
|
|
|
|
// Unset to save memory.
|
|
unset( $results, $r );
|
|
|
|
return $hashtable;
|
|
}
|
|
|
|
/**
|
|
* @param int $blog_id
|
|
* @return int|void
|
|
*/
|
|
public function set_blog( $blog_id ) {
|
|
if ( is_numeric( $blog_id ) ) {
|
|
$blog_id = (int) $blog_id;
|
|
} else {
|
|
$blog = 'http://' . preg_replace( '#^https?://#', '', $blog_id );
|
|
if ( ( ! $parsed = parse_url( $blog ) ) || empty( $parsed['host'] ) ) {
|
|
fwrite( STDERR, "Error: can not determine blog_id from $blog_id\n" );
|
|
exit();
|
|
}
|
|
if ( empty( $parsed['path'] ) ) {
|
|
$parsed['path'] = '/';
|
|
}
|
|
$blogs = get_sites(
|
|
array(
|
|
'domain' => $parsed['host'],
|
|
'number' => 1,
|
|
'path' => $parsed['path'],
|
|
)
|
|
);
|
|
if ( ! $blogs ) {
|
|
fwrite( STDERR, "Error: Could not find blog\n" );
|
|
exit();
|
|
}
|
|
$blog = array_shift( $blogs );
|
|
$blog_id = (int) $blog->blog_id;
|
|
}
|
|
|
|
if ( function_exists( 'is_multisite' ) ) {
|
|
if ( is_multisite() ) {
|
|
switch_to_blog( $blog_id );
|
|
}
|
|
}
|
|
|
|
return $blog_id;
|
|
}
|
|
|
|
/**
|
|
* @param int $user_id
|
|
* @return int|void
|
|
*/
|
|
public function set_user( $user_id ) {
|
|
if ( is_numeric( $user_id ) ) {
|
|
$user_id = (int) $user_id;
|
|
} else {
|
|
$user_id = (int) username_exists( $user_id );
|
|
}
|
|
|
|
if ( ! $user_id || ! wp_set_current_user( $user_id ) ) {
|
|
fwrite( STDERR, "Error: can not find user\n" );
|
|
exit();
|
|
}
|
|
|
|
return $user_id;
|
|
}
|
|
|
|
/**
|
|
* Sort by strlen, longest string first
|
|
*
|
|
* @param string $a
|
|
* @param string $b
|
|
* @return int
|
|
*/
|
|
public function cmpr_strlen( $a, $b ) {
|
|
return strlen( $b ) - strlen( $a );
|
|
}
|
|
|
|
/**
|
|
* GET URL
|
|
*
|
|
* @param string $url
|
|
* @param string $username
|
|
* @param string $password
|
|
* @param bool $head
|
|
* @return array
|
|
*/
|
|
public function get_page( $url, $username = '', $password = '', $head = false ) {
|
|
// Increase the timeout
|
|
add_filter( 'http_request_timeout', array( $this, 'bump_request_timeout' ) );
|
|
|
|
$headers = array();
|
|
$args = array();
|
|
if ( true === $head ) {
|
|
$args['method'] = 'HEAD';
|
|
}
|
|
if ( ! empty( $username ) && ! empty( $password ) ) {
|
|
$headers['Authorization'] = 'Basic ' . base64_encode( "$username:$password" );
|
|
}
|
|
|
|
$args['headers'] = $headers;
|
|
|
|
return wp_safe_remote_request( $url, $args );
|
|
}
|
|
|
|
/**
|
|
* Bump up the request timeout for http requests
|
|
*
|
|
* @param int $val
|
|
* @return int
|
|
*/
|
|
public function bump_request_timeout( $val ) {
|
|
return 60;
|
|
}
|
|
|
|
/**
|
|
* Check if user has exceeded disk quota
|
|
*
|
|
* @return bool
|
|
*/
|
|
public function is_user_over_quota() {
|
|
if ( function_exists( 'upload_is_user_over_quota' ) ) {
|
|
if ( upload_is_user_over_quota() ) {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* Replace newlines, tabs, and multiple spaces with a single space
|
|
*
|
|
* @param string $string
|
|
* @return string
|
|
*/
|
|
public function min_whitespace( $string ) {
|
|
return preg_replace( '|[\r\n\t ]+|', ' ', $string );
|
|
}
|
|
|
|
/**
|
|
* Resets global variables that grow out of control during imports.
|
|
*
|
|
* @since 3.0.0
|
|
*
|
|
* @global wpdb $wpdb WordPress database abstraction object.
|
|
* @global array $wp_actions
|
|
*/
|
|
public function stop_the_insanity() {
|
|
global $wpdb, $wp_actions;
|
|
// Or define( 'WP_IMPORTING', true );
|
|
$wpdb->queries = array();
|
|
// Reset $wp_actions to keep it from growing out of control
|
|
$wp_actions = array();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Returns value of command line params.
|
|
* Exits when a required param is not set.
|
|
*
|
|
* @param string $param
|
|
* @param bool $required
|
|
* @return mixed
|
|
*/
|
|
function get_cli_args( $param, $required = false ) {
|
|
$args = $_SERVER['argv'];
|
|
|
|
$out = array();
|
|
|
|
$last_arg = null;
|
|
$return = null;
|
|
|
|
$il = sizeof( $args );
|
|
|
|
for ( $i = 1, $il; $i < $il; $i++ ) {
|
|
if ( (bool) preg_match( '/^--(.+)/', $args[ $i ], $match ) ) {
|
|
$parts = explode( '=', $match[1] );
|
|
$key = preg_replace( '/[^a-z0-9]+/', '', $parts[0] );
|
|
|
|
if ( isset( $parts[1] ) ) {
|
|
$out[ $key ] = $parts[1];
|
|
} else {
|
|
$out[ $key ] = true;
|
|
}
|
|
|
|
$last_arg = $key;
|
|
} elseif ( (bool) preg_match( '/^-([a-zA-Z0-9]+)/', $args[ $i ], $match ) ) {
|
|
for ( $j = 0, $jl = strlen( $match[1] ); $j < $jl; $j++ ) {
|
|
$key = $match[1]{$j};
|
|
$out[ $key ] = true;
|
|
}
|
|
|
|
$last_arg = $key;
|
|
} elseif ( $last_arg !== null ) {
|
|
$out[ $last_arg ] = $args[ $i ];
|
|
}
|
|
}
|
|
|
|
// Check array for specified param
|
|
if ( isset( $out[ $param ] ) ) {
|
|
// Set return value
|
|
$return = $out[ $param ];
|
|
}
|
|
|
|
// Check for missing required param
|
|
if ( ! isset( $out[ $param ] ) && $required ) {
|
|
// Display message and exit
|
|
echo "\"$param\" parameter is required but was not specified\n";
|
|
exit();
|
|
}
|
|
|
|
return $return;
|
|
}
|