Chris Meller's WP Migration Script

<?php

    // config info
    $import = array(
        'wp_host' => 'localhost',    // hostname
        'wp_user' => 'wordpress',    // username
        'wp_pass' => 'wordpress',    // password
        'wp_name' => 'wordpress',    // name of the database
        'wp_prefix' => 'wp_',        // table prefix
    );

    echo '<pre>';

    // keep habari from executing
    define( 'UNIT_TEST', true );

    // bootstrap it
    include( 'index.php' );

    // create a connection to our wordpress database
    try {
        $wpdb = DatabaseConnection::ConnectionFactory( "mysql:host=" . $import['wp_host'] . ";dbname=" . $import['wp_name'] );
        $wpdb->connect( "mysql:host=" . $import['wp_host'] . ";dbname=" . $import['wp_name'], $import['wp_user'], $import['wp_pass'] );
    }
    catch( Exception $e ) {
        die('Unable to connect to WordPress database. ' . $e->getMessage());
    }


    // users

    $habari_users = Users::get();    // get all the habari users
    $wp_users = array();

    foreach ( $habari_users as $habari_user ) {

        // see if the user exists already in WordPress
        $wp_user = $wpdb->get_row( 'select id, user_login from ' . $import['wp_prefix'] . 'users where user_login = ?', array( $habari_user->username ) );

        // if it doesn't, create it
        if ( !$wp_user ) {

            $wpdb->query( 'insert into ' . $import['wp_prefix'] . 'users (
                    user_login,
                    user_pass,
                    user_nicename,
                    user_email,
                    user_registered,
                    display_name
                ) values ( ?, ?, ?, ?, UTC_TIMESTAMP(), ? ) ', array(
                    $habari_user->username,
                    $habari_user->password,
                    $habari_user->username,
                    $habari_user->email,
                    $habari_user->info->displayname
                )
            );

            $wp_users[ $habari_user->username ] = $wpdb->last_insert_id();

            echo 'Created user ' . $habari_user->username . ' with id ' . $wpdb->last_insert_id() . "\n";

        }
        else {

            echo 'Found existing user ' . $wp_user->user_login . ' with id ' . $wp_user->id . "\n";

            $wp_users[ $wp_user->user_login ] = $wp_user->id;

        }

    }


    // posts and pages

    // get the total number of posts, either published or draft
    $total_posts = Posts::get( array( 'count' => true, 'ignore_permissions' => true, 'content_type' => array( 'entry', 'page' ), 'status' => array( 'published', 'draft' ) ) );
    $wp_posts = array();

    echo 'Total Posts: ' . $total_posts . "\n";

    for ( $i = 0; $i < $total_posts / 10; $i++ ) {

        $posts = Posts::get( array( 'limit' => 10, 'offset' => $i * 10, 'ignore_permissions' => true, 'content_type' => array( 'entry', 'page' ), 'status' => array( 'published', 'draft' ) ) );

        echo 'Got ' . count( $posts ) . ' posts' . "\n";

        foreach ( $posts as $post ) {

            $insert_query = 'insert into ' . $import['wp_prefix'] . 'posts (
                post_author,
                post_date,
                post_date_gmt,
                post_content,
                post_title,
                post_status,
                comment_status,
                post_name,
                post_modified,
                post_modified_gmt,
                guid,
                post_type
            ) values (
                ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?
            )';

            $insert_params = array(
                $wp_users[ $post->author->username ],                    // post author ID, converted to the WordPress user's
                $post->pubdate->format( 'Y-m-d H-i-s' ),
                gmdate( 'Y-m-d H-i-s', $post->pubdate->int ),
                $post->content,
                $post->title,
                Post::status_name( $post->status ) == 'published' ? 'publish' : Post::status_name( $post->status ),                        // post status (published or draft, basically), converted to a string
                $post->info->comments_disabled ? 'closed' : 'open',
                $post->slug,
                $post->modified->format( 'Y-m-d H-i-s' ),
                gmdate( 'Y-m-d H-i-s', $post->modified->int ),
                $post->guid,
                Post::type_name( $post->content_type ) == 'entry' ? 'post' : 'page'
            );

            $result = $wpdb->query( $insert_query, $insert_params );

            if ( !$result ) {
                echo 'Failed to insert post ' . $post->slug . "\n";
            }
            else {
                $wp_posts[ $post->id ] = $wpdb->last_insert_id();
            }

        }

    }

    // tags

    // here we don't use the habari API because it would be cumbersome - if only it were as robust as the Posts API is...
    $total_tags = DB::get_value( 'select count(*) from {tags}' );
    $wp_tax = array();

    echo 'Total Tags: ' . $total_tags . "\n";

    for ( $i = 0; $i < $total_tags / 10; $i++ ) {

        $tags = DB::get_results( 'select id, tag_text, tag_slug from {tags} order by id limit 10 offset ' . $i * 10 );

        echo 'Got ' . count( $tags ) . ' tags' . "\n";

        foreach ( $tags as $tag ) {

            // first, see if it already exists as a term in WordPress
            $wp_tag = $wpdb->get_row( 'select term_id, name, slug from ' . $import['wp_prefix'] . 'terms where slug = ?', array( $tag->tag_slug ) );

            if ( !$wp_tag ) {

                $insert_query = 'insert into ' . $import['wp_prefix'] . 'terms ( name, slug ) values ( ?, ? )';
                $insert_params = array( $tag->tag_text, $tag->tag_slug );

                $wpdb->query( $insert_query, $insert_params );

                //$wp_tags[ $tag->tag_slug ] = $wpdb->last_insert_id();

                $tag_id = $wpdb->last_insert_id();

            }
            else {

                //$wp_tags[ $wp_tag->slug ] = $wp_tag->term_id;

                $tag_id = $wp_tag->term_id;

            }

            // and either way, make sure it's actually specified as a part of the tag taxonomy
            $wp_taxonomy = $wpdb->get_row( 'select term_taxonomy_id from ' . $import['wp_prefix'] . 'term_taxonomy where taxonomy = ? and term_id = ?', array( 'post_tag', $tag_id ) );

            if ( !$wp_taxonomy ) {

                $insert_query = 'insert into ' . $import['wp_prefix'] . 'term_taxonomy ( term_id, taxonomy ) values ( ?, ? )';
                $insert_params = array( $tag_id, 'post_tag' );

                $wpdb->query( $insert_query, $insert_params );

                $wp_tax[ $tag->id ] = $wpdb->last_insert_id();

            }
            else {

                $wp_tax[ $tag->id ] = $wp_taxonomy->term_taxonomy_id;

            }

        }

    }

    // now we have to link all our tags and posts

    $total_joins = DB::get_value( 'select count(*) from {tag2post}' );

    echo 'Total Tag to Post relationships: ' . $total_joins . "\n";

    for ( $i = 0; $i < $total_joins / 10; $i++ ) {

        // get the joins
        $joins = DB::get_results( 'select tag_id, post_id from {tag2post} order by tag_id, post_id limit 10 offset ' . $i * 10 );

        echo 'Got ' . count( $joins ) . ' relationships' . "\n";

        foreach ( $joins as $join ) {

            // if it's not in the list of posts we imported earlier, skip it - it's probably a different content type
            if ( !array_key_exists( $join->post_id, $wp_posts ) ) {
                continue;
            }

            $insert_query = 'insert into ' . $import['wp_prefix'] . 'term_relationships ( object_id, term_taxonomy_id ) values ( ?, ? )';
            $insert_params = array( $wp_posts[ $join->post_id ], $wp_tax[ $join->tag_id ] );

            $wpdb->query( $insert_query, $insert_params );

        }

    }


    // update the tag counts
    $wpdb->query( 'update ' . $import['wp_prefix'] . 'term_taxonomy t set count = ( select count(object_id) from ' . $import['wp_prefix'] . 'term_relationships where term_taxonomy_id = t.term_taxonomy_id )');


    // comments

    $total_comments = DB::get_value ( 'select count(*) from {comments}' );

    echo 'Total comments: ' . $total_comments . "\n";

    // get the comment types from habari
    $comment_types = Comment::list_comment_types();
    $comment_statuses = Comment::list_comment_statuses();

    for ( $i = 0; $i < $total_comments / 10; $i++ ) {

        // get the comments
        $comments = DB::get_results( 'select post_id, name, email, url, ip, date, content, status, type from {comments} order by id limit 10 offset ' . $i * 10 );

        echo 'Got ' . count( $comments ) . ' comments' . "\n";

        foreach ( $comments as $comment ) {

            $comment->date = HabariDateTime::date_create( $comment->date );

            $comment_status = $comment_statuses[ $comment->status ];
            $comment_type = $comment_types[ $comment->type ];

            if ( $comment_status == 'approved' ) {
                $comment_approved = true;
            }
            else {
                $comment_approved = false;
            }

            $insert_query = 'insert into ' . $import['wp_prefix'] . 'comments (
                    comment_post_id,
                    comment_author,
                    comment_author_email,
                    comment_author_url,
                    comment_author_ip,
                    comment_date,
                    comment_date_gmt,
                    comment_content,
                    comment_approved,
                    comment_type
                ) values ( ?, ?, ?, ?, ?, ?, ?, ?, ?, ? )';

            $insert_params = array(
                $wp_posts[ $comment->post_id ],
                $comment->name,
                $comment->email,
                $comment->url,
                long2ip( $comment->ip ),
                $comment->date->format( 'Y-m-d H-i-s' ),
                gmdate( 'Y-m-d H-i-s', $comment->date->int ),
                $comment->content,
                $comment_approved,
                $comment_type
            );

            $wpdb->query( $insert_query, $insert_params );

        }

    }

    // update comment user_id links based on user display names
    $wpdb->query( 'update ' . $import['wp_prefix'] . 'comments c set user_id = ( select ID from ' . $import['wp_prefix'] . 'users where display_name = c.comment_author ) where comment_author in ( select distinct display_name from ' . $import['wp_prefix'] . 'users )');

    // update post comment counts
    $wpdb->query( 'update ' . $import['wp_prefix'] . 'posts p set comment_count = ( select count(*) from ' . $import['wp_prefix'] . 'comments where comment_post_ID = p.ID and comment_approved = \'1\' )' );

    echo '</pre>';

?>