<?php

/**

 * LiveJournal API Importer

 *

 * @package WordPress

 * @subpackage Importer

 */

// XML-RPC library for communicating with LiveJournal API

require_once( ABSPATH . WPINC . '/class-IXR.php' );

/**

 * LiveJournal API Importer class

 *

 * Imports your LiveJournal contents into WordPress using the LJ API

 *

 * @since 2.8

 */

class LJ_API_Import {

	var $comments_url = 'http://www.livejournal.com/export_comments.bml';

	var $ixr_url      = 'http://www.livejournal.com/interface/xmlrpc';

	var $ixr;

	var $username;

	var $password;

	var $comment_meta;

	var $comments;

	var $usermap;

	var $postmap;

	var $commentmap;

	var $pointers = array();

	// This list taken from LJ, they don't appear to have an API for it

	var $moods = array( '1' => 'aggravated',

						'10' => 'discontent',

						'100' => 'rushed',

						'101' => 'contemplative',

						'102' => 'nerdy',

						'103' => 'geeky',

						'104' => 'cynical',

						'105' => 'quixotic',

						'106' => 'crazy',

						'107' => 'creative',

						'108' => 'artistic',

						'109' => 'pleased',

						'11' => 'energetic',

						'110' => 'bitchy',

						'111' => 'guilty',

						'112' => 'irritated',

						'113' => 'blank',

						'114' => 'apathetic',

						'115' => 'dorky',

						'116' => 'impressed',

						'117' => 'naughty',

						'118' => 'predatory',

						'119' => 'dirty',

						'12' => 'enraged',

						'120' => 'giddy',

						'121' => 'surprised',

						'122' => 'shocked',

						'123' => 'rejected',

						'124' => 'numb',

						'125' => 'cheerful',

						'126' => 'good',

						'127' => 'distressed',

						'128' => 'intimidated',

						'129' => 'crushed',

						'13' => 'enthralled',

						'130' => 'devious',

						'131' => 'thankful',

						'132' => 'grateful',

						'133' => 'jealous',

						'134' => 'nervous',

						'14' => 'exhausted',

						'15' => 'happy',

						'16' => 'high',

						'17' => 'horny',

						'18' => 'hungry',

						'19' => 'infuriated',

						'2' => 'angry',

						'20' => 'irate',

						'21' => 'jubilant',

						'22' => 'lonely',

						'23' => 'moody',

						'24' => 'pissed off',

						'25' => 'sad',

						'26' => 'satisfied',

						'27' => 'sore',

						'28' => 'stressed',

						'29' => 'thirsty',

						'3' => 'annoyed',

						'30' => 'thoughtful',

						'31' => 'tired',

						'32' => 'touched',

						'33' => 'lazy',

						'34' => 'drunk',

						'35' => 'ditzy',

						'36' => 'mischievous',

						'37' => 'morose',

						'38' => 'gloomy',

						'39' => 'melancholy',

						'4' => 'anxious',

						'40' => 'drained',

						'41' => 'excited',

						'42' => 'relieved',

						'43' => 'hopeful',

						'44' => 'amused',

						'45' => 'determined',

						'46' => 'scared',

						'47' => 'frustrated',

						'48' => 'indescribable',

						'49' => 'sleepy',

						'5' => 'bored',

						'51' => 'groggy',

						'52' => 'hyper',

						'53' => 'relaxed',

						'54' => 'restless',

						'55' => 'disappointed',

						'56' => 'curious',

						'57' => 'mellow',

						'58' => 'peaceful',

						'59' => 'bouncy',

						'6' => 'confused',

						'60' => 'nostalgic',

						'61' => 'okay',

						'62' => 'rejuvenated',

						'63' => 'complacent',

						'64' => 'content',

						'65' => 'indifferent',

						'66' => 'silly',

						'67' => 'flirty',

						'68' => 'calm',

						'69' => 'refreshed',

						'7' => 'crappy',

						'70' => 'optimistic',

						'71' => 'pessimistic',

						'72' => 'giggly',

						'73' => 'pensive',

						'74' => 'uncomfortable',

						'75' => 'lethargic',

						'76' => 'listless',

						'77' => 'recumbent',

						'78' => 'exanimate',

						'79' => 'embarrassed',

						'8' => 'cranky',

						'80' => 'envious',

						'81' => 'sympathetic',

						'82' => 'sick',

						'83' => 'hot',

						'84' => 'cold',

						'85' => 'worried',

						'86' => 'loved',

						'87' => 'awake',

						'88' => 'working',

						'89' => 'productive',

						'9' => 'depressed',

						'90' => 'accomplished',

						'91' => 'busy',

						'92' => 'blah',

						'93' => 'full',

						'95' => 'grumpy',

						'96' => 'weird',

						'97' => 'nauseated',

						'98' => 'ecstatic',

						'99' => 'chipper' );

	function header() {

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

		screen_icon();

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

	}

	function footer() {

		echo '</div>';

	}

	function greet() {

		?>

		<div class="narrow">

		<form action="admin.php?import=livejournal" method="post">

		<?php wp_nonce_field( 'lj-api-import' ) ?>

		<?php if ( get_option( 'ljapi_username' ) && get_option( 'ljapi_password' ) ) : ?>

			<input type="hidden" name="step" value="<?php echo esc_attr( get_option( 'ljapi_step' ) ) ?>" />

			<p><?php _e( 'It looks like you attempted to import your LiveJournal posts previously and got interrupted.' ) ?></p>

			<p class="submit">

				<input type="submit" class="button-primary" value="<?php esc_attr_e( 'Continue previous import' ) ?>" />

			</p>

			<p class="submitbox"><a href="<?php echo esc_url($_SERVER['PHP_SELF'] . '?import=livejournal&amp;step=-1&amp;_wpnonce=' . wp_create_nonce( 'lj-api-import' ) . '&amp;_wp_http_referer=' . esc_attr( $_SERVER['REQUEST_URI'] )) ?>" class="deletion submitdelete"><?php _e( 'Cancel &amp; start a new import' ) ?></a></p>

			<p>

		<?php else : ?>

			<input type="hidden" name="step" value="1" />

			<input type="hidden" name="login" value="true" />

			<p><?php _e( 'Howdy! This importer allows you to connect directly to LiveJournal and download all your entries and comments' ) ?></p>

			<p><?php _e( 'Enter your LiveJournal username and password below so we can connect to your account:' ) ?></p>

			<table class="form-table">

			<tr>

			<th scope="row"><label for="lj_username"><?php _e( 'LiveJournal Username' ) ?></label></th>

			<td><input type="text" name="lj_username" id="lj_username" class="regular-text" /></td>

			</tr>

			<tr>

			<th scope="row"><label for="lj_password"><?php _e( 'LiveJournal Password' ) ?></label></th>

			<td><input type="password" name="lj_password" id="lj_password" class="regular-text" /></td>

			</tr>

			</table>

			<p><?php _e( 'If you have any entries on LiveJournal which are marked as private, they will be password-protected when they are imported so that only people who know the password can see them.' ) ?></p>

			<p><?php _e( 'If you don&#8217;t enter a password, ALL ENTRIES from your LiveJournal will be imported as public posts in WordPress.' ) ?></p>

			<p><?php _e( 'Enter the password you would like to use for all protected entries here:' ) ?></p>

			<table class="form-table">

			<tr>

			<th scope="row"><label for="protected_password"><?php _e( 'Protected Post Password' ) ?></label></th>

			<td><input type="text" name="protected_password" id="protected_password" class="regular-text" /></td>

			</tr>

			</table>

			<p><?php _e( "<strong>WARNING:</strong> This can take a really long time if you have a lot of entries in your LiveJournal, or a lot of comments. Ideally, you should only start this process if you can leave your computer alone while it finishes the import." ) ?></p>

			<p class="submit">

				<input type="submit" class="button-primary" value="<?php esc_attr_e( 'Connect to LiveJournal and Import' ) ?>" />

			</p>

			<p><?php _e( '<strong>NOTE:</strong> If the import process is interrupted for <em>any</em> reason, come back to this page and it will continue from where it stopped automatically.' ) ?></p>

			<noscript>

				<p><?php _e( '<strong>NOTE:</strong> You appear to have JavaScript disabled, so you will need to manually click through each step of this importer. If you enable JavaScript, it will step through automatically.' ) ?></p>

			</noscript>

		<?php endif; ?>

		</form>

		</div>

		<?php

	}

	function download_post_meta() {

		$total           = (int) get_option( 'ljapi_total' );

		$count           = (int) get_option( 'ljapi_count' );

		$lastsync        = get_option( 'ljapi_lastsync' );

		if ( !$lastsync ) {

			update_option( 'ljapi_lastsync', '1900-01-01 00:00:00' );

		}

		$sync_item_times = get_option( 'ljapi_sync_item_times' );

		if ( !is_array( $sync_item_times ) )

			$sync_item_times = array();

		do {

			$lastsync = date( 'Y-m-d H:i:s', strtotime( get_option( 'ljapi_lastsync' ) ) );

			$synclist = $this->lj_ixr( 'syncitems', array( 'ver' => 1, 'lastsync' => $lastsync ) );

			if ( is_wp_error( $synclist ) )

				return $synclist;

			// Keep track of if we've downloaded everything

			$total = $synclist['total'];

			$count = $synclist['count'];

			foreach ( $synclist['syncitems'] as $event ) {

				if ( substr( $event['item'], 0, 2 ) == 'L-' ) {

					$sync_item_times[ str_replace( 'L-', '', $event['item'] ) ] = $event['time'];

					if ( $event['time'] > $lastsync ) {

						$lastsync = $event['time'];

						update_option( 'ljapi_lastsync', $lastsync );

					}

				}

			}

		} while ( $total > $count );

		// endwhile - all post meta is cached locally

		unset( $synclist );

		update_option( 'ljapi_sync_item_times', $sync_item_times );

		update_option( 'ljapi_total', $total );

		update_option( 'ljapi_count', $count );

		echo '<p>' . __( 'Post metadata has been downloaded, proceeding with posts...' ) . '</p>';

	}

	function download_post_bodies() {

		$imported_count  = (int) get_option( 'ljapi_imported_count' );

		$sync_item_times = get_option( 'ljapi_sync_item_times' );

		$lastsync        = get_option( 'ljapi_lastsync_posts' );

		if ( !$lastsync )

			update_option( 'ljapi_lastsync_posts', date( 'Y-m-d H:i:s', 0 ) );

		$count = 0;

		echo '<ol>';

		do {

			$lastsync = date( 'Y-m-d H:i:s', strtotime( get_option( 'ljapi_lastsync_posts' ) ) );

			// Get the batch of items that match up with the syncitems list

			$itemlist = $this->lj_ixr( 'getevents', array( 'ver' => 1,

															'selecttype' => 'syncitems',

															'lineendings' => 'pc',

															'lastsync' => $lastsync ) );

			if ( is_wp_error( $itemlist ) )

				return $itemlist;

			if ( $num = count( $itemlist['events'] ) ) {

				for ( $e = 0; $e < count( $itemlist['events'] ); $e++ ) {

					$event = $itemlist['events'][$e];

					$imported_count++;

					$inserted = $this->import_post( $event );

					if ( is_wp_error( $inserted ) )

						return $inserted;

					if ( $sync_item_times[ $event['itemid'] ] > $lastsync )

						$lastsync = $sync_item_times[ $event['itemid'] ];

					wp_cache_flush();

				}

				update_option( 'ljapi_lastsync_posts',  $lastsync );

				update_option( 'ljapi_imported_count',  $imported_count );

				update_option( 'ljapi_last_sync_count', $num );

			}

			$count++;

		} while ( $num > 0 && $count < 3 ); // Doing up to 3 requests at a time to avoid memory problems

		// Used so that step1 knows when to stop posting back on itself

		update_option( 'ljapi_last_sync_count', $num );

		// Counter just used to show progress to user

		update_option( 'ljapi_post_batch', ( (int) get_option( 'ljapi_post_batch' ) + 1 ) );

		echo '</ol>';

	}

	function import_post( $post ) {

		global $wpdb;

		// Make sure we haven't already imported this one

		if ( $this->get_wp_post_ID( $post['itemid'] ) )

			return;

		$user = wp_get_current_user();

		$post_author      = $user->ID;

		$post['security'] = !empty( $post['security'] ) ? $post['security'] : '';

		$post_status      = ( 'private' == trim( $post['security'] ) ) ? 'private' : 'publish'; // Only me

		$post_password    = ( 'usemask' == trim( $post['security'] ) ) ? $this->protected_password : ''; // "Friends" via password

		// For some reason, LJ sometimes sends a date as "2004-04-1408:38:00" (no space btwn date/time)

		$post_date = $post['eventtime'];

		if ( 18 == strlen( $post_date ) )

			$post_date = substr( $post_date, 0, 10 ) . ' ' . substr( $post_date, 10 );

		// Cleaning up and linking the title

		$post_title = isset( $post['subject'] ) ? trim( $post['subject'] ) : '';

		$post_title = $this->translate_lj_user( $post_title ); // Translate it, but then we'll strip the link

		$post_title = strip_tags( $post_title ); // Can't have tags in the title in WP

		$post_title = $wpdb->escape( $post_title );

		// Clean up content

		$post_content = $post['event'];

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

		// XHTMLize some tags

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

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

		// lj-cut ==>  <!--more-->

		$post_content = preg_replace( '|<lj-cut text="([^"]*)">|is', '<!--more $1-->', $post_content );

		$post_content = str_replace( array( '<lj-cut>', '</lj-cut>' ), array( '<!--more-->', '' ), $post_content );

		$first = strpos( $post_content, '<!--more' );

		$post_content = substr( $post_content, 0, $first + 1 ) . preg_replace( '|<!--more(.*)?-->|sUi', '', substr( $post_content, $first + 1 ) );

		// lj-user ==>  a href

		$post_content = $this->translate_lj_user( $post_content );

		//$post_content = force_balance_tags( $post_content );

		$post_content = $wpdb->escape( $post_content );

		// Handle any tags associated with the post

		$tags_input = !empty( $post['props']['taglist'] ) ? $post['props']['taglist'] : '';

		// Check if comments are closed on this post

		$comment_status = !empty( $post['props']['opt_nocomments'] ) ? 'closed' : 'open';

		echo '<li>';

		if ( $post_id = post_exists( $post_title, $post_content, $post_date ) ) {

			printf( __( 'Post <strong>%s</strong> already exists.' ), stripslashes( $post_title ) );

		} else {

			printf( __( 'Imported post <strong>%s</strong>...' ), stripslashes( $post_title ) );

			$postdata = compact( 'post_author', 'post_date', 'post_content', 'post_title', 'post_status', 'post_password', 'tags_input', 'comment_status' );

			$post_id = wp_insert_post( $postdata, true );

			if ( is_wp_error( $post_id ) ) {

				if ( 'empty_content' == $post_id->get_error_code() )

					return; // Silent skip on "empty" posts

				return $post_id;

			}

			if ( !$post_id ) {

				_e( 'Couldn’t get post ID (creating post failed!)' );

				echo '</li>';

				return new WP_Error( 'insert_post_failed', __( 'Failed to create post.' ) );

			}

			// Handle all the metadata for this post

			$this->insert_postmeta( $post_id, $post );

		}

		echo '</li>';

	}

	// Convert lj-user tags to links to that user

	function translate_lj_user( $str ) {

		return preg_replace( '|<lj\s+user\s*=\s*["\']([\w-]+)["\']>|', '<a href="http://$1.livejournal.com/" class="lj-user">$1</a>', $str );

	}

	function insert_postmeta( $post_id, $post ) {

		// Need the original LJ id for comments

		add_post_meta( $post_id, 'lj_itemid', $post['itemid'] );

		// And save the permalink on LJ in case we want to link back or something