WordPress Migration: redirecting old URLs

<?php
	#
	# $Id: news.php,v 1.14 2010/02/08 16:29:22 dan Exp $
	#
	# Copyright (c) 1998-2003 DVL Software Limited
	#

	require($_SERVER["DOCUMENT_ROOT"] . "/include/common.php");
	require($_SERVER["DOCUMENT_ROOT"] . "/include/freebsddiary.php");
	require($_SERVER["DOCUMENT_ROOT"] . "/include/databaselogin.php");
	
$MaxArticles = 10000;

$HTML = '';

   $HTML .= '<!DOCTYPE rss PUBLIC "-//Netscape Communications//DTD RSS 0.91//EN"' . "\n";
   $HTML .= '        "http://www.rssboard.org/rss-0.91.dtd">' . "\n";
   $HTML .= '<rss version="0.91">' . "\n";

   $HTML .= "\n";

   $HTML .= '<channel>' . "\n";
   $HTML .= '  <title>The FreeBSD Diary</title>' . "\n";
   $HTML .= '  <link>http://www.freebsddiary.org/</link>' . "\n";
   $HTML .= '  <description>The largest collection of practical examples for FreeBSD!</description>' . "\n";
   $HTML .= '  <language>en-us</language>' . "\n";
   $HTML .= '  <copyright>Copyright ' . GetCopyrightYears() . ', DVL Software Limited.</copyright>' . "\n";

   $HTML .= "\n";

   $sql = "SELECT A.id as article_id,
                  A.name AS article_name, 
                  A.author as author, 
                  A.actual_date as date,
                  A.filename
             FROM articles A
            WHERE A.completed = 'Y'
         order by date, article_id ";

#die("<pre>$sql</pre>");

   $result = pg_query($db, $sql);
   while ($myrow = pg_fetch_array($result)) {
      $email = trim($myrow["email"]);
      $HTML .= '  <item>' . "\n";
      $HTML .= '    <title>' .  htmlentities($myrow["article_name"]) . '</title>' . "\n";
      $HTML .= '    <dc:date>' .  $myrow["date"] . '</dc:date>' . "\n";
      $HTML .= '    <author>' .  $myrow["author"] . '</author>' . "\n";
      $HTML .= '    <filename>' . htmlentities($myrow['filename']) . '</filename>' . "\n";
      $HTML .= '  </item>' . "\n";
   }

   $HTML .= '</channel>' . "\n";
   $HTML .= '</rss>' . "\n";

   header('Content-type: text/xml');

   echo '<?xml version="1.0"?>', "\n";   
   echo $HTML;

<?php
/*
Plugin Name: RSS Importer Redirects
Plugin URI: 
Description: Import article headers from RSS feed and create redirects. Based upon http://wordpress.org/extend/plugins/rss-importer/
Author: Dan Langille
Author URI: http://wordpress.org/
Version: 0.1
Stable tag: 0.1
License: GPL version 2 or later - http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
*/

if ( !defined('WP_LOAD_IMPORTERS') )
	return;

// Load Importer API
require_once ABSPATH . 'wp-admin/includes/import.php';

if ( !class_exists( 'WP_Importer' ) ) {
	$class_wp_importer = ABSPATH . 'wp-admin/includes/class-wp-importer.php';
	if ( file_exists( $class_wp_importer ) )
		require_once $class_wp_importer;
}

/**
 * RSS Importer
 *
 * @package WordPress
 * @subpackage Importer
 */

/**
 * RSS Importer
 *
 * Will process a RSS feed for importing posts into WordPress. This is a very
 * limited importer and should only be used as the last resort, when no other
 * importer is available.
 *
 * @since unknown
 */
if ( class_exists( 'WP_Importer' ) ) {
class RSS_ImportRedirects extends WP_Importer {

	var $posts = array ();
	var $file;
	
	var $authors = array();

function get_all_authors() {
	global $wpdb;

	$query = "SELECT ID, display_name, user_login, user_email FROM $wpdb->users";

	return $wpdb->get_results( $wpdb->prepare($query, $this->authors ));

	return 0;
}

function get_all_posts() {
	global $wpdb;

	$query = "SELECT ID, post_author, post_date, post_title FROM $wpdb->posts WHERE post_type = 'post'";

	return $wpdb->get_results( $wpdb->prepare($query, $posts ));
}

	function header() {
		echo '<div class="wrap">';
		screen_icon();
		echo '<h2>'.__('Import RSS redirects', 'rss-importer-redirects').'</h2>';
	}

	function footer() {
		echo '</div>';
	}

	function greet() {
		echo '<div class="narrow">';
		echo '<p>'.__('Howdy! This importer allows you to import redirects from an RSS 2.0 file into your WordPress site. This is useful if you want to use the old URLS but redirect them to the new URLS. Pick an RSS file to upload and click Import.', 'rss-importer-redirects').'</p>';
		wp_import_upload_form("admin.php?import=rssredirects&amp;step=1");
		echo '</div>';
	}

	function _normalize_tag( $matches ) {
		return '<' . strtolower( $matches[1] );
	}

	function get_posts() {
		global $wpdb;

		set_magic_quotes_runtime(0);
		$datalines = file($this->file); // Read the file into an array
		$importdata = implode('', $datalines); // squish it
		$importdata = str_replace(array ("\r\n", "\r"), "\n", $importdata);

		$authors = $this->get_all_authors();
		$authorLoginLookup = array();
		$authorNameLookup  = array();
		$authorLookup      = array();
		foreach($authors as $author)
		{
		        $key = $author->display_name . ' ## ' . $author->user_email;
		        if (isset($authorLoookup[$key]))
		        {
		                echo 'WARNING: duplicate display name / email combination ' . $key . '<br>';
		        }
                        $authorLoookup[$key] = $author->ID;
		        if (isset($authorNameLookup [$author->display_name]))
		        {
#		                echo 'WARNING: duplicate display_name ' . $author->display_name . '<br>';
                        }
		        if (isset($authorNameLookup [$author->user_login]))
		        {
#		                echo 'WARNING: duplicate user_login ' . $author->user_login . '<br>';
                        }
		        $authorNameLookup [$author->display_name] = $author->ID;
		        $authorLoginLookup[$author->user_login]   = $author->ID;
                }
                
#		echo ' the authors are <pre>' . print_r($authorNameLookup,  true) . '</pre>';
#		echo ' the authors are <pre>' . print_r($authorLoginLookup, true) . '</pre>';

		$posts = $this->get_all_posts();
		$postLookup = array();
		foreach($posts as $post)
		{
		        $key = $post->post_author . ' @@ ' .  $post->post_date . ' @@ ' . $post->post_title;
		        if (isset($postLookup[$key]))
		        {
                		echo ' the posts are <pre>' . print_r($postLookup, true) . '</pre>';
		                die('duplicate post found: ' . $key);
                        }
		        $postLookup[$key] = $post->ID;
                }
#                echo ' the posts are <pre>' . print_r($postLookup, true) . '</pre>';
                echo 'we have ' . count($posts) . ' fetched from WordPress<br>';
                echo 'we have ' . count($postLookup) . ' in postLookup<br>';
                


                // this is the look up array for the redirects added into the system already
                $comments = array();

                $blanks = sprintf("'%50s'", ''); // we use this for padding the filenames so the redirect URLS are aligned in the output
		preg_match_all('|<item>(.*?)</item>|is', $importdata, $this->posts);
		$this->posts = $this->posts[1];
		echo "<pre>";
		foreach ($this->posts as $post) {
			preg_match('|<title>(.*?)</title>|is', $post, $post_title);
			$post_title = str_replace(array('<![CDATA[', ']]>'), '', $wpdb->escape( trim($post_title[1]) ));

			// if we don't already have something from pubDate
			preg_match('|<dc:date>(.*?)</dc:date>|is', $post, $post_date_gmt);
			$post_date_gmt = preg_replace('|([-+])([0-9]+):([0-9]+)$|', '\1\2\3', $post_date_gmt[1]);
			$post_date_gmt = str_replace('T', ' ', $post_date_gmt);
			$post_date_gmt = strtotime($post_date_gmt);

			$post_date_gmt = gmdate('Y-m-d H:i:s', $post_date_gmt);
			$post_date = get_date_from_gmt( $post_date_gmt );

                        // by default, all posts belong to the user id = 1.
			$post_author = 1;

			preg_match('|<filename.*?>(.*?)</filename>|is', $post, $filename);
			$filename = $filename[1];

			preg_match('|<author.*?>(.*?)</author>|is', $post, $authorName);
			if ($authorName)
			{
				$authorName = $wpdb->escape(trim($authorName[1]));
				if (isset($authorNameLookup[$authorName]))
				{
				        $post_author = $authorNameLookup[$authorName];
                                }
                                else
                                {
        				if (isset($authorLoginLookup[$authorName]))
	        			{
		        		        $post_author = $authorLoginLookup[$authorName];
                                        }
                                }
                        }

			// look up the post id for this comment
			
			$key = $post_author . ' @@ ' .  $post_date . ' @@ ' . stripslashes($post_title);
			if (isset($postLookup[$key]))
			{
			        $post_ID = $postLookup[$key];
                        }
                        else
                        {
                                echo 'could not find post id for ' . $authorName . ' ' . $key . '<br>';
                                continue;
                        }
                        
                        $post_URL = get_permalink( $post_ID );
                        $post_URL = str_replace('http://' . $_SERVER['HTTP_HOST'], '', $post_URL);

                        echo "RedirectPermanent /" . str_pad($filename, 50) . "$post_URL\n";
		}
		echo "</pre>";
		exit;
	}

	function import_posts() {
		echo '<ol>';

		foreach ($this->posts as $post) {
			echo "<li>".__('Importing redirects...', 'rss-importer-redirects');

			extract($post);

			if ($post_id = post_exists($post_title, $post_content, $post_date)) {
				_e('Post already imported', 'rss-importer-redirects');
			} else {
				$post_id = wp_insert_post($post);
				if ( is_wp_error( $post_id ) )
					return $post_id;
				if (!$post_id) {
					_e('Couldn&#8217;t get post ID', 'rss-importer-redirects');
					return;
				}

				if (0 != count($categories))
					wp_create_categories($categories, $post_id);
				_e('Done!', 'rss-importer-redirects');
			}
			echo '</li>';
		}

		echo '</ol>';

	}

	function import() {
		$file = wp_import_handle_upload();
		if ( isset($file['error']) ) {
			echo $file['error'];
			return;
		}

		$this->file = $file['file'];
		$this->get_posts();
		$result = $this->import_posts();
		if ( is_wp_error( $result ) )
			return $result;
		wp_import_cleanup($file['id']);
		do_action('import_done', 'rssredirects');

		echo '<h3>';
		printf(__('All done. <a href="%s">Have fun!</a>', 'rss-importer-redirects'), get_option('home'));
		echo '</h3>';
	}

	function dispatch() {
		if (empty ($_GET['step']))
			$step = 0;
		else
			$step = (int) $_GET['step'];

		$this->header();

		switch ($step) {
			case 0 :
				$this->greet();
				break;
			case 1 :
				check_admin_referer('import-upload');
				$result = $this->import();
				if ( is_wp_error( $result ) )
					echo $result->get_error_message();
				break;
		}

		$this->footer();
	}

	function RSS_ImportRedirects() {
		// Nothing.
	}
}

$rss_import = new RSS_ImportRedirects();

register_importer('rssredirects', __('RSS REDIRECTS', 'rss-importer-redirects'), __('Import redirects from an RSS feed.', 'rss-importer-redirects'), array ($rss_import, 'dispatch'));

} // class_exists( 'WP_Importer' )

function rss_importer_redirects_init() {
    load_plugin_textdomain( 'rss-importer-redirects', false, dirname( plugin_basename( __FILE__ ) ) . '/languages' );
}
add_action( 'init', 'rss_importer_redirects_init' );