Changeset 4080 for sites/trunk/wordpress.org/public_html/wp-content/plugins/rosetta/inc/site/class-locale-main.php

Timestamp:

09/16/2016 03:50:43 PM (8 years ago)

Author:

ocean90

Message:

Rosetta: Add role manager for custom roles which are used on the main site.

See #2003, #1398.

File:

• sites/trunk/wordpress.org/public_html/wp-content/plugins/rosetta/inc/site/class-locale-main.php (modified) (4 diffs)

sites/trunk/wordpress.org/public_html/wp-content/plugins/rosetta/inc/site/class-locale-main.php

@@ -5 +5 @@
 use WordPressdotorg\Rosetta\Jetpack;
 use WordPressdotorg\Rosetta\User;
+use WordPressdotorg\Rosetta\User\Role;
 use WP_Site;
+use WP_User;
 
 class Locale_Main implements Site {
@@ -49 +51 @@
         }
 
+        $this->initialize_jetpack_customizations();
+        $this->initialize_user_role_customizations();
+    }
+
+    /**
+     * Initializes customizations for Jetpack.
+     */
+    private function initialize_jetpack_customizations() {
         $jetpack_module_manager = new Jetpack\Module_Manager( [
             'stats',
@@ -58 +68 @@
         $jetpack_module_manager->setup();
 
+        // Options for Jetpack's sharing module.
         $options = new Filter\Options();
-        // Options for Jetpack's sharing module.
         $options->add_option(
             ( new Filter\Option() )
@@ -87 +97 @@
         $options->setup();
     }
+
+    /**
+     * Initializes user role customizations.
+     */
+    private function initialize_user_role_customizations() {
+        $role_manager = new User\Role_Manager();
+        $role_manager->add_role( new Role\Locale_Manager() );
+        $role_manager->add_role( new Role\General_Translation_Editor() );
+        $role_manager->add_role( new Role\Translation_Editor() );
+        $role_manager->setup();
+
+        add_action( 'set_user_role', [ $this, 'restore_translation_editor_role' ], 10, 3 );
+        add_filter( 'editable_roles', [ $this, 'remove_administrator_from_editable_roles' ] );
+    }
+
+    /**
+     * Restores the "(General) Translation Editor" role if an user is promoted.
+     *
+     * @param int    $user_id   The user ID.
+     * @param string $role      The new role.
+     * @param array  $old_roles An array of the user's previous roles.
+     */
+    public function restore_translation_editor_role( $user_id, $role, $old_roles ) {
+        if (
+            Role\General_Translation_Editor::get_name() !== $role &&
+            in_array( Role\Translation_Editor::get_name(), $old_roles, true )
+        ) {
+            $user = new WP_User( $user_id );
+            $user->add_role( Role\Translation_Editor::get_name() );
+        }
+
+        if (
+            Role\Translation_Editor::get_name() !== $role &&
+            in_array( Role\General_Translation_Editor::get_name(), $old_roles, true )
+        ) {
+            $user = new WP_User( $user_id );
+            $user->add_role( Role\General_Translation_Editor::get_name() );
+        }
+    }
+
+    /**
+     * Removes "Administrator" role from the list of editable roles.
+     *
+     * @param array $roles List of roles.
+     * @return array Filtered list of editable roles.
+     */
+    public function remove_administrator_from_editable_roles( $roles ) {
+        if ( ! is_super_admin() && ! is_main_site() ) {
+            unset( $roles['administrator'] );
+        }
+
+        return $roles;
+    }
 }