2007-05-25 03:16:21 -04:00
< ? 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 ;
$level_key = $wpdb -> prefix . 'user_level' ;
$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 ;
}
$level_key = $wpdb -> prefix . 'user_level' ;
$query = " SELECT user_id FROM $wpdb->usermeta WHERE meta_key = ' $level_key ' " ;
if ( $exclude_zeros )
$query .= " AND meta_value != '0' " ;
return $wpdb -> get_col ( $query );
}
function get_nonauthor_user_ids () {
global $wpdb ;
$level_key = $wpdb -> prefix . 'user_level' ;
$query = " SELECT user_id FROM $wpdb->usermeta WHERE meta_key = ' $level_key ' AND meta_value = '0' " ;
return $wpdb -> get_col ( $query );
}
2007-06-14 12:24:28 -04:00
function get_others_unpublished_posts ( $user_id , $type = 'any' ) {
2007-05-25 03:16:21 -04:00
global $wpdb ;
$user = get_userdata ( $user_id );
$level_key = $wpdb -> prefix . 'user_level' ;
$editable = get_editable_user_ids ( $user_id );
2007-06-14 12:24:28 -04:00
if ( in_array ( $type , array ( 'draft' , 'pending' )) )
$type_sql = " post_status = ' $type ' " ;
else
$type_sql = " ( post_status = 'draft' OR post_status = 'pending' ) " ;
2007-06-24 01:58:38 -04:00
$dir = ( 'pending' == $type ) ? 'ASC' : 'DESC' ;
2007-05-25 03:16:21 -04:00
if ( ! $editable ) {
2007-06-14 12:24:28 -04:00
$other_unpubs = '' ;
2007-05-25 03:16:21 -04:00
} else {
$editable = join ( ',' , $editable );
2007-06-24 01:58:38 -04:00
$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 " );
2007-05-25 03:16:21 -04:00
}
2007-06-14 12:24:28 -04:00
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' );
2007-05-25 03:16:21 -04:00
}
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 ;
2007-06-24 01:58:38 -04:00
$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 " ;
2007-05-25 03:16:21 -04:00
$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->users WHERE ID = $id " );
$wpdb -> query ( " DELETE FROM $wpdb->usermeta WHERE user_id = ' $id ' " );
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 ();
}
?>