Set logged in cookie on store api requests

Author: mikejolley

src/RestApi.php

@@ -36,6 +36,7 @@ public function __construct( ExtendRestApi $extend ) {
 	protected function init() {
 		add_action( 'rest_api_init', array( $this, 'register_rest_routes' ), 10 );
 		add_filter( 'rest_authentication_errors', array( $this, 'store_api_authentication' ) );
+		add_action( 'set_logged_in_cookie', array( $this, 'store_api_logged_in_cookie' ) );
 	}
 
 	/**
@@ -81,6 +82,19 @@ public function store_api_authentication( $result ) {
 		return true;
 	}
 
+	/**
+	 * When the login cookies are set, they are not available until the next page reload. For the Store API, specifically
+	 * for returning updated nonces, we need this to be available immediately.
+	 *
+	 * @param string $logged_in_cookie The value for the logged in cookie.
+	 */
+	public function store_api_logged_in_cookie( $logged_in_cookie ) {
+		if ( ! defined( 'LOGGED_IN_COOKIE' ) || ! self::is_request_to_store_api() ) {
+			return;
+		}
+		$_COOKIE[ LOGGED_IN_COOKIE ] = $logged_in_cookie;
+	}
+
 	/**
 	 * Check if is request to the Store API.
 	 *