From d12459ec8d0962874f23300b6cafd91079bd7ca8 Mon Sep 17 00:00:00 2001
From: Christian Gastrell <cgastrell@gmail.com>
Date: Thu, 23 Feb 2023 14:01:01 -0300
Subject: [PATCH] use Jetpack\Connection\Manager to get site ID, remove
 unneeded code and is_wpcom var

---
 ...class-wpcom-rest-api-v2-endpoint-forms.php | 51 ++++++++-----------
 1 file changed, 20 insertions(+), 31 deletions(-)

diff --git a/projects/packages/forms/src/class-wpcom-rest-api-v2-endpoint-forms.php b/projects/packages/forms/src/class-wpcom-rest-api-v2-endpoint-forms.php
index 11d13023d7f91..4de0f763b54c0 100644
--- a/projects/packages/forms/src/class-wpcom-rest-api-v2-endpoint-forms.php
+++ b/projects/packages/forms/src/class-wpcom-rest-api-v2-endpoint-forms.php
@@ -8,7 +8,7 @@
 
 namespace Automattic\Jetpack\Forms;
 
-use Jetpack_Options;
+use Automattic\Jetpack\Connection\Manager;
 use WP_Error;
 use WP_REST_Controller;
 use WP_REST_Server;
@@ -17,19 +17,10 @@
  * Handles the REST routes for Form Responses, aka Feedback.
  */
 class WPCOM_REST_API_V2_Endpoint_Forms extends WP_REST_Controller {
-	/**
-	 * Is WPCOM or not
-	 *
-	 * @var bool
-	 */
-	protected $is_wpcom;
-
 	/**
 	 * Constructor.
 	 */
 	public function __construct() {
-		$this->is_wpcom = defined( 'IS_WPCOM' ) && IS_WPCOM;
-
 		$this->namespace = 'wpcom/v2';
 		$this->rest_base = 'forms';
 
@@ -53,25 +44,25 @@ public function register_rest_routes() {
 				'permissions_check' => array( $this, 'get_responses_permission_check' ),
 				'args'              => array(
 					'limit'   => array(
-						'default'           => 20,
-						'type'              => 'integer',
-						'required'          => false,
-						'minimum'           => 1,
+						'default'  => 20,
+						'type'     => 'integer',
+						'required' => false,
+						'minimum'  => 1,
 					),
 					'offset'  => array(
-						'default'           => 0,
-						'type'              => 'integer',
-						'required'          => false,
-						'minimum'           => 0,
+						'default'  => 0,
+						'type'     => 'integer',
+						'required' => false,
+						'minimum'  => 0,
 					),
 					'form_id' => array(
-						'type'              => 'integer',
-						'required'          => false,
-						'minimum'           => 1,
+						'type'     => 'integer',
+						'required' => false,
+						'minimum'  => 1,
 					),
 					'search'  => array(
-						'type'              => 'string',
-						'required'          => false,
+						'type'     => 'string',
+						'required' => false,
 					),
 				),
 			)
@@ -154,7 +145,12 @@ function ( $response ) {
 	 * @return true|WP_Error Returns true if the user has the required capability, else a WP_Error object.
 	 */
 	public function get_responses_permission_check() {
-		if ( ! is_user_member_of_blog( get_current_user_id(), $this->get_blog_id() ) ) {
+		$site_id = Manager::get_site_id();
+		if ( is_wp_error( $site_id ) ) {
+			return $site_id;
+		}
+
+		if ( ! is_user_member_of_blog( get_current_user_id(), $site_id ) ) {
 			return new WP_Error(
 				'invalid_user_permission_jetpack_form_responses',
 				'unauthorized',
@@ -164,13 +160,6 @@ public function get_responses_permission_check() {
 
 		return true;
 	}
-
-	/**
-	 * Get blog ID selectively depending on IS_WPCOM or not
-	 */
-	protected function get_blog_id() {
-		return $this->is_wpcom ? get_current_blog_id() : Jetpack_Options::get_option( 'id' );
-	}
 }
 
 if ( defined( 'IS_WPCOM' ) && IS_WPCOM ) {