class-wpcom-rest-api-v2-endpoint-following.php

<?php
/**
 * REST API endpoint for the Jetpack Blogroll block.
 *
 * @package automattic/jetpack
 * @since 12.2
 */

use Automattic\Jetpack\Connection\Client;
use Automattic\Jetpack\Status\Visitor;

if ( ! defined( 'ABSPATH' ) ) {
	exit( 0 );
}

/**
 * Class WPCOM_REST_API_V2_Endpoint_Following
 */
class WPCOM_REST_API_V2_Endpoint_Following extends WP_REST_Controller {
	/**
	 * Namespace prefix.
	 *
	 * @var string
	 */
	public $namespace = 'wpcom/v2';

	/**
	 * Endpoint base route.
	 *
	 * @var string
	 */
	public $rest_base = 'following';

	/**
	 * Constructor.
	 */
	public function __construct() {
		$this->wpcom_is_wpcom_only_endpoint    = true;
		$this->wpcom_is_site_specific_endpoint = false;

		add_action( 'rest_api_init', array( $this, 'register_routes' ) );
	}

	/**
	 * Register routes.
	 */
	public function register_routes() {
		register_rest_route(
			$this->namespace,
			$this->rest_base . '/mine',
			array(
				array(
					'methods'             => WP_REST_Server::READABLE,
					'callback'            => array( $this, 'get_following' ),
					'permission_callback' => 'is_user_logged_in',
					'args'                => array(
						'ignore_user_blogs' => array(
							'type' => 'boolean',
						),
					),
				),
			)
		);

		register_rest_route(
			$this->namespace,
			$this->rest_base . '/recommendations',
			array(
				array(
					'methods'             => WP_REST_Server::READABLE,
					'callback'            => array( $this, 'get_recommendations' ),
					'permission_callback' => 'is_user_logged_in',
					'args'                => array(
						'number' => array(
							'type'              => 'number',
							'default'           => 5,
							'validate_callback' => function ( $param ) {
								return is_numeric( $param ) && $param <= 20;
							},
						),
					),
				),
			)
		);
	}

	/**
	 * Gets the sites the user is following
	 *
	 * @param WP_REST_Request $request Full details about the request.
	 * @return array|WP_Error list of followed sites, WP_Error otherwise
	 */
	public function get_following( $request ) {
		$ignore_user_blogs = $request->get_param( 'ignore_user_blogs' );

		if ( defined( 'IS_WPCOM' ) && IS_WPCOM ) {
			require_lib( 'wpcom-get-user-followed-blogs' );
			return get_user_followed_blogs( get_current_user_id(), $ignore_user_blogs );
		}

		$body = Client::wpcom_json_api_request_as_user(
			sprintf( '/me/following%s', $ignore_user_blogs ? '?ignore_user_blogs=true' : '' ),
			'2',
			array(
				'method'  => 'GET',
				'headers' => array(
					'Content-Type'    => 'application/json',
					'X-Forwarded-For' => ( new Visitor() )->get_ip( true ),
				),
			)
		);

		if ( is_wp_error( $body ) ) {
			return $body;
		}

		return json_decode( wp_remote_retrieve_body( $body ) );
	}

	/**
	 * Gets recommended sites for user
	 *
	 * @param WP_REST_Request $request Full details about the request.
	 * @return array|WP_Error list of following recommendations, WP_Error otherwise
	 */
	public function get_recommendations( $request ) {
		$number_of_recommendations = $request->get_param( 'number' );

		if ( defined( 'IS_WPCOM' ) && IS_WPCOM ) {
			require_lib( 'wpcom-get-user-followed-blogs' );
			return get_user_following_recommendations( get_current_user_id(), $number_of_recommendations );
		}

		$body = Client::wpcom_json_api_request_as_user(
			sprintf( '/me/following/recommendations?number=%d', $number_of_recommendations ),
			'2',
			array(
				'method'  => 'GET',
				'headers' => array(
					'Content-Type'    => 'application/json',
					'X-Forwarded-For' => ( new Visitor() )->get_ip( true ),
				),
			)
		);

		if ( is_wp_error( $body ) ) {
			return $body;
		}

		return json_decode( wp_remote_retrieve_body( $body ) );
	}
}

wpcom_rest_api_v2_load_plugin( 'WPCOM_REST_API_V2_Endpoint_Following' );