The conditionals. */ public static function get_conditionals() { return [ AI_Conditional::class, New_Premium_Or_Free_AI_Conditional::class ]; } /** * Class constructor. * * @param AI_Request_Sender_Factory $ai_request_sender_factory The auth strategy factory. * @param Consent_Handler $consent_handler The consent handler instance. * @param WPSEO_Addon_Manager $addon_manager The add-on manager instance. */ public function __construct( AI_Request_Sender_Factory $ai_request_sender_factory, Consent_Handler $consent_handler, WPSEO_Addon_Manager $addon_manager ) { $this->addon_manager = $addon_manager; $this->ai_request_sender_factory = $ai_request_sender_factory; $this->consent_handler = $consent_handler; } /** * Registers routes with WordPress. * * @return void */ public function register_routes() { \register_rest_route( self::ROUTE_NAMESPACE, self::ROUTE_PREFIX, [ 'methods' => 'POST', 'args' => [ 'is_woo_product_entity' => [ 'type' => 'boolean', 'default' => false, ], ], 'callback' => [ $this, 'get_usage' ], 'permission_callback' => [ $this, 'check_permissions' ], ], ); } /** * Runs the callback that gets the monthly usage of the user. * * @param WP_REST_Request $request The request object. * * @return WP_REST_Response The response of the callback action. */ public function get_usage( $request ): WP_REST_Response { $is_woo_product_entity = $request->get_param( 'is_woo_product_entity' ); $user = \wp_get_current_user(); try { $parameters = new Usage_Parameters( $user, $this->is_free( $is_woo_product_entity ), \gmdate( 'Y-m' ) ); $sender = $this->ai_request_sender_factory->create( $user ); $response = $sender->get_usage( $parameters ); $data = \json_decode( $response->get_body() ); } catch ( Remote_Request_Exception | WP_Request_Exception $e ) { if ( $e instanceof Forbidden_Exception && ! $e instanceof Insufficient_Scope_Exception ) { $this->consent_handler->revoke_consent( $user->ID ); } $message = [ 'errorMessage' => $e->getMessage(), 'errorIdentifier' => $e->get_error_identifier(), 'errorCode' => $e->getCode(), ]; if ( $e instanceof Too_Many_Requests_Exception ) { $message['missingLicenses'] = $e->get_missing_licenses(); } return new WP_REST_Response( $message, $e->getCode(), ); } return new WP_REST_Response( $data ); } /** * Whether the current request should read the time-unbound free-usage bucket. * * A user without the relevant paid subscription is on the free tier, whose usage is tracked in * the free-usage bucket rather than per period. * * @param bool $is_woo_product_entity Whether the request is for a WooCommerce product entity. * * @return bool True when the user is on the free tier for this request type. */ public function is_free( $is_woo_product_entity = false ): bool { if ( $is_woo_product_entity ) { return ! $this->addon_manager->has_valid_subscription( WPSEO_Addon_Manager::WOOCOMMERCE_SLUG ); } return ! $this->addon_manager->has_valid_subscription( WPSEO_Addon_Manager::PREMIUM_SLUG ); } }