<?php
defined( 'ABSPATH' ) || exit;

class AgeOnce_Restrictions {

	public function __construct() {
		add_action( 'wp_enqueue_scripts', array( $this, 'enqueue_content_gate_assets' ), 20 );
		add_action( 'add_meta_boxes', array( $this, 'add_universal_meta_box' ) );
		add_action( 'save_post', array( $this, 'save_universal_meta_box' ) );
		add_action( 'woocommerce_product_options_general_product_data', array( $this, 'add_product_meta_field' ) );
		add_action( 'woocommerce_process_product_meta', array( $this, 'save_product_meta_field' ) );
		add_filter( 'the_content', array( $this, 'filter_content' ) );
	}

	/**
	 * True when a post term equals a restricted term, or any ancestor of the post term is restricted (parent category rule matches child-only assignment).
	 *
	 * @param int    $term_id             Term attached to the post.
	 * @param string $tax_slug            Taxonomy slug.
	 * @param array  $restricted_term_ids Term IDs from rules (may be int or string).
	 */
	private static function post_term_matches_restricted_list( $term_id, $tax_slug, array $restricted_term_ids ) {
		if ( in_array( $term_id, $restricted_term_ids ) || in_array( (string) $term_id, $restricted_term_ids ) ) {
			return true;
		}
		$ancestors = get_ancestors( (int) $term_id, $tax_slug );
		foreach ( $ancestors as $ancestor_id ) {
			if ( in_array( $ancestor_id, $restricted_term_ids ) || in_array( (string) $ancestor_id, $restricted_term_ids ) ) {
				return true;
			}
		}
		return false;
	}

	public static function get_required_age_for_content( $post_id ) {
		if ( ! $post_id ) {
			return 0;
		}

		$post_type = get_post_type( $post_id );
		$rules     = (array) get_option( 'ageonce_rules', array() );

		if ( ! isset( $rules[ $post_type ]['enabled'] ) || $rules[ $post_type ]['enabled'] !== 'yes' ) {
			return 0;
		}

		$pt_rules = $rules[ $post_type ];

		if ( isset( $pt_rules['manual_lock_enabled'] ) && $pt_rules['manual_lock_enabled'] === 'yes' ) {
			$manual_check = get_post_meta( $post_id, '_ageonce_restricted', true );
			if ( $manual_check === 'yes' ) {
				return 18;
			}
		}

		$ages_to_check = array( 21, 18, 16 );

		foreach ( $ages_to_check as $age ) {
			if ( ! isset( $pt_rules['enabled_ages'] ) || ! in_array( (string) $age, (array) $pt_rules['enabled_ages'] ) ) {
				continue;
			}

			if ( isset( $pt_rules['ages'][ $age ]['taxonomies'] ) ) {
				foreach ( $pt_rules['ages'][ $age ]['taxonomies'] as $tax_slug => $restricted_term_ids ) {
					if ( empty( $restricted_term_ids ) ) {
						continue;
					}

					$post_terms = wp_get_post_terms( $post_id, $tax_slug, array( 'fields' => 'ids' ) );
					if ( is_wp_error( $post_terms ) ) {
						continue;
					}

					foreach ( $post_terms as $term_id ) {
						if ( self::post_term_matches_restricted_list( $term_id, $tax_slug, $restricted_term_ids ) ) {
							return $age;
						}
					}
				}
			}
		}

		return 0;
	}

	/**
	 * Load content gate styles when the current singular view may show the gate.
	 */
	public function enqueue_content_gate_assets() {
		if ( ! is_singular() ) {
			return;
		}
		if ( 'yes' !== get_option( 'ageonce_enabled' ) ) {
			return;
		}

		$post_id = get_queried_object_id();
		if ( ! $post_id ) {
			return;
		}

		if ( ! self::should_replace_front_content( $post_id ) ) {
			return;
		}

		$required_age = self::get_required_age_for_content( $post_id );
		if ( $required_age <= 0 || AgeOnce_API::is_verified( true, $required_age ) ) {
			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-content-gate',
			plugins_url( 'assets/css/content-gate.css', $base ),
			array(),
			$version
		);
	}

	public static function is_product_restricted( $product_id ) {
		return self::get_required_age_for_content( $product_id ) > 0;
	}

	/**
	 * WooCommerce product pages stay visible by default. Product rules still
	 * require Verify Age at checkout. Posts and pages keep the content gate.
	 *
	 * @param int $post_id Post or product ID.
	 * @return bool
	 */
	public static function should_replace_front_content( $post_id ) {
		$post_id = absint( $post_id );
		if ( $post_id <= 0 ) {
			return false;
		}

		if ( get_post_type( $post_id ) !== 'product' ) {
			return true;
		}

		$rules = (array) get_option( 'ageonce_rules', array() );
		return isset( $rules['product']['gate_product_pages'] ) && $rules['product']['gate_product_pages'] === 'yes';
	}

	public function filter_content( $content ) {
		global $post;
		if ( ! $post ) {
			return $content;
		}

		if ( ! self::should_replace_front_content( $post->ID ) ) {
			return $content;
		}

		$required_age = self::get_required_age_for_content( $post->ID );
		if ( $required_age <= 0 || AgeOnce_API::is_verified( true, $required_age ) ) {
			return $content;
		}

		return $this->get_gate_html( $required_age, $post->ID );
	}

	private function get_gate_html( $age, $post_id ) {
		$verification_url = $this->get_verification_url( $age, $post_id );

		ob_start();
		?>
		<div class="ageonce-content-gate">

			<div class="ageonce-content-gate-sign" aria-hidden="true">
				<?php echo esc_html( $age ); ?>+
			</div>

			<h3 class="ageonce-content-gate-title">
				<?php esc_html_e( 'Restricted Content', 'age-verification-with-ageonce' ); ?>
			</h3>

			<p class="ageonce-content-gate-lead">
				<?php
				printf(
					/* translators: %s: minimum age */
					esc_html__( 'You must be %s years or older to view this content. Please verify your age.', 'age-verification-with-ageonce' ),
					esc_html( $age )
				);
				?>
			</p>

			<a href="<?php echo esc_url( $verification_url ); ?>" class="button button-primary ageonce-gate-btn">
				<?php esc_html_e( 'Verify Age', 'age-verification-with-ageonce' ); ?>
			</a>

		</div>
		<?php
		return ob_get_clean();
	}

	private function get_verification_url( $age = 18, $post_id = null ) {
		$return_url = $post_id ? get_permalink( $post_id ) : get_permalink();
		if ( ! is_string( $return_url ) || $return_url === '' ) {
			$return_url = home_url( '/' );
		}

		return AgeOnce_API::get_verify_age_url( $return_url, 'content', absint( $age ) );
	}

	public function add_universal_meta_box() {
		$rules     = (array) get_option( 'ageonce_rules', array() );
		$post_type = get_post_type();

		if ( isset( $rules[ $post_type ]['manual_lock_enabled'] ) && $rules[ $post_type ]['manual_lock_enabled'] === 'yes' ) {
			add_meta_box(
				'ageonce_restriction_box',
				__( 'AgeOnce Protection', 'age-verification-with-ageonce' ),
				array( $this, 'render_meta_box' ),
				$post_type,
				'side',
				'default'
			);
		}
	}

	public function render_meta_box( $post ) {
		$value = get_post_meta( $post->ID, '_ageonce_restricted', true );
		wp_nonce_field( 'ageonce_save', 'ageonce_nonce' );
		echo '<label><input type="checkbox" name="_ageonce_restricted" value="yes" ' . checked( $value, 'yes', false ) . '> ';
		esc_html_e( 'Require Age Verification', 'age-verification-with-ageonce' );
		echo '</label>';
	}

	public function save_universal_meta_box( $post_id ) {
		if ( ! isset( $_POST['ageonce_nonce'] ) || ! wp_verify_nonce( sanitize_text_field( wp_unslash( $_POST['ageonce_nonce'] ) ), 'ageonce_save' ) ) {
			return;
		}
		if ( defined( 'DOING_AUTOSAVE' ) && DOING_AUTOSAVE ) {
			return;
		}
		if ( ! current_user_can( 'edit_post', $post_id ) ) {
			return;
		}
		update_post_meta( $post_id, '_ageonce_restricted', isset( $_POST['_ageonce_restricted'] ) ? 'yes' : 'no' );
	}

	public function add_product_meta_field() {
		echo '<div class="options_group">';
		wp_nonce_field( 'ageonce_product_meta', 'ageonce_product_meta_nonce' );
		woocommerce_wp_checkbox( array(
			'id'          => '_ageonce_restricted',
			'label'       => __( 'Age Verification Required', 'age-verification-with-ageonce' ),
			'desc_tip'    => true,
			'description' => __( 'Require age verification at checkout for this item.', 'age-verification-with-ageonce' ),
		) );
		echo '</div>';
	}

	public function save_product_meta_field( $post_id ) {
		if ( ! isset( $_POST['ageonce_product_meta_nonce'] ) || ! wp_verify_nonce( sanitize_text_field( wp_unslash( $_POST['ageonce_product_meta_nonce'] ) ), 'ageonce_product_meta' ) ) {
			return;
		}
		if ( ! current_user_can( 'edit_post', $post_id ) ) {
			return;
		}
		$is_restricted = isset( $_POST['_ageonce_restricted'] ) ? 'yes' : 'no';
		update_post_meta( $post_id, '_ageonce_restricted', $is_restricted );
	}
}