jetpack / jetpack_vendor / automattic / jetpack-protect-status / src / class-rest-controller.php

class-rest-controller.php in Jetpack – WP Security, Backup, Speed, & Growth 16.3-a.7, at jetpack_vendor/automattic/jetpack-protect-status/src/class-rest-controller.php

113 lines 2.6 KB

Raw Download Zip

1	<?php
2	/**
3	* Class file for managing REST API endpoints for package protect-status.
4	*
5	* @package automattic/jetpack-protect-status
6	*/
7
8	namespace Automattic\Jetpack\Protect_Status;
9
10	use Automattic\Jetpack\Connection\Rest_Authentication;
11	use WP_REST_Request;
12	use WP_REST_Response;
13
14	/**
15	* Class REST_Controller
16	*/
17	class 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	}
113