Ticket #31532: 31532.2.diff

src/wp-admin/admin-ajax.php

@@ -61 +61 @@
         'query-attachments', 'save-attachment', 'save-attachment-compat', 'send-link-to-editor',
         'send-attachment-to-editor', 'save-attachment-order', 'heartbeat', 'get-revision-diffs',
         'save-user-color-scheme', 'update-widget', 'query-themes', 'parse-embed', 'set-attachment-thumbnail',
-        'parse-media-shortcode', 'destroy-sessions', 'install-plugin', 'update-plugin', 'press-this-save-post',
-        'press-this-add-category',
+        'parse-media-shortcode', 'destroy-sessions', 'install-plugin', 'activate-plugin', 'update-plugin',
+        'press-this-save-post', 'press-this-add-category',
 );
 
 // Register core Ajax calls.

src/wp-admin/includes/ajax-actions.php

@@ -2919 +2919 @@
                 wp_send_json_error( $status );
         }
 
+        if ( is_multisite() ) {
+                wp_send_json_success( $status );
+        }
+
         $plugin_status = install_plugin_install_status( $api );
+        $plugin = $plugin_status['file'];
 
-        if ( ! is_multisite() ) {
-                activate_plugin( $plugin_status['file'] );
+        $url  = admin_url( 'admin-ajax.php' );
+        $args = array(
+                /** This filter is documented in wp-includes/class-http.php */
+                'sslverify' => apply_filters( 'https_local_ssl_verify', false ),
+                'body'      => array(
+                        'action'  => 'activate-plugin',
+                        'nonce'   => wp_create_nonce( 'ajax-activate-plugin_' . $plugin ),
+                        'plugin'  => $plugin,
+                ),
+                'cookies'   => array(
+                        AUTH_COOKIE      => $_COOKIE[ AUTH_COOKIE ],
+                        LOGGED_IN_COOKIE => $_COOKIE[ LOGGED_IN_COOKIE ],
+                ),
+        );
+
+        if ( isset( $_COOKIE[ SECURE_AUTH_COOKIE ] ) ) {
+                $args['cookies'][ SECURE_AUTH_COOKIE ] = $_COOKIE[ SECURE_AUTH_COOKIE ];
         }
 
+        $response = wp_remote_post( $url, $args );
+
+        if ( is_wp_error( $response ) || 200 != wp_remote_retrieve_response_code( $response ) ) {
+                $status['error'] = __( 'Could not activate plugin.' );
+                wp_send_json_error( $status );
+        }
+
+        $body = wp_remote_retrieve_body( $response );
+        $activate_result = json_decode( $body );
+
+        if ( null === $activate_result ) {
+                $status['error'] = __( 'Could not activate plugin.' );
+                wp_send_json_error( $status );
+        }
+
+        if ( false === $activate_result->success ) {
+                $status['error'] = $activate_result->data;
+                wp_send_json_error( $status );
+        }
+
+        $url  = admin_url( 'plugin-install.php' );
+        $args = array(
+                /** This filter is documented in wp-includes/class-http.php */
+                'sslverify' => apply_filters( 'https_local_ssl_verify', false ),
+                'cookies'   => array(
+                        AUTH_COOKIE      => $_COOKIE[ AUTH_COOKIE ],
+                        LOGGED_IN_COOKIE => $_COOKIE[ LOGGED_IN_COOKIE ],
+                ),
+        );
+
+        if ( isset( $_COOKIE[ SECURE_AUTH_COOKIE ] ) ) {
+                $args['cookies'][ SECURE_AUTH_COOKIE ] = $_COOKIE[ SECURE_AUTH_COOKIE ];
+        }
+
+        $response = wp_remote_get( $url, $args );
+
+        if ( ! is_wp_error( $response ) && 200 == wp_remote_retrieve_response_code( $response ) ) {
+                $body = wp_remote_retrieve_body( $response );
+                list( $ignore, $menu ) = explode( '<div id="adminmenuwrap">', $body );
+                list( $menu, $ignore ) = explode( '<div id="wpcontent">', $menu );
+                $menu = preg_replace( '|</div>\s+$|', '', $menu );
+
+                $status['menu'] = $menu;
+        }
+
         wp_send_json_success( $status );
 }
 
 /**
+ * AJAX handler for activating a plugin.
+ *
+ * @since 4.2.0
+ */
+function wp_ajax_activate_plugin() {
+        $plugin = $_POST['plugin'];
+
+        $verify = wp_verify_nonce( $_POST['nonce'], 'ajax-activate-plugin_' . $plugin );
+
+        if ( ! $verify ) {
+                wp_send_json_error( __( 'Could not activate plugin.' ) );
+        }
+
+        include_once( ABSPATH . 'wp-admin/includes/plugin-install.php' );
+
+        $result = activate_plugin( $plugin );
+
+        if ( is_wp_error( $result ) ) {
+                wp_send_json_error( $result->get_error_message() );
+        }
+
+        wp_send_json_success( true );
+}
+
+/**
  * AJAX handler for updating a plugin.
  *
  * @since 4.2.0

src/wp-admin/js/updates.js

@@ -311 +311 @@
                 wp.a11y.speak( wp.updates.l10n.installedMsg );
                 wp.updates.updateDoneSuccessfully = true;
 
+                if ( response.menu ) {
+                        $( '#adminmenuwrap' ).html( response.menu );
+                }
+
                 /*
                  * The lock can be released since the update was successful,
                  * and any other updates can commence.