Trac/43421

src/wp-includes/capabilities.php

@@ -1021,11 +1021,13 @@ function get_role( $role ) {
  * Adds a role, if it does not exist.
  *
  * @since 2.0.0
+ * @since x.y.z Support an array of strings for the capabilities array.
  *
  * @param string $role         Role name.
  * @param string $display_name Display name for role.
- * @param bool[] $capabilities List of capabilities keyed by the capability name,
+ * @param bool[]|string[] $capabilities List of capabilities keyed by the capability name,
  *                             e.g. array( 'edit_posts' => true, 'delete_posts' => false ).
+ *                             Also supports an array of capabilities assumed to be granted.
  * @return WP_Role|void WP_Role object, if the role is added.
  */
 function add_role( $role, $display_name, $capabilities = array() ) {

src/wp-includes/class-wp-roles.php

@@ -147,11 +147,13 @@ public function reinit() {
 	 * To explicitly deny the role a capability, set the value for that capability to false.
 	 *
 	 * @since 2.0.0
+	 * @since x.y.z Support an array of strings for the capabilities array.
 	 *
 	 * @param string $role         Role name.
 	 * @param string $display_name Role display name.
-	 * @param bool[] $capabilities Optional. List of capabilities keyed by the capability name,
-	 *                             e.g. `array( 'edit_posts' => true, 'delete_posts' => false )`.
+	 * @param bool[]|string[] $capabilities List of capabilities keyed by the capability name,
+	 *                             e.g. array( 'edit_posts' => true, 'delete_posts' => false ).
+	 *                             Also supports an array of capabilities assumed to be granted.
 	 *                             Default empty array.
 	 * @return WP_Role|void WP_Role object, if the role is added.
 	 */
@@ -160,6 +162,10 @@ public function add_role( $role, $display_name, $capabilities = array() ) {
 			return;
 		}
 
+		if ( wp_is_numeric_array( $capabilities ) ) {
+			$capabilities = array_fill_keys( $capabilities, true );
+		}
+
 		$this->roles[ $role ] = array(
 			'name'         => $display_name,
 			'capabilities' => $capabilities,

tests/phpunit/tests/user/capabilities.php

@@ -991,6 +991,27 @@ public function test_add_role() {
 		$this->assertFalse( $wp_roles->is_role( $role_name ) );
 	}
 
+	/**
+	 * Test add_role with implied capabilities grant successfully grants capabilities.
+	 */
+	public function test_add_role_with_single_level_capabilities() {
+		$role_name = 'janitor';
+		add_role(
+			$role_name,
+			'Janitor',
+			array(
+				'level_1',
+			)
+		);
+		$this->flush_roles();
+
+		// Assign a user to that role.
+		$id   = self::factory()->user->create( array( 'role' => $role_name ) );
+		$user = new WP_User( $id );
+
+		$this->assertTrue( $user->has_cap( 'level_1' ) );
+	}
+
 	/**
 	 * Change the capabilities associated with a role and make sure the change
 	 * is reflected in has_cap().