root/tags/1.3/wp-admin/includes/user.php

<?php

// Creates a new user from the "Users" form using $_POST information.
function add_user() {
    if ( func_num_args() ) { // The hackiest hack that ever did hack
        global $current_user, $wp_roles;
        $user_id = (int) func_get_arg( 0 );

        if ( isset( $_POST['role'] ) ) {
            if( $user_id != $current_user->id || $wp_roles->role_objects[$_POST['role']]->has_cap( 'edit_users' ) ) {
                $user = new WP_User( $user_id );
                $user->set_role( $_POST['role'] );
            }
        }
    } else {
        add_action( 'user_register', 'add_user' ); // See above
        return edit_user();
    }
}

function edit_user( $user_id = 0 ) {
    global $current_user, $wp_roles, $wpdb;
    if ( $user_id != 0 ) {
        $update = true;
        $user->ID = (int) $user_id;
        $userdata = get_userdata( $user_id );
        $user->user_login = $wpdb->escape( $userdata->user_login );
    } else {
        $update = false;
        $user = '';
    }

    if ( isset( $_POST['user_login'] ))
        $user->user_login = wp_specialchars( trim( $_POST['user_login'] ));

    $pass1 = $pass2 = '';
    if ( isset( $_POST['pass1'] ))
        $pass1 = $_POST['pass1'];
    if ( isset( $_POST['pass2'] ))
        $pass2 = $_POST['pass2'];

    if ( isset( $_POST['role'] ) && current_user_can( 'edit_users' ) ) {
        if( $user_id != $current_user->id || $wp_roles->role_objects[$_POST['role']]->has_cap( 'edit_users' ))
            $user->role = $_POST['role'];
    }

    if ( isset( $_POST['email'] ))
        $user->user_email = wp_specialchars( trim( $_POST['email'] ));
    if ( isset( $_POST['url'] ) ) {
        $user->user_url = clean_url( trim( $_POST['url'] ));
        $user->user_url = preg_match('/^(https?|ftps?|mailto|news|irc|gopher|nntp|feed|telnet):/is', $user->user_url) ? $user->user_url : 'http://'.$user->user_url;
    }
    if ( isset( $_POST['first_name'] ))
        $user->first_name = wp_specialchars( trim( $_POST['first_name'] ));
    if ( isset( $_POST['last_name'] ))
        $user->last_name = wp_specialchars( trim( $_POST['last_name'] ));
    if ( isset( $_POST['nickname'] ))
        $user->nickname = wp_specialchars( trim( $_POST['nickname'] ));
    if ( isset( $_POST['display_name'] ))
        $user->display_name = wp_specialchars( trim( $_POST['display_name'] ));
    if ( isset( $_POST['description'] ))
        $user->description = trim( $_POST['description'] );
    if ( isset( $_POST['jabber'] ))
        $user->jabber = wp_specialchars( trim( $_POST['jabber'] ));
    if ( isset( $_POST['aim'] ))
        $user->aim = wp_specialchars( trim( $_POST['aim'] ));
    if ( isset( $_POST['yim'] ))
        $user->yim = wp_specialchars( trim( $_POST['yim'] ));
    if ( !$update )
        $user->rich_editing = 'true';  // Default to true for new users.
    else if ( isset( $_POST['rich_editing'] ) )
        $user->rich_editing = $_POST['rich_editing'];
    else
        $user->rich_editing = 'false';

    $errors = new WP_Error();

    /* checking that username has been typed */
    if ( $user->user_login == '' )
        $errors->add( 'user_login', __( '<strong>ERROR</strong>: Please enter a username.' ));

    /* checking the password has been typed twice */
    do_action_ref_array( 'check_passwords', array ( $user->user_login, & $pass1, & $pass2 ));

    if (!$update ) {
        if ( $pass1 == '' || $pass2 == '' )
            $errors->add( 'pass', __( '<strong>ERROR</strong>: Please enter your password twice.' ));
    } else {
        if ((empty ( $pass1 ) && !empty ( $pass2 ) ) || (empty ( $pass2 ) && !empty ( $pass1 ) ) )
            $errors->add( 'pass', __( "<strong>ERROR</strong>: you typed your new password only once." ));
    }

    /* Check for "\" in password */
    if( strpos( " ".$pass1, "\\" ) )
        $errors->add( 'pass', __( '<strong>ERROR</strong>: Passwords may not contain the character "\\".' ));

    /* checking the password has been typed twice the same */
    if ( $pass1 != $pass2 )
        $errors->add( 'pass', __( '<strong>ERROR</strong>: Please type the same password in the two password fields.' ));

    if (!empty ( $pass1 ))
        $user->user_pass = $pass1;

    if ( !$update && !validate_username( $user->user_login ) )
        $errors->add( 'user_login', __( '<strong>ERROR</strong>: This username is invalid.  Please enter a valid username.' ));

    if (!$update && username_exists( $user->user_login ))
        $errors->add( 'user_login', __( '<strong>ERROR</strong>: This username is already registered, please choose another one.' ));

    /* checking e-mail address */
    if ( empty ( $user->user_email ) ) {
        $errors->add( 'user_email', __( "<strong>ERROR</strong>: please type an e-mail address" ));
    } else
        if (!is_email( $user->user_email ) ) {
            $errors->add( 'user_email', __( "<strong>ERROR</strong>: the email address isn't correct" ));
        }

    if ( $errors->get_error_codes() )
        return $errors;

    if ( $update ) {
        $user_id = wp_update_user( get_object_vars( $user ));
    } else {
        $user_id = wp_insert_user( get_object_vars( $user ));
        wp_new_user_notification( $user_id );
    }
    return $user_id;
}

function get_author_user_ids() {
    global $wpdb;
    // wpmu site admins don't have user_levels
    $level_key = $wpdb->prefix . 'capabilities';

    $query = "SELECT user_id FROM $wpdb->usermeta WHERE meta_key = '$level_key' AND meta_value != '0'";

    return $wpdb->get_col( $query );
}

function get_editable_authors( $user_id ) {
    global $wpdb;

    $editable = get_editable_user_ids( $user_id );

    if( !$editable ) {
        return false;
    } else {
        $editable = join(',', $editable);
        $authors = $wpdb->get_results( "SELECT * FROM $wpdb->users WHERE ID IN ($editable) ORDER BY display_name" );
    }

    return apply_filters('get_editable_authors', $authors);
}

function get_editable_user_ids( $user_id, $exclude_zeros = true ) {
    global $wpdb;

    $user = new WP_User( $user_id );

    if ( ! $user->has_cap('edit_others_posts') ) {
        if ( $user->has_cap('edit_posts') || $exclude_zeros == false )
            return array($user->id);
        else
            return false;
    }

    // wpmu site admins don't have user_levels
    $level_key = $wpdb->prefix . 'capabilities';

    $query = "SELECT user_id FROM $wpdb->usermeta WHERE meta_key = '$level_key'";
    if ( $exclude_zeros )
        $query .= " AND meta_value != 'a:1:{s:10:\"subscriber\";b:1;}'";

    return $wpdb->get_col( $query );
}

function get_nonauthor_user_ids() {
    global $wpdb;
    // wpmu site admins don't have user_levels
    $level_key = $wpdb->prefix . 'capabilities';

    $query = "SELECT user_id FROM $wpdb->usermeta WHERE meta_key = '$level_key' AND meta_value = '0'";

    return $wpdb->get_col( $query );
}

function get_others_unpublished_posts($user_id, $type='any') {
    global $wpdb;
    $user = get_userdata( $user_id );
    $level_key = $wpdb->prefix . 'user_level';

    $editable = get_editable_user_ids( $user_id );

    if ( in_array($type, array('draft', 'pending')) )
        $type_sql = " post_status = '$type' ";
    else
        $type_sql = " ( post_status = 'draft' OR post_status = 'pending' ) ";

    $dir = ( 'pending' == $type ) ? 'ASC' : 'DESC';

    if( !$editable ) {
        $other_unpubs = '';
    } else {
        $editable = join(',', $editable);
        $other_unpubs = $wpdb->get_results("SELECT ID, post_title, post_author FROM $wpdb->posts WHERE post_type = 'post' AND $type_sql AND post_author IN ($editable) AND post_author != '$user_id' ORDER BY post_modified $dir");
    }

    return apply_filters('get_others_drafts', $other_unpubs);
}

function get_others_drafts($user_id) {
    return get_others_unpublished_posts($user_id, 'draft');
}

function get_others_pending($user_id) {
    return get_others_unpublished_posts($user_id, 'pending');
}

function get_user_to_edit( $user_id ) {
    $user = new WP_User( $user_id );
    $user->user_login   = attribute_escape($user->user_login);
    $user->user_email   = attribute_escape($user->user_email);
    $user->user_url     = clean_url($user->user_url);
    $user->first_name   = attribute_escape($user->first_name);
    $user->last_name    = attribute_escape($user->last_name);
    $user->display_name = attribute_escape($user->display_name);
    $user->nickname     = attribute_escape($user->nickname);
    $user->aim          = attribute_escape($user->aim);
    $user->yim          = attribute_escape($user->yim);
    $user->jabber       = attribute_escape($user->jabber);
    $user->description  =  wp_specialchars($user->description);

    return $user;
}

function get_users_drafts( $user_id ) {
    global $wpdb;
    $user_id = (int) $user_id;
    $query = "SELECT ID, post_title FROM $wpdb->posts WHERE post_type = 'post' AND post_status = 'draft' AND post_author = $user_id ORDER BY post_modified DESC";
    $query = apply_filters('get_users_drafts', $query);
    return $wpdb->get_results( $query );
}

function wp_delete_user($id, $reassign = 'novalue') {
    global $wpdb;

    $id = (int) $id;
    $user = get_userdata($id);

    if ($reassign == 'novalue') {
        $post_ids = $wpdb->get_col("SELECT ID FROM $wpdb->posts WHERE post_author = $id");

        if ($post_ids) {
            foreach ($post_ids as $post_id)
                wp_delete_post($post_id);
        }

        // Clean links
        $wpdb->query("DELETE FROM $wpdb->links WHERE link_owner = $id");
    } else {
        $reassign = (int) $reassign;
        $wpdb->query("UPDATE $wpdb->posts SET post_author = {$reassign} WHERE post_author = {$id}");
        $wpdb->query("UPDATE $wpdb->links SET link_owner = {$reassign} WHERE link_owner = {$id}");
    }

    // FINALLY, delete user
    do_action('delete_user', $id);

    $wpdb->query("DELETE FROM $wpdb->usermeta WHERE user_id = $id AND meta_key = '{$wpdb->prefix}capabilities'");

    wp_cache_delete($id, 'users');
    wp_cache_delete($user->user_login, 'userlogins');

    return true;
}

function wp_revoke_user($id) {
    $id = (int) $id;

    $user = new WP_User($id);
    $user->remove_all_caps();
}

?>