<?php
/**
 * REST handlers for the setup wizard.
 *
 * @package AcidStudios\AEO
 */

declare(strict_types=1);

namespace AcidStudios\AEO\Rest\Handlers;

use AcidStudios\AEO\Audit\Loopback;
use AcidStudios\AEO\Audit\Runner;
use AcidStudios\AEO\Cache;
use AcidStudios\AEO\Modules;
use AcidStudios\AEO\Options;
use AcidStudios\AEO\Rest\Controller;
use WP_Error;
use WP_REST_Request;

defined( 'ABSPATH' ) || exit;

/**
 * `POST /wizard/step` and `POST /wizard/complete`.
 *
 * A step saves whatever it collected and records how far the administrator has
 * got; the recorded step never goes backwards, so returning to an earlier step
 * to change an answer does not make the wizard forget the later ones.
 *
 * Completing does not flush the rewrite rules, it asks for a flush. The modules
 * the wizard has just switched on did not register their files on this request
 * — they were off when the plugin booted — so writing the rule set out now
 * would persist a set that still lacks them. `Plugin::maybe_flush_rewrites()`
 * does it on the next `init`, by which time the modules have registered.
 */
final class Wizard {

	/**
	 * The last step of the wizard, matching `general.wizard_step` in the schema.
	 *
	 * @var int
	 */
	private const LAST_STEP = 6;

	/**
	 * Argument schema of `POST /wizard/step`.
	 *
	 * @return array<string,array<string,mixed>> Argument schemas.
	 */
	public static function step_args(): array {
		return [
			'step' => [
				'type'              => 'integer',
				'required'          => true,
				'minimum'           => 0,
				'maximum'           => self::LAST_STEP,
				'sanitize_callback' => 'rest_sanitize_request_arg',
				'validate_callback' => 'rest_validate_request_arg',
			],
			'data' => Controller::object_arg(),
		];
	}

	/**
	 * Argument schema of `POST /wizard/complete`.
	 *
	 * @return array<string,array<string,mixed>> Argument schemas.
	 */
	public static function complete_args(): array {
		return [
			'submit_indexnow' => Controller::bool_arg( false ),
		];
	}

	/**
	 * Saves one wizard step.
	 *
	 * @param WP_REST_Request $request The request.
	 * @return array<string,mixed>|WP_Error Keys `settings` and `flush_needed`, or a 400.
	 */
	public function step( WP_REST_Request $request ) {
		$data  = $request->get_param( 'data' );
		$saved = Controller::save_sections( is_array( $data ) ? $data : [] );

		if ( is_wp_error( $saved ) ) {
			return $saved;
		}

		$step    = (int) $request->get_param( 'step' );
		$reached = (int) Options::get( 'general.wizard_step', 0 );

		if ( $step > $reached ) {
			Options::set( 'general.wizard_step', $step );
		}

		return [
			'settings'     => Options::all(),
			'flush_needed' => (bool) $saved['flush_needed'],
		];
	}

	/**
	 * Marks the wizard finished and reports where the site stands.
	 *
	 * @param WP_REST_Request $request The request.
	 * @return array<string,mixed> Keys `audit` and `indexnow_queued`.
	 */
	public function complete( WP_REST_Request $request ): array {
		Options::update_section(
			'general',
			[
				'wizard_completed' => true,
				'wizard_step'      => self::LAST_STEP,
			]
		);

		Cache::invalidate_all();
		Loopback::flush();
		Controller::forget_derived_data();
		update_option( Controller::FLUSH_OPTION, 1 );

		$queued = null;

		if ( (bool) $request->get_param( 'submit_indexnow' ) ) {
			$indexnow = Controller::module( 'indexnow', Modules\IndexNow::class );
			$queued   = $indexnow instanceof Modules\IndexNow ? $indexnow->submit_all() : 0;
		}

		return [
			'audit'           => Runner::run( true ),
			'indexnow_queued' => $queued,
		];
	}
}