site_iri: comparison web/wp-includes/capabilities.php

equal deleted inserted replaced

-:f507feede89a
+:09a1c134465b
 		if ( empty( $this->roles ) )
 			return;
 		$this->role_objects = array();
 		$this->role_names =  array();
-		foreach ( (array) $this->roles as $role => $data ) {
+		foreach ( array_keys( $this->roles ) as $role ) {
+			$this->role_objects[$role] = new WP_Role( $role, $this->roles[$role]['capabilities'] );
+			$this->role_names[$role] = $this->roles[$role]['name'];
+		}
+	}
+	/**
+	 * Reinitialize the object
+	 *
+	 * Recreates the role objects. This is typically called only by switch_to_blog()
+	 * after switching wpdb to a new blog ID.
+	 *
+	 * @since 3.5.0
+	 * @access public
+	 */
+	function reinit() {
+		// There is no need to reinit if using the wp_user_roles global.
+		if ( ! $this->use_db )
+			return;
+		global $wpdb, $wp_user_roles;
+		// Duplicated from _init() to avoid an extra function call.
+		$this->role_key = $wpdb->prefix . 'user_roles';
+		$this->roles = get_option( $this->role_key );
+		if ( empty( $this->roles ) )
+			return;
+		$this->role_objects = array();
+		$this->role_names =  array();
+		foreach ( array_keys( $this->roles ) as $role ) {
 			$this->role_objects[$role] = new WP_Role( $role, $this->roles[$role]['capabilities'] );
 			$this->role_names[$role] = $this->roles[$role]['name'];
 		}
 	}
 	 * @param string $role Role name.
 	 * @param string $cap Capability name.
 	 * @param bool $grant Optional, default is true. Whether role is capable of performing capability.
 	 */
 	function add_cap( $role, $cap, $grant = true ) {
+		if ( ! isset( $this->roles[$role] ) )
+			return;
 		$this->roles[$role]['capabilities'][$cap] = $grant;
 		if ( $this->use_db )
 			update_option( $this->role_key, $this->roles );
 	}
 	 *
 	 * @param string $role Role name.
 	 * @param string $cap Capability name.
 	 */
 	function remove_cap( $role, $cap ) {
+		if ( ! isset( $this->roles[$role] ) )
+			return;
 		unset( $this->roles[$role]['capabilities'][$cap] );
 		if ( $this->use_db )
 			update_option( $this->role_key, $this->roles );
 	}
 	 * @access public
 	 *
 	 * @param string $role Role name to look up.
 	 * @return bool
 	 */
-	function is_role( $role )
+	function is_role( $role ) {
-	{
 		return isset( $this->role_names[$role] );
 	}
 }
 /**
 	 * Retrieves the userdata and passes it to {@link WP_User::init()}.
 	 *
 	 * @since 2.0.0
 	 * @access public
 	 *
-	 * @param int|string $id User's ID
+	 * @param int|string|stdClass|WP_User $id User's ID, a WP_User object, or a user object from the DB.
 	 * @param string $name Optional. User's username
 	 * @param int $blog_id Optional Blog ID, defaults to current blog.
 	 * @return WP_User
 	 */
 	function __construct( $id = 0, $name = '', $blog_id = '' ) {
 				'user_description' => 'description',
 				'user_level' => $prefix . 'user_level',
 				$prefix . 'usersettings' => $prefix . 'user-settings',
 				$prefix . 'usersettingstime' => $prefix . 'user-settings-time',
 			);
+		}
+		if ( is_a( $id, 'WP_User' ) ) {
+			$this->init( $id->data, $blog_id );
+			return;
+		} elseif ( is_object( $id ) ) {
+			$this->init( $id, $blog_id );
+			return;
 		}
 		if ( ! empty( $id ) && ! is_numeric( $id ) ) {
 			$name = $id;
 			$id = 0;
 	 */
 	function has_prop( $key ) {
 		return $this->__isset( $key );
 	}
+	/*
+	 * Return an array representation.
+	 *
+	 * @since 3.5.0
+	 *
+	 * @return array Array representation.
+	 */
+	function to_array() {
+		return get_object_vars( $this->data );
+	}
 	/**
 	 * Set up capability object properties.
 	 *
 	 * Will set the value for the 'cap_key' property to current database table
 	 * prefix, followed by 'capabilities'. Will then check to see if the
 		if ( ! isset( $wp_roles ) )
 			$wp_roles = new WP_Roles();
 		//Filter out caps that are not role names and assign to $this->roles
 		if ( is_array( $this->caps ) )
-			$this->roles = array_filter( array_keys( $this->caps ), array( &$wp_roles, 'is_role' ) );
+			$this->roles = array_filter( array_keys( $this->caps ), array( $wp_roles, 'is_role' ) );
 		//Build $allcaps from role caps, overlay user's $caps
 		$this->allcaps = array();
 		foreach ( (array) $this->roles as $role ) {
 			$the_role = $wp_roles->get_role( $role );
 	 * @access public
 	 *
 	 * @param string $role Role name.
 	 */
 	function set_role( $role ) {
+		if ( 1 == count( $this->roles ) && $role == current( $this->roles ) )
+			return;
 		foreach ( (array) $this->roles as $oldrole )
 			unset( $this->caps[$oldrole] );
-		if ( 1 == count( $this->roles ) && $role == $this->roles[0] )
-			return;
 		if ( !empty( $role ) ) {
 			$this->caps[$role] = true;
 			$this->roles = array( $role => true );
 		} else {
 	 * @access public
 	 *
 	 * @param string $cap Capability name.
 	 */
 	function remove_cap( $cap ) {
-		if ( empty( $this->caps[$cap] ) )
+		if ( ! isset( $this->caps[$cap] ) )
 			return;
 		unset( $this->caps[$cap] );
 		update_user_meta( $this->ID, $this->cap_key, $this->caps );
 	}
 		break;
 	case 'promote_user':
 		$caps[] = 'promote_users';
 		break;
 	case 'edit_user':
+	case 'edit_users':
 		// Allow user to edit itself
-		if ( isset( $args[0] ) && $user_id == $args[0] )
+		if ( 'edit_user' == $cap && isset( $args[0] ) && $user_id == $args[0] )
 			break;
-		// Fall through
-	case 'edit_users':
 		// If multisite these caps are allowed only for super admins.
 		if ( is_multisite() && !is_super_admin( $user_id ) )
 			$caps[] = 'do_not_allow';
 		else
-			$caps[] = 'edit_users'; // Explicit due to primitive fall through
+			$caps[] = 'edit_users'; // edit_user maps to edit_users.
 		break;
 	case 'delete_post':
 	case 'delete_page':
-		$author_data = get_userdata( $user_id );
 		$post = get_post( $args[0] );
 		if ( 'revision' == $post->post_type ) {
 			$post = get_post( $post->post_parent );
 		}
 			if ( 'delete_post' == $cap )
 				$cap = $post_type->cap->$cap;
 			break;
 		}
-		if ( '' != $post->post_author ) {
+		$post_author_id = $post->post_author;
-			$post_author_data = get_userdata( $post->post_author );
-		} else {
+		// If no author set yet, default to current user for cap checks.
-			// No author set yet, so default to current user for cap checks.
+		if ( ! $post_author_id )
-			$post_author_data = $author_data;
+			$post_author_id = $user_id;
-		}
+		$post_author_data = $post_author_id == get_current_user_id() ? wp_get_current_user() : get_userdata( $post_author_id );
 		// If the user is the author...
 		if ( is_object( $post_author_data ) && $user_id == $post_author_data->ID ) {
 			// If the post is published...
 			if ( 'publish' == $post->post_status ) {
 		break;
 		// edit_post breaks down to edit_posts, edit_published_posts, or
 		// edit_others_posts
 	case 'edit_post':
 	case 'edit_page':
-		$author_data = get_userdata( $user_id );
 		$post = get_post( $args[0] );
 		if ( 'revision' == $post->post_type ) {
 			$post = get_post( $post->post_parent );
 		}
 			if ( 'edit_post' == $cap )
 				$cap = $post_type->cap->$cap;
 			break;
 		}
-		if ( '' != $post->post_author ) {
+		$post_author_id = $post->post_author;
-			$post_author_data = get_userdata( $post->post_author );
-		} else {
+		// If no author set yet, default to current user for cap checks.
-			// No author set yet, so default to current user for cap checks.
+		if ( ! $post_author_id )
-			$post_author_data = $author_data;
+			$post_author_id = $user_id;
-		}
+		$post_author_data = $post_author_id == get_current_user_id() ? wp_get_current_user() : get_userdata( $post_author_id );
-		//echo "current user id : $user_id, post author id: " . $post_author_data->ID . "<br />";
 		// If the user is the author...
 		if ( is_object( $post_author_data ) && $user_id == $post_author_data->ID ) {
 			// If the post is published...
 			if ( 'publish' == $post->post_status ) {
 				$caps[] = $post_type->cap->edit_published_posts;
 				$caps[] = $post_type->cap->edit_private_posts;
 		}
 		break;
 	case 'read_post':
 	case 'read_page':
-		$author_data = get_userdata( $user_id );
 		$post = get_post( $args[0] );
 		if ( 'revision' == $post->post_type ) {
 			$post = get_post( $post->post_parent );
 		}
 		if ( $status_obj->public ) {
 			$caps[] = $post_type->cap->read;
 			break;
 		}
-		if ( '' != $post->post_author ) {
+		$post_author_id = $post->post_author;
-			$post_author_data = get_userdata( $post->post_author );
-		} else {
+		// If no author set yet, default to current user for cap checks.
-			// No author set yet, so default to current user for cap checks.
+		if ( ! $post_author_id )
-			$post_author_data = $author_data;
+			$post_author_id = $user_id;
-		}
+		$post_author_data = $post_author_id == get_current_user_id() ? wp_get_current_user() : get_userdata( $post_author_id );
 		if ( is_object( $post_author_data ) && $user_id == $post_author_data->ID )
 			$caps[] = $post_type->cap->read;
 		elseif ( $status_obj->private )
 			$caps[] = $post_type->cap->read_private_posts;
 		else
 			$caps = map_meta_cap( 'edit_post', $user_id, $post->ID );
+		break;
+	case 'publish_post':
+		$post = get_post( $args[0] );
+		$post_type = get_post_type_object( $post->post_type );
+		$caps[] = $post_type->cap->publish_posts;
 		break;
 	case 'edit_post_meta':
 	case 'delete_post_meta':
 	case 'add_post_meta':
 		$post = get_post( $args[0] );
 			$caps[] = $cap;
 		break;
 	case 'edit_files':
 	case 'edit_plugins':
 	case 'edit_themes':
-		if ( defined('DISALLOW_FILE_EDIT') && DISALLOW_FILE_EDIT ) {
+		// Disallow the file editors.
+		if ( defined( 'DISALLOW_FILE_EDIT' ) && DISALLOW_FILE_EDIT )
 			$caps[] = 'do_not_allow';
-			break;
+		elseif ( defined( 'DISALLOW_FILE_MODS' ) && DISALLOW_FILE_MODS )
-		}
+			$caps[] = 'do_not_allow';
-		// Fall through if not DISALLOW_FILE_EDIT.
+		elseif ( is_multisite() && ! is_super_admin( $user_id ) )
+			$caps[] = 'do_not_allow';
+		else
+			$caps[] = $cap;
+		break;
 	case 'update_plugins':
 	case 'delete_plugins':
 	case 'install_plugins':
 	case 'update_themes':
 	case 'delete_themes':
 	case 'install_themes':
 	case 'update_core':
-		// Disallow anything that creates, deletes, or edits core, plugin, or theme files.
+		// Disallow anything that creates, deletes, or updates core, plugin, or theme files.
 		// Files in uploads are excepted.
-		if ( defined('DISALLOW_FILE_MODS') && DISALLOW_FILE_MODS ) {
+		if ( defined( 'DISALLOW_FILE_MODS' ) && DISALLOW_FILE_MODS )
 			$caps[] = 'do_not_allow';
-			break;
+		elseif ( is_multisite() && ! is_super_admin( $user_id ) )
-		}
+			$caps[] = 'do_not_allow';
-		// Fall through if not DISALLOW_FILE_MODS.
+		else
+			$caps[] = $cap;
+		break;
+	case 'activate_plugins':
+		$caps[] = $cap;
+		if ( is_multisite() ) {
+			// update_, install_, and delete_ are handled above with is_super_admin().
+			$menu_perms = get_site_option( 'menu_items', array() );
+			if ( empty( $menu_perms['plugins'] ) )
+				$caps[] = 'manage_network_plugins';
+		}
+		break;
 	case 'delete_user':
 	case 'delete_users':
-		// If multisite these caps are allowed only for super admins.
+		// If multisite only super admins can delete users.
-		if ( is_multisite() && !is_super_admin( $user_id ) ) {
+		if ( is_multisite() && ! is_super_admin( $user_id ) )
 			$caps[] = 'do_not_allow';
-		} else {
+		else
-			if ( 'delete_user' == $cap )
+			$caps[] = 'delete_users'; // delete_user maps to delete_users.
-				$cap = 'delete_users';
-			$caps[] = $cap;
-		}
 		break;
 	case 'create_users':
 		if ( !is_multisite() )
 			$caps[] = $cap;
 		elseif ( is_super_admin() || get_site_option( 'add_new_users' ) )
 			$caps[] = $cap;
 		else
 			$caps[] = 'do_not_allow';
 		break;
+	case 'manage_links' :
+		if ( get_option( 'link_manager_enabled' ) )
+			$caps[] = $cap;
+		else
+			$caps[] = 'do_not_allow';
+		break;
 	default:
 		// Handle meta capabilities for custom post types.
 		$post_type_meta_caps = _post_type_meta_capabilities();
 		if ( isset( $post_type_meta_caps[ $cap ] ) ) {
 			$args = array_merge( array( $post_type_meta_caps[ $cap ], $user_id ), $args );
 * @param int $blog_id Blog ID
 * @param string $capability Capability or role name.
 * @return bool
 */
 function current_user_can_for_blog( $blog_id, $capability ) {
+	if ( is_multisite() )
+		switch_to_blog( $blog_id );
 	$current_user = wp_get_current_user();
 	if ( empty( $current_user ) )
 		return false;
-	// Create new object to avoid stomping the global current_user.
-	$user = new WP_User( $current_user->ID) ;
-	// Set the blog id. @todo add blog id arg to WP_User constructor?
-	$user->for_blog( $blog_id );
 	$args = array_slice( func_get_args(), 2 );
 	$args = array_merge( array( $capability ), $args );
-	return call_user_func_array( array( &$user, 'has_cap' ), $args );
+	$can = call_user_func_array( array( $current_user, 'has_cap' ), $args );
+	if ( is_multisite() )
+		restore_current_blog();
+	return $can;
 }
 /**
 * Whether author of supplied post has capability or role.
 *
 */
 function author_can( $post, $capability ) {
 	if ( !$post = get_post($post) )
 		return false;
-	$author = new WP_User( $post->post_author );
+	$author = get_userdata( $post->post_author );
-	if ( empty( $author->ID ) )
+	if ( ! $author )
 		return false;
 	$args = array_slice( func_get_args(), 2 );
 	$args = array_merge( array( $capability ), $args );
-	return call_user_func_array( array( &$author, 'has_cap' ), $args );
+	return call_user_func_array( array( $author, 'has_cap' ), $args );
 }
 /**
 * Whether a particular user has capability or role.
 *
 * @param string $capability Capability or role name.
 * @return bool
 */
 function user_can( $user, $capability ) {
 	if ( ! is_object( $user ) )
-		$user = new WP_User( $user );
+		$user = get_userdata( $user );
 	if ( ! $user || ! $user->exists() )
 		return false;
 	$args = array_slice( func_get_args(), 2 );
 	$args = array_merge( array( $capability ), $args );
-	return call_user_func_array( array( &$user, 'has_cap' ), $args );
+	return call_user_func_array( array( $user, 'has_cap' ), $args );
 }
 /**
 * Retrieve role object.
 *
 *
 * @param int $user_id (Optional) The ID of a user. Defaults to the current user.
 * @return bool True if the user is a site admin.
 */
 function is_super_admin( $user_id = false ) {
-	if ( $user_id )
+	if ( ! $user_id || $user_id == get_current_user_id() )
-		$user = new WP_User( $user_id );
+		$user = wp_get_current_user();
 	else
-		$user = wp_get_current_user();
+		$user = get_userdata( $user_id );
-	if ( ! $user->exists() )
+	if ( ! $user || ! $user->exists() )
 		return false;
 	if ( is_multisite() ) {
 		$super_admins = get_super_admins();
 		if ( is_array( $super_admins ) && in_array( $user->user_login, $super_admins ) )

changeset 204	09a1c134465b
parent 194	32102edaa81b