<?php

[0] Fix | Delete

/**

[1] Fix | Delete

* WP_Importer base class

[2] Fix | Delete

*/

[3] Fix | Delete

#[AllowDynamicProperties]

[4] Fix | Delete

class WP_Importer {

[5] Fix | Delete

/**

[6] Fix | Delete

* Class Constructor

[7] Fix | Delete

*/

[8] Fix | Delete

public function __construct() {}

[9] Fix | Delete

[10] Fix | Delete

/**

[11] Fix | Delete

* Returns array with imported permalinks from WordPress database.

[12] Fix | Delete

*

[13] Fix | Delete

* @global wpdb $wpdb WordPress database abstraction object.

[14] Fix | Delete

*

[15] Fix | Delete

* @param string $importer_name

[16] Fix | Delete

* @param string $blog_id

[17] Fix | Delete

* @return array

[18] Fix | Delete

*/

[19] Fix | Delete

public function get_imported_posts( $importer_name, $blog_id ) {

[20] Fix | Delete

global $wpdb;

[21] Fix | Delete

[22] Fix | Delete

$hashtable = array();

[23] Fix | Delete

[24] Fix | Delete

$limit = 100;

[25] Fix | Delete

$offset = 0;

[26] Fix | Delete

[27] Fix | Delete

// Grab all posts in chunks.

[28] Fix | Delete

do {

[29] Fix | Delete

$meta_key = $importer_name . '_' . $blog_id . '_permalink';

[30] Fix | Delete

$results = $wpdb->get_results(

[31] Fix | Delete

$wpdb->prepare(

[32] Fix | Delete

"SELECT post_id, meta_value FROM $wpdb->postmeta WHERE meta_key = %s LIMIT %d,%d",

[33] Fix | Delete

$meta_key,

[34] Fix | Delete

$offset,

[35] Fix | Delete

$limit

[36] Fix | Delete

)

[37] Fix | Delete

);

[38] Fix | Delete

[39] Fix | Delete

// Increment offset.

[40] Fix | Delete

$offset = ( $limit + $offset );

[41] Fix | Delete

[42] Fix | Delete

if ( ! empty( $results ) ) {

[43] Fix | Delete

foreach ( $results as $r ) {

[44] Fix | Delete

// Set permalinks into array.

[45] Fix | Delete

$hashtable[ $r->meta_value ] = (int) $r->post_id;

[46] Fix | Delete

}

[47] Fix | Delete

}

[48] Fix | Delete

} while ( count( $results ) === $limit );

[49] Fix | Delete

[50] Fix | Delete

return $hashtable;

[51] Fix | Delete

}

[52] Fix | Delete

[53] Fix | Delete

/**

[54] Fix | Delete

* Returns count of imported permalinks from WordPress database.

[55] Fix | Delete

*

[56] Fix | Delete

* @global wpdb $wpdb WordPress database abstraction object.

[57] Fix | Delete

*

[58] Fix | Delete

* @param string $importer_name

[59] Fix | Delete

* @param string $blog_id

[60] Fix | Delete

* @return int

[61] Fix | Delete

*/

[62] Fix | Delete

public function count_imported_posts( $importer_name, $blog_id ) {

[63] Fix | Delete

global $wpdb;

[64] Fix | Delete

[65] Fix | Delete

$count = 0;

[66] Fix | Delete

[67] Fix | Delete

// Get count of permalinks.

[68] Fix | Delete

$meta_key = $importer_name . '_' . $blog_id . '_permalink';

[69] Fix | Delete

$result = $wpdb->get_results(

[70] Fix | Delete

$wpdb->prepare(

[71] Fix | Delete

"SELECT COUNT( post_id ) AS cnt FROM $wpdb->postmeta WHERE meta_key = %s",

[72] Fix | Delete

$meta_key

[73] Fix | Delete

)

[74] Fix | Delete

);

[75] Fix | Delete

[76] Fix | Delete

if ( ! empty( $result ) ) {

[77] Fix | Delete

$count = (int) $result[0]->cnt;

[78] Fix | Delete

}

[79] Fix | Delete

[80] Fix | Delete

return $count;

[81] Fix | Delete

}

[82] Fix | Delete

[83] Fix | Delete

/**

[84] Fix | Delete

* Sets array with imported comments from WordPress database.

[85] Fix | Delete

*

[86] Fix | Delete

* @global wpdb $wpdb WordPress database abstraction object.

[87] Fix | Delete

*

[88] Fix | Delete

* @param string $blog_id

[89] Fix | Delete

* @return array

[90] Fix | Delete

*/

[91] Fix | Delete

public function get_imported_comments( $blog_id ) {

[92] Fix | Delete

global $wpdb;

[93] Fix | Delete

[94] Fix | Delete

$hashtable = array();

[95] Fix | Delete

[96] Fix | Delete

$limit = 100;

[97] Fix | Delete

$offset = 0;

[98] Fix | Delete

[99] Fix | Delete

// Grab all comments in chunks.

[100] Fix | Delete

do {

[101] Fix | Delete

$results = $wpdb->get_results(

[102] Fix | Delete

$wpdb->prepare(

[103] Fix | Delete

"SELECT comment_ID, comment_agent FROM $wpdb->comments LIMIT %d,%d",

[104] Fix | Delete

$offset,

[105] Fix | Delete

$limit

[106] Fix | Delete

)

[107] Fix | Delete

);

[108] Fix | Delete

[109] Fix | Delete

// Increment offset.

[110] Fix | Delete

$offset = ( $limit + $offset );

[111] Fix | Delete

[112] Fix | Delete

if ( ! empty( $results ) ) {

[113] Fix | Delete

foreach ( $results as $r ) {

[114] Fix | Delete

// Explode comment_agent key.

[115] Fix | Delete

list ( $comment_agent_blog_id, $source_comment_id ) = explode( '-', $r->comment_agent );

[116] Fix | Delete

[117] Fix | Delete

$source_comment_id = (int) $source_comment_id;

[118] Fix | Delete

[119] Fix | Delete

// Check if this comment came from this blog.

[120] Fix | Delete

if ( (int) $blog_id === (int) $comment_agent_blog_id ) {

[121] Fix | Delete

$hashtable[ $source_comment_id ] = (int) $r->comment_ID;

[122] Fix | Delete

}

[123] Fix | Delete

}

[124] Fix | Delete

}

[125] Fix | Delete

} while ( count( $results ) === $limit );

[126] Fix | Delete

[127] Fix | Delete

return $hashtable;

[128] Fix | Delete

}

[129] Fix | Delete

[130] Fix | Delete

/**

[131] Fix | Delete

* @param int $blog_id

[132] Fix | Delete

* @return int|void

[133] Fix | Delete

*/

[134] Fix | Delete

public function set_blog( $blog_id ) {

[135] Fix | Delete

if ( is_numeric( $blog_id ) ) {

[136] Fix | Delete

$blog_id = (int) $blog_id;

[137] Fix | Delete

} else {

[138] Fix | Delete

$blog = 'http://' . preg_replace( '#^https?://#', '', $blog_id );

[139] Fix | Delete

$parsed = parse_url( $blog );

[140] Fix | Delete

if ( ! $parsed || empty( $parsed['host'] ) ) {

[141] Fix | Delete

fwrite( STDERR, "Error: can not determine blog_id from $blog_id\n" );

[142] Fix | Delete

exit;

[143] Fix | Delete

}

[144] Fix | Delete

if ( empty( $parsed['path'] ) ) {

[145] Fix | Delete

$parsed['path'] = '/';

[146] Fix | Delete

}

[147] Fix | Delete

$blogs = get_sites(

[148] Fix | Delete

array(

[149] Fix | Delete

'domain' => $parsed['host'],

[150] Fix | Delete

'number' => 1,

[151] Fix | Delete

'path' => $parsed['path'],

[152] Fix | Delete

)

[153] Fix | Delete

);

[154] Fix | Delete

if ( ! $blogs ) {

[155] Fix | Delete

fwrite( STDERR, "Error: Could not find blog\n" );

[156] Fix | Delete

exit;

[157] Fix | Delete

}

[158] Fix | Delete

$blog = array_shift( $blogs );

[159] Fix | Delete

$blog_id = (int) $blog->blog_id;

[160] Fix | Delete

}

[161] Fix | Delete

[162] Fix | Delete

if ( function_exists( 'is_multisite' ) ) {

[163] Fix | Delete

if ( is_multisite() ) {

[164] Fix | Delete

switch_to_blog( $blog_id );

[165] Fix | Delete

}

[166] Fix | Delete

}

[167] Fix | Delete

[168] Fix | Delete

return $blog_id;

[169] Fix | Delete

}

[170] Fix | Delete

[171] Fix | Delete

/**

[172] Fix | Delete

* @param int $user_id

[173] Fix | Delete

* @return int|void

[174] Fix | Delete

*/

[175] Fix | Delete

public function set_user( $user_id ) {

[176] Fix | Delete

if ( is_numeric( $user_id ) ) {

[177] Fix | Delete

$user_id = (int) $user_id;

[178] Fix | Delete

} else {

[179] Fix | Delete

$user_id = (int) username_exists( $user_id );

[180] Fix | Delete

}

[181] Fix | Delete

[182] Fix | Delete

if ( ! $user_id || ! wp_set_current_user( $user_id ) ) {

[183] Fix | Delete

fwrite( STDERR, "Error: can not find user\n" );

[184] Fix | Delete

exit;

[185] Fix | Delete

}

[186] Fix | Delete

[187] Fix | Delete

return $user_id;

[188] Fix | Delete

}

[189] Fix | Delete

[190] Fix | Delete

/**

[191] Fix | Delete

* Sorts by strlen, longest string first.

[192] Fix | Delete

*

[193] Fix | Delete

* @param string $a

[194] Fix | Delete

* @param string $b

[195] Fix | Delete

* @return int

[196] Fix | Delete

*/

[197] Fix | Delete

public function cmpr_strlen( $a, $b ) {

[198] Fix | Delete

return strlen( $b ) - strlen( $a );

[199] Fix | Delete

}

[200] Fix | Delete

[201] Fix | Delete

/**

[202] Fix | Delete

* Gets URL.

[203] Fix | Delete

*

[204] Fix | Delete

* @param string $url

[205] Fix | Delete

* @param string $username

[206] Fix | Delete

* @param string $password

[207] Fix | Delete

* @param bool $head

[208] Fix | Delete

* @return array

[209] Fix | Delete

*/

[210] Fix | Delete

public function get_page(

[211] Fix | Delete

$url,

[212] Fix | Delete

$username = '',

[213] Fix | Delete

#[\SensitiveParameter]

[214] Fix | Delete

$password = '',

[215] Fix | Delete

$head = false

[216] Fix | Delete

) {

[217] Fix | Delete

// Increase the timeout.

[218] Fix | Delete

add_filter( 'http_request_timeout', array( $this, 'bump_request_timeout' ) );

[219] Fix | Delete

[220] Fix | Delete

$headers = array();

[221] Fix | Delete

$args = array();

[222] Fix | Delete

if ( true === $head ) {

[223] Fix | Delete

$args['method'] = 'HEAD';

[224] Fix | Delete

}

[225] Fix | Delete

if ( ! empty( $username ) && ! empty( $password ) ) {

[226] Fix | Delete

$headers['Authorization'] = 'Basic ' . base64_encode( "$username:$password" );

[227] Fix | Delete

}

[228] Fix | Delete

[229] Fix | Delete

$args['headers'] = $headers;

[230] Fix | Delete

[231] Fix | Delete

return wp_safe_remote_request( $url, $args );

[232] Fix | Delete

}

[233] Fix | Delete

[234] Fix | Delete

/**

[235] Fix | Delete

* Bumps up the request timeout for http requests.

[236] Fix | Delete

*

[237] Fix | Delete

* @param int $val

[238] Fix | Delete

* @return int

[239] Fix | Delete

*/

[240] Fix | Delete

public function bump_request_timeout( $val ) {

[241] Fix | Delete

return 60;

[242] Fix | Delete

}

[243] Fix | Delete

[244] Fix | Delete

/**

[245] Fix | Delete

* Checks if user has exceeded disk quota.

[246] Fix | Delete

*

[247] Fix | Delete

* @return bool

[248] Fix | Delete

*/

[249] Fix | Delete

public function is_user_over_quota() {

[250] Fix | Delete

if ( function_exists( 'upload_is_user_over_quota' ) ) {

[251] Fix | Delete

if ( upload_is_user_over_quota() ) {

[252] Fix | Delete

return true;

[253] Fix | Delete

}

[254] Fix | Delete

}

[255] Fix | Delete

[256] Fix | Delete

return false;

[257] Fix | Delete

}

[258] Fix | Delete

[259] Fix | Delete

/**

[260] Fix | Delete

* Replaces newlines, tabs, and multiple spaces with a single space.

[261] Fix | Delete

*

[262] Fix | Delete

* @param string $text

[263] Fix | Delete

* @return string

[264] Fix | Delete

*/

[265] Fix | Delete

public function min_whitespace( $text ) {

[266] Fix | Delete

return preg_replace( '|[\r\n\t ]+|', ' ', $text );

[267] Fix | Delete

}

[268] Fix | Delete

[269] Fix | Delete

/**

[270] Fix | Delete

* Resets global variables that grow out of control during imports.

[271] Fix | Delete

*

[272] Fix | Delete

* @since 3.0.0

[273] Fix | Delete

*

[274] Fix | Delete

* @global wpdb $wpdb WordPress database abstraction object.

[275] Fix | Delete

* @global int[] $wp_actions

[276] Fix | Delete

*/

[277] Fix | Delete

public function stop_the_insanity() {

[278] Fix | Delete

global $wpdb, $wp_actions;

[279] Fix | Delete

// Or define( 'WP_IMPORTING', true );

[280] Fix | Delete

$wpdb->queries = array();

[281] Fix | Delete

// Reset $wp_actions to keep it from growing out of control.

[282] Fix | Delete

$wp_actions = array();

[283] Fix | Delete

}

[284] Fix | Delete

}

[285] Fix | Delete

[286] Fix | Delete

/**

[287] Fix | Delete

* Returns value of command line params.

[288] Fix | Delete

* Exits when a required param is not set.

[289] Fix | Delete

*

[290] Fix | Delete

* @param string $param

[291] Fix | Delete

* @param bool $required

[292] Fix | Delete

* @return mixed

[293] Fix | Delete

*/

[294] Fix | Delete

function get_cli_args( $param, $required = false ) {

[295] Fix | Delete

$args = $_SERVER['argv'];

[296] Fix | Delete

if ( ! is_array( $args ) ) {

[297] Fix | Delete

$args = array();

[298] Fix | Delete

}

[299] Fix | Delete

[300] Fix | Delete

$out = array();

[301] Fix | Delete

[302] Fix | Delete

$last_arg = null;

[303] Fix | Delete

$return = null;

[304] Fix | Delete

[305] Fix | Delete

$il = count( $args );

[306] Fix | Delete

[307] Fix | Delete

for ( $i = 1, $il; $i < $il; $i++ ) {

[308] Fix | Delete

if ( (bool) preg_match( '/^--(.+)/', $args[ $i ], $match ) ) {

[309] Fix | Delete

$parts = explode( '=', $match[1] );

[310] Fix | Delete

$key = preg_replace( '/[^a-z0-9]+/', '', $parts[0] );

[311] Fix | Delete

[312] Fix | Delete

if ( isset( $parts[1] ) ) {

[313] Fix | Delete

$out[ $key ] = $parts[1];

[314] Fix | Delete

} else {

[315] Fix | Delete

$out[ $key ] = true;

[316] Fix | Delete

}

[317] Fix | Delete

[318] Fix | Delete

$last_arg = $key;

[319] Fix | Delete

} elseif ( (bool) preg_match( '/^-([a-zA-Z0-9]+)/', $args[ $i ], $match ) ) {

[320] Fix | Delete

for ( $j = 0, $jl = strlen( $match[1] ); $j < $jl; $j++ ) {

[321] Fix | Delete

$key = $match[1][ $j ];

[322] Fix | Delete

$out[ $key ] = true;

[323] Fix | Delete

}

[324] Fix | Delete

[325] Fix | Delete

$last_arg = $key;

[326] Fix | Delete

} elseif ( null !== $last_arg ) {

[327] Fix | Delete

$out[ $last_arg ] = $args[ $i ];

[328] Fix | Delete

}

[329] Fix | Delete

}

[330] Fix | Delete

[331] Fix | Delete

// Check array for specified param.

[332] Fix | Delete

if ( isset( $out[ $param ] ) ) {

[333] Fix | Delete

// Set return value.

[334] Fix | Delete

$return = $out[ $param ];

[335] Fix | Delete

}

[336] Fix | Delete

[337] Fix | Delete

// Check for missing required param.

[338] Fix | Delete

if ( ! isset( $out[ $param ] ) && $required ) {

[339] Fix | Delete

// Display message and exit.

[340] Fix | Delete

echo "\"$param\" parameter is required but was not specified\n";

[341] Fix | Delete

exit;

[342] Fix | Delete

}

[343] Fix | Delete

[344] Fix | Delete

return $return;

[345] Fix | Delete

}

[346] Fix | Delete

[347] Fix | Delete