<?php

/**

 * WordPress Importer

 *

 * @package WordPress

 * @subpackage Importer

 */

/**

 * WordPress Importer

 *

 * Will process the WordPress eXtended RSS files that you upload from the export

 * file.

 *

 * @since unknown

 */

class WP_Import {

	var $post_ids_processed = array ();

	var $orphans = array ();

	var $file;

	var $id;

	var $mtnames = array ();

	var $newauthornames = array ();

	var $allauthornames = array ();

	var $author_ids = array ();

	var $tags = array ();

	var $categories = array ();

	var $j = -1;

	var $fetch_attachments = false;

	var $url_remap = array ();

	function header() {

		echo '<div class="wrap">';

		screen_icon();

		echo '<h2>'.__('Import WordPress').'</h2>';

	}

	function footer() {

		echo '</div>';

	}

	function unhtmlentities($string) { // From php.net for < 4.3 compat

		$trans_tbl = get_html_translation_table(HTML_ENTITIES);

		$trans_tbl = array_flip($trans_tbl);

		return strtr($string, $trans_tbl);

	}

	function greet() {

		echo '<div class="narrow">';

		echo '<p>'.__('Howdy! Upload your WordPress eXtended RSS (WXR) file and we&#8217;ll import the posts, pages, comments, custom fields, categories, and tags into this blog.').'</p>';

		echo '<p>'.__('Choose a WordPress WXR file to upload, then click Upload file and import.').'</p>';

		wp_import_upload_form("admin.php?import=wordpress&step=1");

		echo '</div>';

	}

	function get_tag( $string, $tag ) {

		global $wpdb;

		preg_match("|<$tag.*?>(.*?)</$tag>|is", $string, $return);

		$return = preg_replace('|^<!\[CDATA\[(.*)\]\]>$|s', '$1', $return[1]);

		$return = $wpdb->escape( trim( $return ) );

		return $return;

	}

	function has_gzip() {

		return is_callable('gzopen');

	}

	function fopen($filename, $mode='r') {

		if ( $this->has_gzip() )

			return gzopen($filename, $mode);

		return fopen($filename, $mode);

	}

	function feof($fp) {

		if ( $this->has_gzip() )

			return gzeof($fp);

		return feof($fp);

	}

	function fgets($fp, $len=8192) {

		if ( $this->has_gzip() )

			return gzgets($fp, $len);

		return fgets($fp, $len);

	}

	function fclose($fp) {

		if ( $this->has_gzip() )

			return gzclose($fp);

		return fclose($fp);

	}

	function get_entries($process_post_func=NULL) {

		set_magic_quotes_runtime(0);

		$doing_entry = false;

		$is_wxr_file = false;

		$fp = $this->fopen($this->file, 'r');

		if ($fp) {

			while ( !$this->feof($fp) ) {

				$importline = rtrim($this->fgets($fp));

				// this doesn't check that the file is perfectly valid but will at least confirm that it's not the wrong format altogether

				if ( !$is_wxr_file && preg_match('|xmlns:wp="http://wordpress[.]org/export/\d+[.]\d+/"|', $importline) )

					$is_wxr_file = true;

				if ( false !== strpos($importline, '<wp:base_site_url>') ) {

					preg_match('|<wp:base_site_url>(.*?)</wp:base_site_url>|is', $importline, $url);

					$this->base_url = $url[1];

					continue;

				}

				if ( false !== strpos($importline, '<wp:category>') ) {

					preg_match('|<wp:category>(.*?)</wp:category>|is', $importline, $category);

					$this->categories[] = $category[1];

					continue;

				}

				if ( false !== strpos($importline, '<wp:tag>') ) {

					preg_match('|<wp:tag>(.*?)</wp:tag>|is', $importline, $tag);

					$this->tags[] = $tag[1];

					continue;

				}

				if ( false !== strpos($importline, '<item>') ) {

					$this->post = '';

					$doing_entry = true;

					continue;

				}

				if ( false !== strpos($importline, '</item>') ) {

					$doing_entry = false;

					if ($process_post_func)

						call_user_func($process_post_func, $this->post);

					continue;

				}

				if ( $doing_entry ) {

					$this->post .= $importline . "\n";

				}

			}

			$this->fclose($fp);

		}

		return $is_wxr_file;

	}

	function get_wp_authors() {

		// We need to find unique values of author names, while preserving the order, so this function emulates the unique_value(); php function, without the sorting.

		$temp = $this->allauthornames;

		$authors[0] = array_shift($temp);

		$y = count($temp) + 1;

		for ($x = 1; $x < $y; $x ++) {

			$next = array_shift($temp);

			if (!(in_array($next, $authors)))

				array_push($authors, "$next");

		}

		return $authors;

	}

	function get_authors_from_post() {

		global $current_user;

		// this will populate $this->author_ids with a list of author_names => user_ids

		foreach ( $_POST['author_in'] as $i => $in_author_name ) {

			if ( !empty($_POST['user_select'][$i]) ) {

				// an existing user was selected in the dropdown list

				$user = get_userdata( intval($_POST['user_select'][$i]) );

				if ( isset($user->ID) )

					$this->author_ids[$in_author_name] = $user->ID;

			}

			elseif ( $this->allow_create_users() ) {

				// nothing was selected in the dropdown list, so we'll use the name in the text field

				$new_author_name = trim($_POST['user_create'][$i]);

				// if the user didn't enter a name, assume they want to use the same name as in the import file

				if ( empty($new_author_name) )

					$new_author_name = $in_author_name;

				$user_id = username_exists($new_author_name);

				if ( !$user_id ) {

					$user_id = wp_create_user($new_author_name, wp_generate_password());

				}

				$this->author_ids[$in_author_name] = $user_id;

			}

			// failsafe: if the user_id was invalid, default to the current user

			if ( empty($this->author_ids[$in_author_name]) ) {

				$this->author_ids[$in_author_name] = intval($current_user->ID);

			}

		}

	}

	function wp_authors_form() {

?>

<?php screen_icon(); ?>

<h2><?php _e('Assign Authors'); ?></h2>

<p><?php _e('To make it easier for you to edit and save the imported posts and drafts, you may want to change the name of the author of the posts. For example, you may want to import all the entries as <code>admin</code>s entries.'); ?></p>

<?php

	if ( $this->allow_create_users() ) {

		echo '<p>'.__('If a new user is created by WordPress, a password will be randomly generated. Manually change the user&#8217;s details if necessary.')."</p>\n";

	}

		$authors = $this->get_wp_authors();

		echo '<form action="?import=wordpress&amp;step=2&amp;id=' . $this->id . '" method="post">';

		wp_nonce_field('import-wordpress');

		echo '<ol id="authors">';

		$j = -1;

		foreach ($authors as $author) {

			++ $j;

			echo '<li>'.__('Import author:').' <strong>'.$author.'</strong><br />';

			$this->users_form($j, $author);

			echo '</li>';

		}

		if ( $this->allow_fetch_attachments() ) {

?>

</ol>

<?php screen_icon(); ?>

<h2><?php _e('Import Attachments'); ?></h2>

<p>

	<input type="checkbox" value="1" name="attachments" id="import-attachments" />

	<label for="import-attachments"><?php _e('Download and import file attachments') ?></label>

</p>

<?php

		}

		echo '<p class="submit">';

		echo '<input type="submit" class="button" value="'. esc_attr__('Submit') .'" />'.'<br />';

		echo '</p>';

		echo '</form>';

	}

	function users_form($n, $author) {

		if ( $this->allow_create_users() ) {

			printf('<label>'.__('Create user %1$s or map to existing'), ' <input type="text" value="'. esc_attr($author) .'" name="'.'user_create['.intval($n).']'.'" maxlength="30" /></label> <br />');

		}

		else {

			echo __('Map to existing').'<br />';

		}

		// keep track of $n => $author name

		echo '<input type="hidden" name="author_in['.intval($n).']" value="' . esc_attr($author).'" />';

		$users = get_users_of_blog();

?><select name="user_select[<?php echo $n; ?>]">

	<option value="0"><?php _e('- Select -'); ?></option>

	<?php

		foreach ($users as $user) {

			echo '<option value="'.$user->user_id.'">'.$user->user_login.'</option>';

		}

?>

	</select>

	<?php

	}

	function select_authors() {

		$is_wxr_file = $this->get_entries(array(&$this, 'process_author'));

		if ( $is_wxr_file ) {

			$this->wp_authors_form();

		}

		else {

			echo '<h2>'.__('Invalid file').'</h2>';

			echo '<p>'.__('Please upload a valid WXR (WordPress eXtended RSS) export file.').'</p>';

		}

	}

	// fetch the user ID for a given author name, respecting the mapping preferences

	function checkauthor($author) {

		global $current_user;

		if ( !empty($this->author_ids[$author]) )

			return $this->author_ids[$author];

		// failsafe: map to the current user

		return $current_user->ID;

	}

	function process_categories() {

		global $wpdb;

		$cat_names = (array) get_terms('category', 'fields=names');

		while ( $c = array_shift($this->categories) ) {

			$cat_name = trim($this->get_tag( $c, 'wp:cat_name' ));

			// If the category exists we leave it alone

			if ( in_array($cat_name, $cat_names) )

				continue;

			$category_nicename	= $this->get_tag( $c, 'wp:category_nicename' );

			$category_description = $this->get_tag( $c, 'wp:category_description' );

			$posts_private		= (int) $this->get_tag( $c, 'wp:posts_private' );

			$links_private		= (int) $this->get_tag( $c, 'wp:links_private' );

			$parent = $this->get_tag( $c, 'wp:category_parent' );

			if ( empty($parent) )

				$category_parent = '0';

			else

				$category_parent = category_exists($parent);

			$catarr = compact('category_nicename', 'category_parent', 'posts_private', 'links_private', 'posts_private', 'cat_name', 'category_description');

			$cat_ID = wp_insert_category($catarr);

		}

	}

	function process_tags() {

		global $wpdb;

		$tag_names = (array) get_terms('post_tag', 'fields=names');

		while ( $c = array_shift($this->tags) ) {

			$tag_name = trim($this->get_tag( $c, 'wp:tag_name' ));

			// If the category exists we leave it alone

			if ( in_array($tag_name, $tag_names) )

				continue;

			$slug = $this->get_tag( $c, 'wp:tag_slug' );

			$description = $this->get_tag( $c, 'wp:tag_description' );

			$tagarr = compact('slug', 'description');

			$tag_ID = wp_insert_term($tag_name, 'post_tag', $tagarr);

		}

	}

	function process_author($post) {

		$author = $this->get_tag( $post, 'dc:creator' );

		if ($author)

			$this->allauthornames[] = $author;

	}

	function process_posts() {

		echo '<ol>';

		$this->get_entries(array(&$this, 'process_post'));

		echo '</ol>';

		wp_import_cleanup($this->id);

		do_action('import_done', 'wordpress');

		echo '<h3>'.sprintf(__('All done.').' <a href="%s">'.__('Have fun!').'</a>', get_option('home')).'</h3>';

	}

	function process_post($post) {

		global $wpdb;

		$post_ID = (int) $this->get_tag( $post, 'wp:post_id' );

  		if ( $post_ID && !empty($this->post_ids_processed[$post_ID]) ) // Processed already

			return 0;

		set_time_limit( 60 );

		// There are only ever one of these

		$post_title     = $this->get_tag( $post, 'title' );

		$post_date      = $this->get_tag( $post, 'wp:post_date' );

		$post_date_gmt  = $this->get_tag( $post, 'wp:post_date_gmt' );

		$comment_status = $this->get_tag( $post, 'wp:comment_status' );

		$ping_status    = $this->get_tag( $post, 'wp:ping_status' );

		$post_status    = $this->get_tag( $post, 'wp:status' );

		$post_name      = $this->get_tag( $post, 'wp:post_name' );

		$post_parent    = $this->get_tag( $post, 'wp:post_parent' );

		$menu_order     = $this->get_tag( $post, 'wp:menu_order' );

		$post_type      = $this->get_tag( $post, 'wp:post_type' );

		$post_password  = $this->get_tag( $post, 'wp:post_password' );

		$guid           = $this->get_tag( $post, 'guid' );

		$post_author    = $this->get_tag( $post, 'dc:creator' );

		$post_excerpt = $this->get_tag( $post, 'excerpt:encoded' );

		$post_excerpt = preg_replace_callback('|<(/?[A-Z]+)|', create_function('$match', 'return "<" . strtolower($match[1]);'), $post_excerpt);

		$post_excerpt = str_replace('<br>', '<br />', $post_excerpt);

		$post_excerpt = str_replace('<hr>', '<hr />', $post_excerpt);

		$post_content = $this->get_tag( $post, 'content:encoded' );

		$post_content = preg_replace_callback('|<(/?[A-Z]+)|', create_function('$match', 'return "<" . strtolower($match[1]);'), $post_content);

		$post_content = str_replace('<br>', '<br />', $post_content);

		$post_content = str_replace('<hr>', '<hr />', $post_content);

		preg_match_all('|<category domain="tag">(.*?)</category>|is', $post, $tags);

		$tags = $tags[1];

		$tag_index = 0;

		foreach ($tags as $tag) {

			$tags[$tag_index] = $wpdb->escape($this->unhtmlentities(str_replace(array ('<![CDATA[', ']]>'), '', $tag)));

			$tag_index++;

		}