Changeset 4927 for sites/trunk/wordpress.org/public_html/wp-content/plugins/wporg-gp-customizations/inc/class-plugin.php

Timestamp:

02/18/2017 06:12:44 PM (8 years ago)

Author:

ocean90

Message:

Translate: Add redirect parameter for current page to the login URL.

File:

• sites/trunk/wordpress.org/public_html/wp-content/plugins/wporg-gp-customizations/inc/class-plugin.php (modified) (3 diffs)

sites/trunk/wordpress.org/public_html/wp-content/plugins/wporg-gp-customizations/inc/class-plugin.php

@@ -47 +47 @@
         // Toolbar.
         add_action( 'admin_bar_menu', array( $this, 'add_profile_settings_to_admin_bar' ) );
+        add_action( 'admin_bar_menu', array( $this, 'replace_login_url_in_admin_bar' ), 20 );
         add_action( 'admin_bar_init', array( $this, 'show_admin_bar' ) );
         add_action( 'add_admin_bar_menus', array( $this, 'remove_admin_bar_menus' ) );
@@ -89 +90 @@
      * Adds the link to profile settings to the user actions toolbar menu.
      *
-     * @param \WP_Admin_Bar $wp_admin_bar
+     * @param \WP_Admin_Bar $wp_admin_bar WP_Admin_Bar instance.
      */
     public function add_profile_settings_to_admin_bar( $wp_admin_bar ) {
@@ -105 +106 @@
             $wp_admin_bar->add_node( $logout_node ); // Ensures that logout is the last action.
         }
+    }
+
+    /**
+     * Changes the login URL to include a redirect parameter for the current page.
+     *
+     * @param \WP_Admin_Bar $wp_admin_bar WP_Admin_Bar instance.
+     */
+    public function replace_login_url_in_admin_bar( $wp_admin_bar ) {
+        if ( ! $wp_admin_bar->get_node( 'log-in' ) ) {
+            return;
+        }
+
+        $menu = [
+            'id' => 'log-in',
+            'href' => wp_login_url( gp_url_current() ),
+        ];
+        $wp_admin_bar->add_menu( $menu );
     }