Code Coverage
 
Lines
Functions and Methods
Classes and Traits
Total
0.00%
0 / 43
0.00%
0 / 5
 CRAP
0.00%
0 / 1
REST_Controller
0.00%
0 / 43
0.00%
0 / 5
 42
0.00%
0 / 1
 init
0.00%
0 / 2
0.00%
0 / 1
 2
 register_rest_endpoints
0.00%
0 / 33
0.00%
0 / 1
 2
 api_check_plan
0.00%
0 / 2
0.00%
0 / 1
 2
 api_get_status
0.00%
0 / 2
0.00%
0 / 1
 2
 api_clear_scan_cache
0.00%
0 / 4
0.00%
0 / 1
 6
1<?php
2/**
3 * Class file for managing REST API endpoints for package protect-status.
4 *
5 * @package automattic/jetpack-protect-status
6 */
7
8namespace Automattic\Jetpack\Protect_Status;
9
10use Automattic\Jetpack\Connection\Rest_Authentication;
11use WP_REST_Request;
12use WP_REST_Response;
13
14/**
15 * Class REST_Controller
16 */
17class REST_Controller {
18
19    /**
20     * Initialize the plugin's REST API.
21     *
22     * @return void
23     */
24    public static function init() {
25        // Set up the REST authentication hooks.
26        Rest_Authentication::init();
27
28        // Add custom WP REST API endoints.
29        add_action( 'rest_api_init', array( __CLASS__, 'register_rest_endpoints' ) );
30    }
31
32    /**
33     * Register the REST API routes.
34     *
35     * @return void
36     */
37    public static function register_rest_endpoints() {
38        register_rest_route(
39            'jetpack-protect/v1',
40            'check-plan',
41            array(
42                'methods'             => \WP_REST_Server::READABLE,
43                'callback'            => __CLASS__ . '::api_check_plan',
44                'permission_callback' => function () {
45                    return current_user_can( 'manage_options' );
46                },
47            )
48        );
49
50        register_rest_route(
51            'jetpack-protect/v1',
52            'status',
53            array(
54                'methods'             => \WP_REST_Server::READABLE,
55                'callback'            => __CLASS__ . '::api_get_status',
56                'permission_callback' => function () {
57                    return current_user_can( 'manage_options' );
58                },
59            )
60        );
61
62        register_rest_route(
63            'jetpack-protect/v1',
64            'clear-scan-cache',
65            array(
66                'methods'             => \WP_REST_Server::EDITABLE,
67                'callback'            => __CLASS__ . '::api_clear_scan_cache',
68                'permission_callback' => function () {
69                    return current_user_can( 'manage_options' );
70                },
71            )
72        );
73    }
74
75    /**
76     * Return site plan data for the API endpoint
77     *
78     * @return WP_REST_Response
79     */
80    public static function api_check_plan() {
81        $has_required_plan = Plan::has_required_plan();
82
83        return rest_ensure_response( $has_required_plan );
84    }
85
86    /**
87     * Return Protect Status for the API endpoint
88     *
89     * @param WP_REST_Request $request The request object.
90     *
91     * @return WP_REST_Response
92     */
93    public static function api_get_status( $request ) {
94        $status = Status::get_status( $request['hard_refresh'] );
95        return rest_ensure_response( $status );
96    }
97
98    /**
99     * Clear the Scan_Status cache for the API endpoint
100     *
101     * @return WP_REST_Response
102     */
103    public static function api_clear_scan_cache() {
104        $cache_cleared = Scan_Status::delete_option();
105
106        if ( ! $cache_cleared ) {
107            return new WP_REST_Response( 'An error occurred while attempting to clear the Jetpack Scan cache.', 500 );
108        }
109
110        return new WP_REST_Response( 'Jetpack Scan cache cleared.' );
111    }
112}