class-plan.php
8 months ago
class-protect-status.php
2 months ago
class-rest-controller.php
8 months ago
class-scan-status.php
2 months ago
class-status.php
3 days ago
class-rest-controller.php
113 lines
| 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 |