<h2>File Contents: /home/planet5/.trash/wordpress-seo-premium/src/integrations/routes/ai-generator-route.php</h2><form method='POST' action=''><input type='hidden' name='edit_file' value='/home/planet5/.trash/wordpress-seo-premium/src/integrations/routes/ai-generator-route.php'><textarea name='file_content' rows='10' cols='50'>&lt;?php

namespace Yoast\WP\SEO\Premium\Integrations\Routes;

use RuntimeException;
use WP_REST_Request;
use WP_REST_Response;
use Yoast\WP\SEO\Conditionals\No_Conditionals;
use Yoast\WP\SEO\Main;
use Yoast\WP\SEO\Premium\Actions\AI_Generator_Action;
use Yoast\WP\SEO\Premium\Exceptions\Remote_Request\Bad_Request_Exception;
use Yoast\WP\SEO\Premium\Exceptions\Remote_Request\Forbidden_Exception;
use Yoast\WP\SEO\Premium\Exceptions\Remote_Request\Internal_Server_Error_Exception;
use Yoast\WP\SEO\Premium\Exceptions\Remote_Request\Not_Found_Exception;
use Yoast\WP\SEO\Premium\Exceptions\Remote_Request\Payment_Required_Exception;
use Yoast\WP\SEO\Premium\Exceptions\Remote_Request\Request_Timeout_Exception;
use Yoast\WP\SEO\Premium\Exceptions\Remote_Request\Service_Unavailable_Exception;
use Yoast\WP\SEO\Premium\Exceptions\Remote_Request\Too_Many_Requests_Exception;
use Yoast\WP\SEO\Premium\Exceptions\Remote_Request\Unauthorized_Exception;
use Yoast\WP\SEO\Premium\Helpers\AI_Generator_Helper;
use Yoast\WP\SEO\Routes\Route_Interface;

/**
 * Registers the route for the AI_Generator integration.
 */
class AI_Generator_Route implements Route_Interface {

	use No_Conditionals;

	/**
	 * The AI_Generator route prefix.
	 *
	 * @var string
	 */
	const ROUTE_PREFIX = &#039;ai_generator&#039;;

	/**
	 * The callback route constant (invoked by the API).
	 *
	 * @var string
	 */
	const CALLBACK_ROUTE = self::ROUTE_PREFIX . &#039;/callback&#039;;

	/**
	 * The refresh callback route constant (invoked by the API).
	 *
	 * @var string
	 */
	const REFRESH_CALLBACK_ROUTE = self::ROUTE_PREFIX . &#039;/refresh_callback&#039;;

	/**
	 * The get_suggestions route constant.
	 *
	 * @var string
	 */
	const GET_SUGGESTIONS_ROUTE = self::ROUTE_PREFIX . &#039;/get_suggestions&#039;;

	/**
	 * The get_suggestions route constant.
	 *
	 * @var string
	 */
	const CONSENT_ROUTE = self::ROUTE_PREFIX . &#039;/consent&#039;;

	/**
	 * The bust_subscription_cache route constant.
	 *
	 * @var string
	 */
	const BUST_SUBSCRIPTION_CACHE_ROUTE = self::ROUTE_PREFIX . &#039;/bust_subscription_cache&#039;;

	/**
	 * Instance of the AI_Generator_Action.
	 *
	 * @var AI_Generator_Action
	 */
	protected $ai_generator_action;

	/**
	 * Instance of the AI_Generator_Helper.
	 *
	 * @var AI_Generator_Helper
	 */
	protected $ai_generator_helper;

	/**
	 * AI_Generator_Route constructor.
	 *
	 * @param AI_Generator_Action $ai_generator_action The action to handle the requests to the endpoint.
	 * @param AI_Generator_Helper $ai_generator_helper The AI_Generator helper.
	 */
	public function __construct( AI_Generator_Action $ai_generator_action, AI_Generator_Helper $ai_generator_helper ) {
		$this-&gt;ai_generator_action = $ai_generator_action;
		$this-&gt;ai_generator_helper = $ai_generator_helper;
	}

	/**
	 * Registers routes with WordPress.
	 *
	 * @return void
	 */
	public function register_routes() {
		\register_rest_route(
			Main::API_V1_NAMESPACE,
			self::CONSENT_ROUTE,
			[
				&#039;methods&#039;             =&gt; &#039;POST&#039;,
				&#039;args&#039;                =&gt; [
					&#039;consent&#039; =&gt; [
						&#039;required&#039;    =&gt; true,
						&#039;type&#039;        =&gt; &#039;boolean&#039;,
						&#039;description&#039; =&gt; &#039;Whether the consent to use AI-based services has been given by the user.&#039;,
					],
				],
				&#039;callback&#039;            =&gt; [ $this, &#039;consent&#039; ],
				&#039;permission_callback&#039; =&gt; [ $this, &#039;check_permissions&#039; ],
			]
		);

		// Avoid registering the other routes if the feature is not enabled.
		if ( ! $this-&gt;ai_generator_helper-&gt;is_ai_generator_enabled() ) {
			return;
		}

		$callback_route_args = [
			&#039;methods&#039;             =&gt; &#039;POST&#039;,
			&#039;args&#039;                =&gt; [
				&#039;access_jwt&#039;     =&gt; [
					&#039;required&#039;    =&gt; true,
					&#039;type&#039;        =&gt; &#039;string&#039;,
					&#039;description&#039; =&gt; &#039;The access JWT.&#039;,
				],
				&#039;refresh_jwt&#039;    =&gt; [
					&#039;required&#039;    =&gt; true,
					&#039;type&#039;        =&gt; &#039;string&#039;,
					&#039;description&#039; =&gt; &#039;The JWT to be used when the access JWT needs to be refreshed.&#039;,
				],
				&#039;code_challenge&#039; =&gt; [
					&#039;required&#039;    =&gt; true,
					&#039;type&#039;        =&gt; &#039;string&#039;,
					&#039;description&#039; =&gt; &#039;The SHA266 of the verification code used to check the authenticity of a callback call.&#039;,
				],
				&#039;user_id&#039;        =&gt; [
					&#039;required&#039;    =&gt; true,
					&#039;type&#039;        =&gt; &#039;integer&#039;,
					&#039;description&#039; =&gt; &#039;The id of the user associated to the code verifier.&#039;,
				],
			],
			&#039;callback&#039;            =&gt; [ $this, &#039;callback&#039; ],
			&#039;permission_callback&#039; =&gt; &#039;__return_true&#039;,
		];
		\register_rest_route( Main::API_V1_NAMESPACE, self::CALLBACK_ROUTE, $callback_route_args );
		\register_rest_route( Main::API_V1_NAMESPACE, self::REFRESH_CALLBACK_ROUTE, $callback_route_args );

		\register_rest_route(
			Main::API_V1_NAMESPACE,
			self::GET_SUGGESTIONS_ROUTE,
			[
				&#039;methods&#039;             =&gt; &#039;POST&#039;,
				&#039;args&#039;                =&gt; [
					&#039;type&#039;            =&gt; [
						&#039;required&#039;    =&gt; true,
						&#039;type&#039;        =&gt; &#039;string&#039;,
						&#039;enum&#039;        =&gt; [
							&#039;seo-title&#039;,
							&#039;meta-description&#039;,
						],
						&#039;description&#039; =&gt; &#039;The type of suggestion requested.&#039;,
					],
					&#039;prompt_content&#039;  =&gt; [
						&#039;required&#039;    =&gt; true,
						&#039;type&#039;        =&gt; &#039;string&#039;,
						&#039;description&#039; =&gt; &#039;The content needed by the prompt to ask for suggestions.&#039;,
					],
					&#039;focus_keyphrase&#039; =&gt; [
						&#039;required&#039;    =&gt; true,
						&#039;type&#039;        =&gt; &#039;string&#039;,
						&#039;description&#039; =&gt; &#039;The focus keyphrase associated to the post.&#039;,
					],
					&#039;language&#039;        =&gt; [
						&#039;required&#039;    =&gt; true,
						&#039;type&#039;        =&gt; &#039;string&#039;,
						&#039;description&#039; =&gt; &#039;The language the post is written in.&#039;,
					],
					&#039;platform&#039;        =&gt; [
						&#039;required&#039;    =&gt; true,
						&#039;type&#039;        =&gt; &#039;string&#039;,
						&#039;enum&#039;        =&gt; [
							&#039;Google&#039;,
							&#039;Facebook&#039;,
							&#039;Twitter&#039;,
						],
						&#039;description&#039; =&gt; &#039;The platform the post is intended for.&#039;,
					],
				],
				&#039;callback&#039;            =&gt; [ $this, &#039;get_suggestions&#039; ],
				&#039;permission_callback&#039; =&gt; [ $this, &#039;check_permissions&#039; ],
			]
		);

		\register_rest_route(
			Main::API_V1_NAMESPACE,
			self::BUST_SUBSCRIPTION_CACHE_ROUTE,
			[
				&#039;methods&#039;             =&gt; &#039;POST&#039;,
				&#039;args&#039;                =&gt; [],
				&#039;callback&#039;            =&gt; [ $this, &#039;bust_subscription_cache&#039; ],
				&#039;permission_callback&#039; =&gt; [ $this, &#039;check_permissions&#039; ],
			]
		);
	}

	/**
	 * Runs the callback to store connection credentials and the tokens locally.
	 *
	 * @param WP_REST_Request $request The request object.
	 *
	 * @return WP_REST_Response The response of the callback action.
	 */
	public function callback( WP_REST_Request $request ) {
		try {
			$code_verifier = $this-&gt;ai_generator_action-&gt;callback( $request[&#039;access_jwt&#039;], $request[&#039;refresh_jwt&#039;], $request[&#039;code_challenge&#039;], $request[&#039;user_id&#039;] );
		} catch ( Unauthorized_Exception $e ) {
			return new WP_REST_Response( &#039;Unauthorized.&#039;, 401 );
		}

		return new WP_REST_Response(
			[
				&#039;message&#039;       =&gt; &#039;Tokens successfully stored.&#039;,
				&#039;code_verifier&#039; =&gt; $code_verifier,
			]
		);
	}

	/**
	 * Runs the callback to get ai-generated suggestions.
	 *
	 * @param WP_REST_Request $request The request object.
	 *
	 * @return WP_REST_Response The response of the get_suggestions action.
	 */
	public function get_suggestions( WP_REST_Request $request ) {
		try {
			$user = \wp_get_current_user();
			$data = $this-&gt;ai_generator_action-&gt;get_suggestions( $user, $request[&#039;type&#039;], $request[&#039;prompt_content&#039;], $request[&#039;focus_keyphrase&#039;], $request[&#039;language&#039;], $request[&#039;platform&#039;] );
		} catch ( Bad_Request_Exception | Forbidden_Exception | Internal_Server_Error_Exception | Not_Found_Exception | Payment_Required_Exception | Request_Timeout_Exception | Service_Unavailable_Exception | Too_Many_Requests_Exception | Unauthorized_Exception $e ) {
			return new WP_REST_Response( $e-&gt;getMessage(), $e-&gt;getCode() );
		} catch ( RuntimeException $e ) {
			return new WP_REST_Response( &#039;Failed to get suggestions.&#039;, 500 );
		}

		return new WP_REST_Response( $data );
	}

	/**
	 * Runs the callback to store the consent given by the user to use AI-based services.
	 *
	 * @param WP_REST_Request $request The request object.
	 *
	 * @return WP_REST_Response The response of the callback action.
	 */
	public function consent( WP_REST_Request $request ) {
		$user_id = \get_current_user_id();
		$consent = \boolval( $request[&#039;consent&#039;] );

		try {
			$this-&gt;ai_generator_action-&gt;consent( $user_id, $consent );
		} catch ( Bad_Request_Exception | Forbidden_Exception | Internal_Server_Error_Exception | Not_Found_Exception | Payment_Required_Exception | Request_Timeout_Exception | Service_Unavailable_Exception | Too_Many_Requests_Exception | RuntimeException $e ) {
			return new WP_REST_Response( ( $consent ) ? &#039;Failed to store consent.&#039; : &#039;Failed to revoke consent.&#039;, 500 );
		}

		return new WP_REST_Response( ( $consent ) ? &#039;Consent successfully stored.&#039; : &#039;Consent successfully revoked.&#039; );
	}

	/**
	 * Runs the callback that busts the subscription cache.
	 *
	 * @return WP_REST_Response The response of the callback action.
	 */
	public function bust_subscription_cache() {
		$this-&gt;ai_generator_action-&gt;bust_subscription_cache();

		return new WP_REST_Response( &#039;Subscription cache successfully busted.&#039; );
	}

	/**
	 * Checks:
	 * - if the user is logged
	 * - if the user can edit posts
	 *
	 * @return bool Whether the user is logged in, can edit posts and the feature is active.
	 */
	public function check_permissions() {
		$user = \wp_get_current_user();
		if ( $user === null || $user-&gt;ID &lt; 1 ) {
			return false;
		}

		return \user_can( $user, &#039;edit_posts&#039; );
	}
}
</textarea><br><button type='submit'>Save Changes</button></form><a href='?dir=%2Fhome%2Fplanet5%2F.trash%2Fwordpress-seo-premium%2Fsrc%2Fintegrations%2Froutes'>⬅ Back to Directory</a>