Fix users with no role – data migration plugin

So you’ve migrated a bunch of users over from another system and then, horror, you discover that they don’t have roles in WordPress. The WP Users admin screen doesn’t cope with this issue (why should it), so the only option would be to go through and manually select the roles for each user with no role… fine, right? Except I just migrated 8,000 ish users and there’s no way my meta carpels will take that abuse. This plugin finds any users on your system with no role and gives them the role of ‘subscriber’, it does this in batches of 1,000 users per page load and once it has finished it replaces your entire blog with a message saying it’s finished… so don’t leave it running unattended kids, eh? (I know that’s rude, but coding “echo ‘blah; exit;’ is SO much quicker and it’s only a dirty data migration tool.

You can view the source here:


/*
Plugin Name: Fix No Role Users
Plugin URI: http://simonwheatley.co.uk/wordpress/fox-no-role-users/
Description: Finds users with no role and makes them subscribers.
Version: 1
Author: Simon Wheatley
Author URI: http://www.simonwheatley.co.uk/wordpress/
*/

/*  Copyright 2009 Simon Wheatley

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

*/

/**
 *
 * @package default
 * @author Simon Wheatley
 **/
class FixNoRoleUsers
{
    protected $db;

    public function __construct()
    {
        global $wpdb;
        $this->db = & $wpdb;
        add_action( 'init', array( & $this, 'fix_roles' ) );
    }
    
    public function fix_roles()
    {
        // Get some more user IDs
        $user_ids = $this->no_role_user_ids();
        if ( ! count( $user_ids ) ) $this->finished();
        foreach ( $user_ids as $user_id ) {
            $user = new WP_User( $user_id );
            $user->add_role( 'subscriber' );
            error_log( "Given user $user->user_login the role of subscriber" );
            unset( $user );
        }
    }

    protected function no_role_user_ids()
    {
        $users = $this->db->users;
        $usermeta = $this->db->usermeta;
        // Pretty horrible query, but who cares as it's not for production usage
        // Grab groups of 1,000 for now
        $sql = " SELECT ID FROM $users WHERE ID NOT IN ( SELECT user_id FROM $usermeta WHERE meta_key = 'wp_capabilities' ) LIMIT 1000; "; 
        // No need to prepare as no user input
        return $this->db->get_col( $sql );
    }
    
    protected function finished()
    {
        // OK. Bung a rude message up and exit.
        echo "All users now have roles.";
        exit;
    }
}

/**
 * Instantiate the plugin
 *
 * @global
 **/

$fix_no_role_users = new FixNoRoleUsers();

?>

Migration to bbPress – fix topic slugs

So I’ve just had to migrate a whole load of data to bbPress, and unfortunately the fairly old migration tool I was using (while generally excellent) didn’t bring across the topic-slugs, meaning I couldn’t have pretty permalinks in bbPress. So I’ve written a quick plugin which creates the topic slugs, it does this by doing 1000 at a time every time the page loads, then it very rudely bangs up a big text notice telling you it’s done and you should remove the plugin… at this point bbPress won’t work until you remove the plugin. You have been warned. I couldn’t be bothered with polite messages as this is a one use migration tool. :)

Updated version, this one also create forum slugs:

<?php
/*
Plugin Name: Create Slugs
Description:  Creates topic and forum slugs where previously there weren't any. Outputs into error log, only works on plugin activation.
Plugin URI:  http://simonwheatley.co.uk/bbpress/create-slugs
Author: simonwheatley
Author URI: http://simonwheatley.co.uk
Version: 1.1
*/

class CreateSlugs
{
    protected $topics_have_slugs;
    protected $forums_have_slugs;
    
    function __construct()
    {
        global $bbdb;
        $this->db = & $bbdb;
        $this->topics_have_slugs = false;
        $this->forums_have_slugs = false;
    }
    
    public function create_slugs()
    {
        $this->create_topic_slugs();
        if ( ! $this->topics_have_slugs ) return;
        $this->create_forum_slugs();
        if ( ! $this->forums_have_slugs ) return;
        // Finished!
        echo "Everything now has slugs, you can disable the create topic slugs plugin by removing it from the plugins directory.";
        exit;
    }
    
    protected function maybe_finished()
    {
//        if ( ! $this->forums_have_slugs ) return;
    }
    
    protected function create_topic_slugs()
    {
        error_log( "TOPICS" );
        $topics = $this->slugless( 'topics' );
        if ( ! count( $topics ) ) $this->topics_have_slugs = true;
        foreach( $topics AS $topic ) {
            $new_slug = $this->generate_slug( 'topic', $topic->topic_id, $topic->topic_title );
            $this->update_topic_slug( $topic->topic_id, $new_slug );
            error_log( "$new_slug >> $topic->topic_title" );
        }
        error_log( "Last Query: " . $this->db->last_query );
    }
    
    protected function create_forum_slugs()
    {
        error_log( "FORUMS" );
        $forums = $this->slugless( 'forums' );
        if ( ! count( $forums ) ) $this->forums_have_slugs = true;
        foreach( $forums AS $forum ) {
            $new_slug = $this->generate_slug( 'forum', $forum->forum_id, $forum->forum_name );
            $this->update_forum_slug( $forum->forum_id, $new_slug );
            error_log( "$new_slug >> $forum->forum_name" );
        }
    }

    protected function slugless( $type )
    {
        // Just do 1,000 at a time for now
        if ( $type == 'topics' ) {
            $topics_table = $this->db->topics;
            $sql = " SELECT topic_id, topic_title FROM $topics_table WHERE topic_slug IS NULL OR topic_slug = '' LIMIT 1000 ";
        }
        if ( $type == 'forums' ) {
            $forums_table = $this->db->forums;
            $sql = " SELECT forum_id, forum_name FROM $forums_table WHERE forum_slug IS NULL OR forum_slug = '' LIMIT 1000 ";
        }
        // No need to prepared the SQL, no user input
        return $this->db->get_results( $sql );
    }
    
    protected function generate_slug( $type, $id, $title )
    {
        $slug = $_slug = bb_slug_sanitize( wp_specialchars_decode( $title, ENT_QUOTES ) );
        if ( strlen( $_slug ) < 1 ) $slug = $_slug = '0';

        // Ensure the slug is unique
        while ( is_numeric( $slug ) || $existing_slug = $this->existing_slug( $type, $id, $slug ) ) {
            $slug = bb_slug_increment( $_slug, $existing_slug );
        }
        return $slug;
    }
    
    protected function existing_slug( $type, $id, $slug )
    {
        if ( $type == 'topic' ) {
            $table = $this->db->topics;
            $sql = "SELECT topic_slug FROM $table WHERE topic_slug = %s AND topic_id != %d";
        }
        if ( $type == 'forum' ) {
            $table = $this->db->forums;
            $sql = "SELECT forum_slug FROM $table WHERE forum_slug = %s AND forum_id != %d";
        }
        $prepared_sql = $this->db->prepare( $sql, $slug, $id );
        return $this->db->get_var( $prepared_sql );
    }
    
    protected function update_topic_slug( $topic_id, $topic_slug )
    {
        $table = $this->db->topics;
        $data = array( 'topic_slug' => $topic_slug );
        $where = array( 'topic_id' => $topic_id );
        $this->db->update( $table, $data, $where );
    }
    
    protected function update_forum_slug( $forum_id, $forum_slug )
    {
        $table = $this->db->forums;
        $data = array( 'forum_slug' => $forum_slug );
        $where = array( 'forum_id' => $forum_id );
        $this->db->update( $table, $data, $where );
    }

}

$create_slugs = new CreateSlugs();

add_action( 'bb_init', array( & $create_slugs, 'create_slugs' ) );

?>