<?php

declare( strict_types=1 );

namespace Rooster\Core;

defined( 'ABSPATH' ) || exit;

final class DeviceClass {

	private static ?string $current = null;

	
	private static array $siblings = [];

	public static function boot(): void {
		add_action( 'send_headers', [ self::class, 'headers' ] );
		add_filter( 'body_class', [ self::class, 'body_class' ] );
		add_filter( 'render_block_data', [ self::class, 'resolve_part' ], 6 ); 
		self::cache_adapters();
	}

	
	public static function current(): string {
		if ( null !== self::$current ) {
			return self::$current;
		}
		$hint = $_SERVER['HTTP_SEC_CH_UA_MOBILE'] ?? null;
		if ( null !== $hint ) {
			$class = '?1' === trim( (string) $hint ) ? 'mobile' : 'desktop';
		} else {
			$ua    = (string) ( $_SERVER['HTTP_USER_AGENT'] ?? '' );
			$class = wp_is_mobile() || preg_match( '/iPad|Tablet|Android(?!.*Mobile)/i', $ua ) ? 'mobile' : 'desktop';
		}
		self::$current = (string) apply_filters( 'rooster/device/class', $class );
		return self::$current;
	}

	public static function enabled(): bool {
		return (bool) \AIEO_Engine_Settings::get( 'device_variants', true );
	}

	
	public static function reset(): void {
		self::$current  = null;
		self::$siblings = [];
	}

	public static function headers(): void {
		if ( headers_sent() || is_admin() ) {
			return;
		}
		header( 'Accept-CH: Sec-CH-UA-Mobile' );
		header( 'Critical-CH: Sec-CH-UA-Mobile' );
		if ( self::enabled() ) {
			
			header( 'Vary: Sec-CH-UA-Mobile', false );
		}
	}

	
	public static function body_class( $classes ): array {
		$classes   = (array) $classes;
		$classes[] = 'rooster-device-' . self::current();
		return $classes;
	}

	
	public static function resolve_part( $parsed_block ) {
		if ( ! is_array( $parsed_block ) || 'core/template-part' !== ( $parsed_block['blockName'] ?? '' ) ) {
			return $parsed_block;
		}
		if ( is_admin() ) {
			return $parsed_block;
		}
		$slug = (string) ( $parsed_block['attrs']['slug'] ?? '' );
		if ( '' === $slug ) {
			return $parsed_block;
		}
		if ( ! self::enabled() || 'mobile' !== self::current() || str_ends_with( $slug, '-mobile' ) ) {
			return $parsed_block;
		}
		$variant = $slug . '-mobile';
		if ( ! isset( self::$siblings[ $slug ] ) ) {
			self::$siblings[ $slug ] = (bool) apply_filters(
				'rooster/device/has_variant',
				is_file( get_stylesheet_directory() . '/parts/' . $variant . '.html' ) || is_file( get_template_directory() . '/parts/' . $variant . '.html' ),
				$slug
			);
		}
		if ( self::$siblings[ $slug ] ) {
			$parsed_block['attrs']['slug'] = $variant;
		}
		return $parsed_block;
	}

	
	private static function cache_adapters(): void {
		if ( ! self::enabled() ) {
			return;
		}
		
		add_filter( 'rocket_cache_mobile', '__return_true' );
		add_filter( 'rocket_cache_mobile_files_tablet', static fn () => 'mobile' );
		add_filter( 'rocket_cache_query_strings', static fn ( $qs ) => array_values( array_unique( array_merge( (array) $qs, [ 'rooster_layer' ] ) ) ) );
		
		add_filter( 'litespeed_vary', static fn ( $vary ) => array_merge( (array) $vary, [ 'rooster_device' => self::current() ] ) );
		add_filter( 'litespeed_vary_cookies', static fn ( $cookies ) => (array) $cookies );
		
		do_action( 'rooster/device/cache_adapters', self::current() );
	}
}