<?php
/*
Plugin Name: Redirect users to different locations after logging in
Plugin URI: http://www.theblog.ca/wplogin-front-page
Description: Redirect users to different locations after logging in. Define a set of rules for specific users, user with specific roles, users with specific capabilities, and a blanket rule for all other users.
Author: Peter
Version: 1.1
Author URI: http://www.theblog.ca
*/

/*
--------------
Config section
--------------
*/

/*
--------------
Redirect specific usernames to different URLs

Example:
$redirect_per_user['admin'] = 'http://yoursite.com/specialadminpage.php';
--------------
*/

$redirect_per_user = array(); // Do not touch this line. Enter rules below this line

/*
--------------
Redirect specific user roles to different URLs

Example:
$redirect_by_role['contributor'] = 'http://yoursite.com/specialcontribpage.php';

Note that the first rule that matches is applied
Default WordPress roles are administrator, editor, author, contributor, subscriber
--------------
*/

$redirect_by_role = array(); // Do not touch this line. Enter rules below this line

/*
--------------
Redirect specific user levels to different URLs

Example:
$redirect_by_level['level_8'] = 'http://yoursite.com/specialadminpage.php';

Note that the first rule that matches is applied
See http://codex.wordpress.org/Roles_and_Capabilities#Capabilities for a full list of capabilities
--------------
*/

$redirect_by_level = array(); // Do not touch this line. Enter rules below this line


/* 
--------------
Redirect all users who don't match the rules above to a specific URL

Example:
$redirect_all_users = get_option('siteurl');

Leave as false to direct them all to the admin section is per usual
--------------
*/

$redirect_all_users = false;


/*
------------------------
Code begins, config over
------------------------
*/


// Thanks to http://wordpress.org/support/topic/97314 for this function

function redirect_current_user_can($capability, $current_user) {
    global $wpdb;

    $roles = get_option($wpdb->prefix . 'user_roles');
    $user_roles = $current_user->{$wpdb->prefix . 'capabilities'};
    $user_roles = array_keys($user_roles, true);
    $role = $user_roles[0];
    $capabilities = $roles[$role]['capabilities'];

    if ( in_array( $capability, array_keys( $capabilities, true) ) ) {
        // check array keys of capabilities for match against requested capability
        return true;
    }
    return false;
}

function redirect_to_front_page( $user_login ) {
    global $wpdb, $redirect_to, $redirect_per_user, $redirect_by_role, $redirect_by_level, $redirect_all_users;

    if ( isset( $redirect_per_user[$user_login] ) ) {
        $redirect_to = $redirect_per_user[$user_login];
    }
    else {
        $user = get_userdatabylogin($user_login);

        if ($redirect_by_role) {
            foreach ( $redirect_by_role as $role => $redirect_url ) {
                if ( isset ( $user->{$wpdb->prefix . 'capabilities'}[$role] ) ) {
                    $redirect_to = $redirect_url;
                    return false;
                }
            }
        }
        
        if ($redirect_by_level) {
            foreach ( $redirect_by_level as $level => $redirect_url ) {
                if ( redirect_current_user_can ( $level, $user ) ) {
                    $redirect_to = $redirect_url;
                    return false;
                }
            }
        }

        if ($redirect_all_users) {
            $redirect_to = $redirect_all_users;
        }
    }
}

add_action('wp_login', 'redirect_to_front_page');
?>