[not verified] Use ( new Host() )->is_wpcom_simple()

Automattic · Jun 1, 2023 · b04ac83 · b04ac83
1 parent 0aa43f4
commit b04ac83
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/..._inc/lib/core-api/wpcom-endpoints/class-wpcom-rest-api-v2-endpoint-send-email-preview.php b/..._inc/lib/core-api/wpcom-endpoints/class-wpcom-rest-api-v2-endpoint-send-email-preview.php
@@ -6,6 +6,7 @@
  */
 
 use Automattic\Jetpack\Connection\Manager;
+use Automattic\Jetpack\Status\Host;
 
 require_once __DIR__ . '/trait-wpcom-rest-api-proxy-request-trait.php';
 
@@ -39,11 +40,12 @@ public function __construct() {
 	 */
 	public function register_routes() {
 		$options = array(
-			'show_in_index' => true,
-			'methods'       => 'POST',
+			'show_in_index'       => true,
+			'methods'             => 'POST',
 			// if this is not a wpcom site, we need to proxy the request to wpcom
-			'callback'      => ( defined( 'IS_WPCOM' ) && IS_WPCOM ) ? array( $this, 'send_email_preview' ) : array( $this, 'proxy_request_to_wpcom_as_user' ),
-			'args'          => array(
+			'callback'            => ( ( new Host() )->is_wpcom_simple() ) ? array( $this, 'send_email_preview' ) : array( $this, 'proxy_request_to_wpcom_as_user' ),
+			'permission_callback' => array( $this, 'permissions_check' ),
+			'args'                => array(
 				'id' => array(
 					'description' => __( 'Unique identifier for the post.', 'jetpack' ),
 					'type'        => 'integer',
@@ -66,12 +68,14 @@ public function register_routes() {
 	 * @return true|WP_Error True if the request has edit access, WP_Error object otherwise.
 	 */
 	public function permissions_check( $request ) {
-		if ( ! ( new Manager() )->is_user_connected() ) {
-			return new WP_Error(
-				'rest_cannot_send_email_preview',
-				__( 'Please connect your user account to WordPress.com', 'jetpack' ),
-				array( 'status' => rest_authorization_required_code() )
-			);
+		if ( ! ( new Host() )->is_wpcom_simple() ) {
+			if ( ! ( new Manager() )->is_user_connected() ) {
+				return new WP_Error(
+					'rest_cannot_send_email_preview',
+					__( 'Please connect your user account to WordPress.com', 'jetpack' ),
+					array( 'status' => rest_authorization_required_code() )
+				);
+			}
 		}
 
 		$post = get_post( $request->get_param( 'id' ) );