Changeset 4289

Timestamp:

11/02/2012 05:26:26 AM (12 years ago)

Author:

johnjamesjacoby

Message:

Capabilities:

• Introduce bbp_set_user_role() and add $context parameter to bbp_get_user_role().
• See #1939.

File:

• trunk/includes/users/functions.php (modified) (2 diffs)

trunk/includes/users/functions.php

@@ -1380 +1380 @@
  * @return string
  */
-function bbp_get_user_role( $user_id = 0 ) {
+function bbp_set_user_role( $user_id = 0, $new_role = '', $context = '' ) {
 
     // Validate user id
@@ -1391 +1391 @@
 
     // Get the user's main role
-    $role = isset( $user->roles ) ? array_shift( $user->roles ) : '';
-
-    return apply_filters( 'bbp_get_user_role', $role, $user_id, $user );
+    switch ( strtolower( $context ) ) {
+
+        // bbPress context
+        case 'forum'     :
+        case 'forums'    :
+        case 'bbpress'   :          
+            $forum_roles = array_intersect( array_values( $user->roles ), array_keys( bbp_get_forums_editable_roles() ) );
+
+            foreach ( $forum_roles as $forum_role ) {
+                $user->remove_role( $forum_role );
+            }
+
+            $user->add_role( $new_role );
+
+            break;
+
+        // WordPress core context
+        case ''          :
+        case 'blog'      :
+        case 'wordpress' :
+        default          :
+            $blog_roles = array_intersect( array_values( $user->roles ), array_keys( get_editable_roles() ) );
+
+            foreach ( $blog_roles as $blog_role ) {
+                $user->remove_role( $blog_role );
+            }
+
+            $user->add_role( $new_role );
+
+            break;
+    }
+
+    return apply_filters( 'bbp_get_user_role', $new_role, $context, $user_id, $user );
+}
+
+/**
+ * Return a user's main role
+ *
+ * @since bbPress (r3860)
+ *
+ * @param int $user_id
+ * @uses bbp_get_user_id() To get the user id
+ * @uses get_userdata() To get the user data
+ * @uses apply_filters() Calls 'bbp_get_user_role' with the
+ *                        role and user id
+ * @return string
+ */
+function bbp_get_user_role( $user_id = 0, $context = '' ) {
+
+    // Validate user id
+    $user_id = bbp_get_user_id( $user_id, false, false );
+    if ( empty( $user_id ) )
+        return false;
+
+    // Get userdata
+    $user = get_userdata( $user_id );
+
+    // User has no roles so set as empty string
+    if ( empty( $user->roles ) ) {
+
+        // Assume the anonymous role if looking for a bbPress role
+        if ( in_array( $context, array( 'forum', 'forums', 'bbpress' ) ) ) {
+            $role = bbp_get_spectator_role();
+
+        // Otherwise trust that there is no role
+        } else {
+            $role = '';
+        }
+
+    // User has roles so lets
+    } else {
+
+        // Get the user's main role
+        switch ( strtolower( $context ) ) {
+
+            // bbPress context
+            case 'forum'     :
+            case 'forums'    :
+            case 'bbpress'   :
+                $role = array_intersect( array_values( $user->roles ), array_keys( bbp_get_forums_editable_roles() ) );
+
+                // If there's a role in the array, use the first one
+                if ( !empty( $role ) ) {
+                    $role = array_shift( array_values( $role ) );
+
+                // Otherwise, give them the anonymous role
+                } else {
+                    $role = bbp_get_spectator_role();
+                }
+
+                break;
+
+            // WordPress core context
+            case ''          :
+            case 'blog'      :
+            case 'wordpress' :
+            default          :
+                $role = array_intersect( array_values( $user->roles ), array_keys( get_editable_roles() ) );
+
+                // If there's a role in the array, use the first one
+                if ( !empty( $role ) ) {
+                    $role = array_shift( array_values( $role ) );
+                }
+
+                break;
+        }
+    }
+
+    return apply_filters( 'bbp_get_user_role', $role, $context, $user_id, $user );
 }