Membership Block

_inc/lib/core-api/wpcom-endpoints/memberships.php

@@ -0,0 +1,180 @@
+<?php // phpcs:disable WordPress.Files.FileName.InvalidClassFileName
+/**
+ * Memberships: API to communicate with "product" database.
+ *
+ * @package    Jetpack
+ * @since      7.3.0
+ */
+
+/**
+ * Class WPCOM_REST_API_V2_Endpoint_Memberships
+ * This introduces V2 endpoints.
+ */
+class WPCOM_REST_API_V2_Endpoint_Memberships extends WP_REST_Controller {
+
+	/**
+	 * WPCOM_REST_API_V2_Endpoint_Memberships constructor.
+	 */
+	public function __construct() {
+		$this->namespace                       = 'wpcom/v2';
+		$this->rest_base                       = 'memberships';
+		$this->wpcom_is_wpcom_only_endpoint    = true;
+		$this->wpcom_is_site_specific_endpoint = true;
+		add_action( 'rest_api_init', array( $this, 'register_routes' ) );
+	}
+
+	/**
+	 * Called automatically on `rest_api_init()`.
+	 */
+	public function register_routes() {
+		register_rest_route(
+			$this->namespace,
+			$this->rest_base . '/status',
+			array(
+				array(
+					'methods'             => WP_REST_Server::READABLE,
+					'callback'            => array( $this, 'get_status' ),
+					'permission_callback' => array( $this, 'get_status_permission_check' ),
+				),
+			)
+		);
+		register_rest_route(
+			$this->namespace,
+			$this->rest_base . '/product',
+			array(
+				array(
+					'methods'             => WP_REST_Server::CREATABLE,
+					'callback'            => array( $this, 'create_product' ),
+					'permission_callback' => array( $this, 'get_status_permission_check' ),
+					'args'                => array(
+						'title'    => array(
+							'type'     => 'string',
+							'required' => true,
+						),
+						'price'    => array(
+							'type'     => 'float',
+							'required' => true,
+						),
+						'currency' => array(
+							'type'     => 'string',
+							'required' => true,
+						),
+						'interval' => array(
+							'type'     => 'string',
+							'required' => true,
+						),
+					),
+				),
+			)
+		);
+	}
+
+	/**
+	 * Ensure the user has proper permissions
+	 *
+	 * @return boolean
+	 */
+	public function get_status_permission_check() {
+		return current_user_can( 'edit_posts' );
+	}
+
+	/**
+	 * Do create a product based on data, or pass request to wpcom.
+	 *
+	 * @param object $request - request passed from WP.
+	 *
+	 * @return array|WP_Error
+	 */
+	public function create_product( $request ) {
+		if ( ( defined( 'IS_WPCOM' ) && IS_WPCOM ) ) {
+			require_lib( 'memberships' );
+			$connected_destination_account_id = Jetpack_Memberships::get_connected_account_id();
+			if ( ! $connected_destination_account_id ) {
+				return new WP_Error( 'no-destination-account', __( 'Please set up a Stripe account for this site first', 'jetpack' ) );
+			}
+			$product = Memberships_Product::create(
+				get_current_blog_id(),
+				array(
+					'title'                            => $request['title'],
+					'price'                            => $request['price'],
+					'currency'                         => $request['currency'],
+					'interval'                         => $request['interval'],
+					'connected_destination_account_id' => $connected_destination_account_id,
+				)
+			);
+			return $product->to_array();
+		} else {
+			$blog_id  = Jetpack_Options::get_option( 'id' );
+			$response = Jetpack_Client::wpcom_json_api_request_as_user(
+				"/sites/$blog_id/{$this->rest_base}/product",
+				'v2',
+				array(
+					'method' => 'POST',
+				),
+				array(
+					'title'    => $request['title'],
+					'price'    => $request['price'],
+					'currency' => $request['currency'],
+					'interval' => $request['interval'],
+				)
+			);
+			if ( is_wp_error( $response ) ) {
+				if ( $response->get_error_code() === 'missing_token' ) {
+					return new WP_Error( 'missing_token', __( 'Please connect your user account to WordPress.com', 'jetpack' ), 404 );
+				}
+				return new WP_Error( 'wpcom_connection_error', __( 'Could not connect to WordPress.com', 'jetpack' ), 404 );
+			}
+			$data = isset( $response['body'] ) ? json_decode( $response['body'], true ) : null;
+			return $data;
+		}
+
+		return $request;
+	}
+
+	/**
+	 * Get a status of connection for the site. If this is Jetpack, pass the request to wpcom.
+	 *
+	 * @return array|WP_Error
+	 */
+	public function get_status() {
+		$connected_account_id = Jetpack_Memberships::get_connected_account_id();
+		$connect_url          = '';
+		if ( ( defined( 'IS_WPCOM' ) && IS_WPCOM ) ) {
+			require_lib( 'memberships' );
+			$blog_id = get_current_blog_id();
+			if ( ! $connected_account_id ) {
+				$connect_url = get_memberships_connected_account_redirect( get_current_user_id(), $blog_id );
+			}
+			$products = get_memberships_plans( $blog_id );
+		} else {
+			$blog_id  = Jetpack_Options::get_option( 'id' );
+			$response = Jetpack_Client::wpcom_json_api_request_as_user(
+				"/sites/$blog_id/{$this->rest_base}/status",
+				'v2',
+				array(),
+				null
+			);
+			if ( is_wp_error( $response ) ) {
+				if ( $response->get_error_code() === 'missing_token' ) {
+					return new WP_Error( 'missing_token', __( 'Please connect your user account to WordPress.com', 'jetpack' ), 404 );
+				}
+				return new WP_Error( 'wpcom_connection_error', __( 'Could not connect to WordPress.com', 'jetpack' ), 404 );
+			}
+			$data = isset( $response['body'] ) ? json_decode( $response['body'], true ) : null;
+			if ( ! $connected_account_id ) {
+				$connect_url = empty( $data['connect_url'] ) ? '' : $data['connect_url'];
+			}
+			$products = empty( $data['products'] ) ? array() : $data['products'];
+		}
+		return array(
+			'connected_account_id' => $connected_account_id,
+			'connect_url'          => $connect_url,
+			'products'             => $products,
+		);
+	}
+}
+
+if ( ( defined( 'IS_WPCOM' ) && IS_WPCOM ) || Jetpack::is_active() ) {
+	wpcom_rest_api_v2_load_plugin( 'WPCOM_REST_API_V2_Endpoint_Memberships' );
+}
+

bin/phpcs-whitelist.js

@@ -8,8 +8,10 @@ module.exports = [
 	'functions.global.php',
 	'functions.opengraph.php',
 	'_inc/lib/debugger/',
+	'_inc/lib/core-api/wpcom-endpoints/memberships.php',
 	'_inc/lib/class.jetpack-password-checker.php',
 	'modules/masterbar/',
+	'modules/memberships/',
 	'modules/module-extras.php',
 	'modules/module-info.php',
 	'modules/theme-tools/social-menu/',