<?php
defined( 'ABSPATH' ) || exit;

/**
 * Age verification gate for WooCommerce My Account registration.
 */
class AgeOnce_Registration {

	/** Allowed minimum ages for registration. */
	const ALLOWED_AGES = array( '16', '18', '21' );

	public function __construct() {
		add_action( 'wp_enqueue_scripts', array( $this, 'enqueue_assets' ), 20 );
		add_filter( 'body_class', array( $this, 'registration_body_class' ) );
		add_action( 'woocommerce_register_form', array( $this, 'render_age_gate_area' ), 5 );
		add_filter( 'woocommerce_process_registration_errors', array( $this, 'validate_registration' ), 10, 1 );
		add_action( 'woocommerce_created_customer', array( $this, 'save_user_verification_meta' ), 20, 1 );
	}

	/**
	 * Whether registration age verification is enabled in settings (and plugin is on).
	 *
	 * @return bool
	 */
	public static function is_feature_enabled() {
		if ( 'yes' !== get_option( 'ageonce_enabled' ) ) {
			return false;
		}
		return 'yes' === get_option( 'ageonce_registration_enabled', 'no' );
	}

	/**
	 * Configured minimum age for registration (16, 18, or 21).
	 *
	 * @return int
	 */
	public static function get_required_age() {
		$age = (string) absint( get_option( 'ageonce_registration_age', '18' ) );
		if ( ! in_array( $age, self::ALLOWED_AGES, true ) ) {
			$age = '18';
		}
		return (int) $age;
	}

	/**
	 * My Account page with registration form available to guests.
	 *
	 * @return bool
	 */
	private function is_registration_context() {
		if ( ! function_exists( 'is_account_page' ) || ! is_account_page() ) {
			return false;
		}
		if ( is_user_logged_in() ) {
			return false;
		}
		if ( function_exists( 'is_wc_endpoint_url' ) && ( is_wc_endpoint_url( 'lost-password' ) || is_wc_endpoint_url( 'customer-logout' ) ) ) {
			return false;
		}
		if ( 'yes' !== get_option( 'woocommerce_enable_myaccount_registration' ) ) {
			return false;
		}
		return true;
	}

	/**
	 * Whether the registration submit control should stay locked.
	 *
	 * @return bool
	 */
	private function registration_submit_locked() {
		if ( ! self::is_feature_enabled() || ! $this->is_registration_context() ) {
			return false;
		}
		return ! AgeOnce_API::is_verified( true, self::get_required_age() );
	}

	/**
	 * @param string[] $classes Body classes.
	 * @return string[]
	 */
	public function registration_body_class( $classes ) {
		if ( $this->registration_submit_locked() ) {
			$classes[] = 'ageonce-registration-requires-verification';
		}
		return $classes;
	}

	/**
	 * Enqueue styles always on registration when feature is on; JS only while locked.
	 */
	public function enqueue_assets() {
		if ( ! self::is_feature_enabled() || ! $this->is_registration_context() ) {
			return;
		}

		$version = defined( 'AGEONCE_VERIFICATION_VERSION' ) ? AGEONCE_VERIFICATION_VERSION : '1.0.0';
		$base    = defined( 'AGEONCE_VERIFICATION_PLUGIN_FILE' ) ? AGEONCE_VERIFICATION_PLUGIN_FILE : dirname( __DIR__ ) . '/ageonce-verification.php';

		wp_enqueue_style(
			'ageonce-checkout-age-gate',
			plugins_url( 'assets/css/checkout-age-gate.css', $base ),
			array(),
			$version
		);
		wp_enqueue_style(
			'ageonce-registration-age-gate',
			plugins_url( 'assets/css/registration-age-gate.css', $base ),
			array( 'ageonce-checkout-age-gate' ),
			$version
		);
		wp_enqueue_style( 'dashicons' );

		if ( ! $this->registration_submit_locked() ) {
			return;
		}

		wp_enqueue_script(
			'ageonce-registration-age-gate',
			plugins_url( 'assets/js/registration-age-gate.js', $base ),
			array( 'jquery' ),
			$version,
			true
		);

		wp_localize_script(
			'ageonce-registration-age-gate',
			'ageonceRegistration',
			array(
				'disableButtonTitle' => __( 'Please verify your age above', 'age-verification-with-ageonce' ),
			)
		);
	}

	/**
	 * Render age gate (or verified notice) inside the WooCommerce register form.
	 */
	public function render_age_gate_area() {
		if ( ! self::is_feature_enabled() ) {
			return;
		}

		$required_age = self::get_required_age();
		$is_verified  = AgeOnce_API::is_verified( true, $required_age );

		if ( $is_verified ) {
			?>
			<div class="woocommerce-message ageonce-checkout-verified">
				<span class="dashicons dashicons-yes ageonce-checkout-verified-icon" aria-hidden="true"></span>
				<span class="ageonce-checkout-verified-text">
					<?php esc_html_e( 'Age verified. You can create your store account.', 'age-verification-with-ageonce' ); ?>
				</span>
			</div>
			<?php
			return;
		}

		$return_url = function_exists( 'wc_get_page_permalink' ) ? wc_get_page_permalink( 'myaccount' ) : home_url( '/' );
		$oauth_url  = AgeOnce_API::get_verify_age_url( is_string( $return_url ) && $return_url !== '' ? $return_url : home_url( '/' ), 'signup', $required_age );
		?>
		<div class="ageonce-gate-container ageonce-registration-gate">
			<h4 class="ageonce-gate-title"><?php esc_html_e( 'Age Verification Required', 'age-verification-with-ageonce' ); ?></h4>
			<p class="ageonce-gate-lead">
				<?php
				printf(
					/* translators: %d: minimum age required */
					esc_html__( 'You must verify that you are %d+ before creating a store account.', 'age-verification-with-ageonce' ),
					absint( $required_age )
				);
				?>
			</p>
			<a href="<?php echo esc_url( $oauth_url ); ?>" class="button alt ageonce-verify-btn ageonce-verify-btn--full">
				<?php
				printf(
					/* translators: %d: minimum age required */
					esc_html__( 'Verify Age (%d+)', 'age-verification-with-ageonce' ),
					absint( $required_age )
				);
				?>
			</a>
			<div class="ageonce-gate-footnote">
				<?php esc_html_e( 'Secure one-time check', 'age-verification-with-ageonce' ); ?>
			</div>
		</div>
		<?php
	}

	/**
	 * Server-side block for My Account registration.
	 *
	 * @param WP_Error $validation_error Existing validation errors.
	 * @return WP_Error
	 */
	public function validate_registration( $validation_error ) {
		if ( ! self::is_feature_enabled() ) {
			return $validation_error;
		}

		if ( ! ( $validation_error instanceof WP_Error ) ) {
			$validation_error = new WP_Error();
		}

		if ( ! AgeOnce_API::is_verified( true, self::get_required_age() ) ) {
			$validation_error->add(
				'ageonce_registration_unverified',
				__( 'Please verify your age before creating a store account.', 'age-verification-with-ageonce' )
			);
		}

		return $validation_error;
	}

	/**
	 * Persist verification outcome on the new customer when session is verified.
	 *
	 * @param int $customer_id New user ID.
	 */
	public function save_user_verification_meta( $customer_id ) {
		$customer_id = absint( $customer_id );
		if ( $customer_id <= 0 ) {
			return;
		}
		if ( ! AgeOnce_API::is_verified( true, self::get_required_age() ) ) {
			return;
		}

		update_user_meta( $customer_id, '_ageonce_status', 'verified' );

		$tx_id = ( function_exists( 'WC' ) && WC()->session ) ? WC()->session->get( 'ageonce_transaction_id' ) : null;
		if ( $tx_id ) {
			update_user_meta( $customer_id, '_ageonce_transaction_id', sanitize_text_field( (string) $tx_id ) );
		}

		$verified_at = ( function_exists( 'WC' ) && WC()->session ) ? WC()->session->get( 'ageonce_timestamp' ) : null;
		if ( ! $verified_at ) {
			$verified_at = current_time( 'mysql' );
		}
		update_user_meta( $customer_id, '_ageonce_timestamp', sanitize_text_field( (string) $verified_at ) );
		update_user_meta( $customer_id, '_ageonce_required_age', self::get_required_age() );
	}
}