<?php
// If this file is called directly, abort.
if ( ! defined( 'WPINC' ) ) {
	die;
}

/**
 * Manages AIO Tools standalone WordPress Drop-in files:
 * 1. wp-content/advanced-cache.php + WP_CACHE constant in wp-config.php
 * 2. Runtime static config in wp-content/uploads/aio-cache/cache-config.php
 */
class AIO_Dropin_Manager {

	const ADV_SIGNATURE = 'AIO_TOOLS_ADVANCED_CACHE_DROPIN';

	/**
	 * Flush PHP stat cache and Zend OPcache for a drop-in or config file immediately.
	 */
	public static function invalidate_php_file_cache( $path ) {
		if ( empty( $path ) ) {
			return;
		}
		clearstatcache( true, $path );
		if ( function_exists( 'opcache_invalidate' ) ) {
			@opcache_invalidate( $path, true ); // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged
		}
	}

	/**
	 * Return normalized path to wp-content/advanced-cache.php.
	 */
	public static function get_advanced_cache_path() {
		return wp_normalize_path( trailingslashit( WP_CONTENT_DIR ) . 'advanced-cache.php' );
	}

	/**
	 * Locate wp-config.php (standard root or one directory above ABSPATH).
	 */
	public static function get_wp_config_path() {
		$root_cfg = wp_normalize_path( ABSPATH . 'wp-config.php' );
		if ( file_exists( $root_cfg ) ) {
			return $root_cfg;
		}
		$parent_cfg = wp_normalize_path( dirname( ABSPATH ) . '/wp-config.php' );
		if ( file_exists( $parent_cfg ) && ! file_exists( dirname( ABSPATH ) . '/wp-settings.php' ) ) {
			return $parent_cfg;
		}
		return '';
	}

	/**
	 * Return normalized cache directory path inside uploads.
	 */
	public static function get_cache_dir() {
		$upload_dir = wp_upload_dir();
		return trailingslashit( wp_normalize_path( $upload_dir['basedir'] ) ) . 'aio-cache/';
	}

	/**
	 * Inspect current advanced-cache.php & WP_CACHE status.
	 */
	public static function get_advanced_cache_status() {
		$adv_path     = self::get_advanced_cache_path();
		$cfg_path     = self::get_wp_config_path();
		$exists       = file_exists( $adv_path );
		$is_ours      = false;
		$foreign_name = '';

		if ( $exists && is_readable( $adv_path ) ) {
			$head    = (string) @file_get_contents( $adv_path, false, null, 0, 2048 ); // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged
			$is_ours = strpos( $head, self::ADV_SIGNATURE ) !== false;
			if ( ! $is_ours ) {
				if ( stripos( $head, 'WP Rocket' ) !== false ) {
					$foreign_name = 'WP Rocket';
				} elseif ( stripos( $head, 'LiteSpeed' ) !== false ) {
					$foreign_name = 'LiteSpeed Cache';
				} elseif ( stripos( $head, 'W3 Total Cache' ) !== false ) {
					$foreign_name = 'W3 Total Cache';
				} elseif ( stripos( $head, 'WP-Super-Cache' ) !== false ) {
					$foreign_name = 'WP Super Cache';
				} else {
					$foreign_name = __( 'Another Cache Plugin', 'aio-tools' );
				}
			}
		}

		$wp_cache_on      = defined( 'WP_CACHE' ) && WP_CACHE;
		$wp_config_write  = ! empty( $cfg_path ) && is_writable( $cfg_path );
		$wp_content_write = ( $exists && is_writable( $adv_path ) ) || ( ! $exists && is_writable( WP_CONTENT_DIR ) );

		return array(
			'exists'           => $exists,
			'is_ours'          => $is_ours,
			'foreign_name'     => $foreign_name,
			'wp_cache_const'   => $wp_cache_on,
			'wp_config_write'  => $wp_config_write,
			'wp_content_write' => $wp_content_write,
			'active'           => $exists && $is_ours && $wp_cache_on,
		);
	}

	/**
	 * Write static runtime cache configuration & MIME .htaccess inside wp-content/uploads/aio-cache/.
	 */
	public static function write_runtime_config( $options = null ) {
		if ( ! is_array( $options ) ) {
			$options = get_option( 'aio_options', array() );
		}

		$cache_dir = self::get_cache_dir();
		if ( ! file_exists( $cache_dir ) ) {
			wp_mkdir_p( $cache_dir );
		}
		if ( ! is_dir( $cache_dir ) || ! is_writable( $cache_dir ) ) {
			return false;
		}

		// Protect .php files in aio-cache and configure Apache/LiteSpeed .gz HTML serving
		$cache_htaccess = $cache_dir . '.htaccess';
		$ht_rules       = implode(
			"\n",
			array(
				'# AIO Tools Cache Directory Rules',
				'<FilesMatch "\.php$">',
				'	<IfModule mod_authz_core.c>',
				'		Require all denied',
				'	</IfModule>',
				'	<IfModule !mod_authz_core.c>',
				'		Order allow,deny',
				'		Deny from all',
				'	</IfModule>',
				'</FilesMatch>',
				'<IfModule mod_mime.c>',
				'	AddType text/html .gz',
				'	AddEncoding gzip .gz',
				'	AddDefaultCharset UTF-8',
				'</IfModule>',
				'<IfModule mod_headers.c>',
				'	Header set Vary "Accept-Encoding, Cookie"',
				'	Header set Cache-Control "no-cache, no-store, must-revalidate"',
				'</IfModule>',
				'',
			)
		);
		@file_put_contents( $cache_htaccess, $ht_rules ); // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged
		@file_put_contents( $cache_dir . 'index.php', "<?php\n// Silence is golden.\n" ); // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged

		$perf_active   = ! isset( $options['opt-module-performance'] ) || ! empty( $options['opt-module-performance'] );
		$maint_active  = ! empty( $options['optEnableMaintenance'] );
		$cache_enabled = $perf_active && ! empty( $options['optEnablePageCache'] ) && ! $maint_active;

		$ttl_hours = isset( $options['optCacheTtlHours'] ) ? absint( $options['optCacheTtlHours'] ) : 24;
		$ttl_secs  = $ttl_hours > 0 ? ( $ttl_hours * HOUR_IN_SECONDS ) : 0;

		$exclude_urls = array();
		if ( ! empty( $options['optCacheExcludeUrls'] ) ) {
			foreach ( preg_split( '/[\r\n]+/', (string) $options['optCacheExcludeUrls'] ) as $u ) {
				$u = trim( $u );
				if ( $u !== '' ) {
					$exclude_urls[] = $u;
				}
			}
		}

		$exclude_cookies = array();
		if ( ! empty( $options['optCacheExcludeCookies'] ) ) {
			foreach ( preg_split( '/[\r\n]+/', (string) $options['optCacheExcludeCookies'] ) as $c ) {
				$c = trim( $c );
				if ( $c !== '' ) {
					$exclude_cookies[] = $c;
				}
			}
		}

		$runtime = array(
			'enabled'         => (bool) $cache_enabled,
			'cache_dir'       => $cache_dir,
			'expected_host'   => strtolower( (string) wp_parse_url( home_url(), PHP_URL_HOST ) ),
			'ttl'             => (int) $ttl_secs,
			'separate_mobile' => ! empty( $options['optCacheSeparateMobile'] ),
			'ignore_tracking' => ! isset( $options['optCacheIgnoreTracking'] ) || ! empty( $options['optCacheIgnoreTracking'] ),
			'exclude_urls'    => $exclude_urls,
			'exclude_cookies' => $exclude_cookies,
		);

		$cfg_file = $cache_dir . 'cache-config.php';
		$export   = "<?php\nif ( ! defined( 'ABSPATH' ) ) {\n\texit;\n}\nreturn " . var_export( $runtime, true ) . ";\n";
		$ok       = (bool) @file_put_contents( $cfg_file, $export, LOCK_EX ); // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged
		if ( $ok ) {
			self::invalidate_php_file_cache( $cfg_file );
		}
		return $ok;
	}

	/**
	 * Synchronize wp-content/advanced-cache.php & WP_CACHE in wp-config.php.
	 *
	 * @param bool $enable        Whether drop-in should be installed.
	 * @param bool $force_replace Overwrite foreign advanced-cache.php if true.
	 * @param array|null $options Current aio_options array.
	 * @return bool
	 */
	public static function sync_advanced_cache( $enable, $force_replace = false, $options = null ) {
		self::write_runtime_config( $options );

		$adv_path = self::get_advanced_cache_path();

		if ( ! $enable ) {
			if ( file_exists( $adv_path ) && is_readable( $adv_path ) ) {
				$head = (string) @file_get_contents( $adv_path, false, null, 0, 2048 ); // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged
				if ( strpos( $head, self::ADV_SIGNATURE ) !== false ) {
					@unlink( $adv_path ); // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged
					self::invalidate_php_file_cache( $adv_path );
					self::set_wp_cache_constant( false );
				}
			}
			return true;
		}

		// Do not clobber another plugin's advanced-cache.php unless user explicitly clicked Overwrite
		if ( file_exists( $adv_path ) && is_readable( $adv_path ) && ! $force_replace ) {
			$head = (string) @file_get_contents( $adv_path, false, null, 0, 2048 ); // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged
			if ( strpos( $head, self::ADV_SIGNATURE ) === false ) {
				return false;
			}
		}

		if ( ( file_exists( $adv_path ) && ! is_writable( $adv_path ) ) || ( ! file_exists( $adv_path ) && ! is_writable( WP_CONTENT_DIR ) ) ) {
			return false;
		}

		$config_file_path = self::get_cache_dir() . 'cache-config.php';
		$dropin_code      = self::build_advanced_cache_code( $config_file_path );

		$written = @file_put_contents( $adv_path, $dropin_code, LOCK_EX ); // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged
		if ( $written === false ) {
			return false;
		}
		self::invalidate_php_file_cache( $adv_path );

		self::set_wp_cache_constant( true );
		return true;
	}

	/**
	 * Safely toggle define( 'WP_CACHE', true/false ) inside wp-config.php.
	 */
	public static function set_wp_cache_constant( $enable ) {
		$cfg_path = self::get_wp_config_path();
		if ( empty( $cfg_path ) || ! is_readable( $cfg_path ) || ! is_writable( $cfg_path ) ) {
			return false;
		}

		$contents = (string) @file_get_contents( $cfg_path ); // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged
		if ( strlen( $contents ) < 100 || strpos( $contents, '<?php' ) === false ) {
			return false;
		}

		$pattern = '/^[ \t]*define\s*\(\s*[\'"]WP_CACHE[\'"]\s*,\s*(?:true|false|1|0)\s*\)\s*;.*$/mi';

		if ( $enable ) {
			$line = "define( 'WP_CACHE', true ); // Added by AIO Tools";
			if ( preg_match( $pattern, $contents ) ) {
				$updated = preg_replace( $pattern, $line, $contents, 1 );
			} else {
				$updated = preg_replace( '/(<\?php\s*)/i', "$1\n" . $line . "\n", $contents, 1 );
			}
		} else {
			// Only remove or disable if added by AIO Tools or set to false safely
			if ( strpos( $contents, '// Added by AIO Tools' ) !== false ) {
				$updated = preg_replace( '/^[ \t]*define\s*\(\s*[\'"]WP_CACHE[\'"].*Added by AIO Tools.*(?:\r?\n)?/mi', '', $contents );
			} else {
				return true;
			}
		}

		if ( ! is_string( $updated ) || strlen( $updated ) < 80 ) {
			return false;
		}

		if ( $updated === $contents ) {
			return true;
		}

		$ok = (bool) @file_put_contents( $cfg_path, $updated, LOCK_EX ); // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged
		if ( $ok ) {
			self::invalidate_php_file_cache( $cfg_path );
		}
		return $ok;
	}

	/**
	 * Generate standalone zero-DB advanced-cache.php drop-in PHP source.
	 */
	private static function build_advanced_cache_code( $config_file_path ) {
		$escaped_cfg_path = addslashes( wp_normalize_path( $config_file_path ) );

		return <<<PHP
<?php
/**
 * AIO Tools Standalone Full-Page Cache Drop-in
 * Signature: AIO_TOOLS_ADVANCED_CACHE_DROPIN
 * Version: 3.1.0
 */
if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

if ( ! defined( 'AIO_ADVANCED_CACHE_ACTIVE' ) ) {
	define( 'AIO_ADVANCED_CACHE_ACTIVE', true );
}

( static function () {
	if ( ! isset( \$_SERVER['REQUEST_METHOD'] ) || \$_SERVER['REQUEST_METHOD'] !== 'GET' ) {
		return;
	}
	if ( defined( 'WP_ADMIN' ) || defined( 'DOING_AJAX' ) || defined( 'DOING_CRON' ) || defined( 'XMLRPC_REQUEST' ) || defined( 'REST_REQUEST' ) ) {
		return;
	}
	if ( isset( \$_GET['aio_no_opt'] ) && (string) \$_GET['aio_no_opt'] === '1' ) {
		return;
	}

	\$cfg_file = '{$escaped_cfg_path}';
	if ( ! is_file( \$cfg_file ) ) {
		return;
	}
	\$cfg = include \$cfg_file;
	if ( ! is_array( \$cfg ) || empty( \$cfg['enabled'] ) || empty( \$cfg['cache_dir'] ) || empty( \$cfg['expected_host'] ) ) {
		return;
	}

	// Verify Host Header matches site host (allowing www. prefix equivalence)
	\$raw_host = isset( \$_SERVER['HTTP_HOST'] ) ? strtolower( trim( (string) \$_SERVER['HTTP_HOST'] ) ) : '';
	\$req_host = preg_replace( '/:\d+$/', '', \$raw_host );
	\$norm_req = preg_replace( '/^www\./', '', \$req_host );
	\$norm_exp = preg_replace( '/^www\./', '', strtolower( (string) \$cfg['expected_host'] ) );
	if ( \$norm_req === '' || \$norm_req !== \$norm_exp ) {
		return;
	}

	\$raw_uri = isset( \$_SERVER['REQUEST_URI'] ) ? (string) \$_SERVER['REQUEST_URI'] : '/';
	if ( strpos( \$raw_uri, "\\0" ) !== false || strpos( \$raw_uri, '\\\\' ) !== false || strpos( \$raw_uri, '..' ) !== false ) {
		return;
	}

	\$uri_path = (string) parse_url( \$raw_uri, PHP_URL_PATH );

	// Fast WebP fallback & static 404 shield for /wp-content/ assets when .htaccess/Nginx rules are absent
	if ( strpos( \$uri_path, '/wp-content/' ) === 0 && preg_match( '/\.(?:jpe?g|png|gif|webp|avif|svg|ico|woff2?|ttf|otf|eot|css|js)$/i', \$uri_path ) ) {
		\$rel_path = rawurldecode( ltrim( \$uri_path, '/' ) );
		if ( strpos( \$rel_path, '..' ) === false && strpos( \$rel_path, "\\0" ) === false ) {
			\$abs_file = rtrim( str_replace( '\\\\', '/', ABSPATH ), '/' ) . '/' . \$rel_path;
			if ( preg_match( '/\.(?:jpe?g|png)$/i', \$abs_file ) && ! is_file( \$abs_file ) ) {
				\$webp_cand = preg_replace( '/\.(?:jpe?g|png)$/i', '.webp', \$abs_file );
				if ( \$webp_cand && is_file( \$webp_cand ) && filesize( \$webp_cand ) > 100 ) {
					header( 'Content-Type: image/webp' );
					header( 'Content-Length: ' . filesize( \$webp_cand ) );
					header( 'Cache-Control: public, max-age=31536000, immutable' );
					header( 'X-AIO-WebP: HIT-DROPIN' );
					readfile( \$webp_cand );
					exit;
				}
			}
			if ( ! is_file( \$abs_file ) ) {
				http_response_code( 404 );
				header( 'Content-Type: text/plain; charset=UTF-8' );
				header( 'Cache-Control: public, max-age=60' );
				header( 'X-AIO-Static-404: 1' );
				echo '404 Not Found';
				exit;
			}
		}
		return;
	}

	// Inspect query parameters (optionally ignore marketing/analytics tracking params)
	if ( ! empty( \$_GET ) ) {
		if ( empty( \$cfg['ignore_tracking'] ) ) {
			return;
		}
		\$ignored = array(
			'utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content', 'utm_id',
			'fbclid', 'gclid', 'dclid', 'msclkid', 'ttclid', 'twclid', 'yclid', 'zarsrc',
			'_ga', '_gl', 'mc_cid', 'mc_eid',
		);
		foreach ( array_keys( \$_GET ) as \$qkey ) {
			if ( ! in_array( strtolower( (string) \$qkey ), \$ignored, true ) ) {
				return;
			}
		}
	}

	// Check login, WooCommerce cart, password-protected post, and custom excluded cookies
	if ( ! empty( \$_COOKIE ) ) {
		\$cookie_prefixes = array(
			'wordpress_logged_in_',
			'wp-postpass_',
			'woocommerce_items_in_cart',
			'wp_woocommerce_session_',
			'aio_maint_bypass',
			'comment_author_',
		);
		if ( ! empty( \$cfg['exclude_cookies'] ) && is_array( \$cfg['exclude_cookies'] ) ) {
			\$cookie_prefixes = array_merge( \$cookie_prefixes, \$cfg['exclude_cookies'] );
		}
		foreach ( array_keys( \$_COOKIE ) as \$ckey ) {
			foreach ( \$cookie_prefixes as \$pref ) {
				if ( \$pref !== '' && stripos( (string) \$ckey, \$pref ) !== false ) {
					return;
				}
			}
		}
	}

	if ( preg_match( '/\.(?:xml|txt|json|php|ico|css|js|jpe?g|png|gif|webp|avif|svg|woff2?|ttf|otf|eot)$/i', \$uri_path ) ) {
		return;
	}
	if ( preg_match( '#/wp-(?:admin|login|json|cron)#i', \$uri_path ) ) {
		return;
	}

	if ( ! empty( \$cfg['exclude_urls'] ) && is_array( \$cfg['exclude_urls'] ) ) {
		foreach ( \$cfg['exclude_urls'] as \$ex_pat ) {
			if ( \$ex_pat !== '' && stripos( \$uri_path, \$ex_pat ) !== false ) {
				return;
			}
		}
	}

	\$segments      = array_filter( explode( '/', trim( \$uri_path, '/' ) ), 'strlen' );
	\$safe_segments = array();
	foreach ( \$segments as \$seg ) {
		\$clean = preg_replace( '/[^A-Za-z0-9._~%-]/', '_', \$seg );
		if ( \$clean === '' || \$clean === '.' || \$clean === '..' ) {
			return;
		}
		\$safe_segments[] = substr( \$clean, 0, 120 );
	}

	\$is_https = ( ! empty( \$_SERVER['HTTPS'] ) && strtolower( (string) \$_SERVER['HTTPS'] ) !== 'off' )
		|| ( isset( \$_SERVER['SERVER_PORT'] ) && (string) \$_SERVER['SERVER_PORT'] === '443' )
		|| ( isset( \$_SERVER['HTTP_X_FORWARDED_PROTO'] ) && strtolower( (string) \$_SERVER['HTTP_X_FORWARDED_PROTO'] ) === 'https' );

	\$is_mobile = false;
	if ( ! empty( \$cfg['separate_mobile'] ) && ! empty( \$_SERVER['HTTP_USER_AGENT'] ) ) {
		\$ua        = (string) \$_SERVER['HTTP_USER_AGENT'];
		\$is_mobile = (bool) preg_match( '/Mobile|Android|Silk\/|Kindle|BlackBerry|Opera (?:Mini|Mobi)|iPhone|iPod/i', \$ua );
	}

	\$scheme_prefix = \$is_https ? 'https-' : 'http-';
	\$sub_path      = ! empty( \$safe_segments ) ? implode( '/', \$safe_segments ) . '/' : '';
	\$file_name     = \$is_mobile ? 'index-mobile.html' : 'index.html';
	\$cache_dir     = str_replace( '\\\\', '/', \$cfg['cache_dir'] );
	\$cache_file    = \$cache_dir . \$scheme_prefix . \$cfg['expected_host'] . '/' . \$sub_path . \$file_name;

	if ( strpos( \$cache_file, \$cache_dir ) !== 0 || ! is_file( \$cache_file ) ) {
		return;
	}

	\$ttl = isset( \$cfg['ttl'] ) ? (int) \$cfg['ttl'] : 86400;
	if ( \$ttl > 0 && ( time() - (int) filemtime( \$cache_file ) ) > \$ttl ) {
		return;
	}

	\$accept_enc = isset( \$_SERVER['HTTP_ACCEPT_ENCODING'] ) ? strtolower( (string) \$_SERVER['HTTP_ACCEPT_ENCODING'] ) : '';
	\$gz_file    = \$cache_file . '.gz';
	\$can_gzip   = ( strpos( \$accept_enc, 'gzip' ) !== false ) && ! ini_get( 'zlib.output_compression' ) && is_file( \$gz_file );

	header( 'Content-Type: text/html; charset=UTF-8' );
	header( 'Vary: Accept-Encoding' );

	if ( \$can_gzip ) {
		header( 'Content-Encoding: gzip' );
		header( 'Content-Length: ' . filesize( \$gz_file ) );
		header( 'X-AIO-Cache: HIT-DROPIN-GZIP' );
		readfile( \$gz_file );
		exit;
	}

	header( 'X-AIO-Cache: HIT-DROPIN' );
	readfile( \$cache_file );
	exit;
} )();

PHP;
	}
}