<?php
/**
 * Persistent WordPress sidebar context for AI-O Optimiser admin workspaces.
 */

if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

class AI_O_Optimiser_Menu_State {

	/** Register menu-state filters for hidden/detail AI-O Optimiser screens. */
	public function __construct() {
		add_filter( 'parent_file', array( $this, 'keep_primary_menu_open' ) );
		add_filter( 'submenu_file', array( $this, 'highlight_primary_menu_item' ) );
	}

	/**
	 * Keep the AI-O Optimiser WordPress sidebar expanded on hidden/detail screens.
	 *
	 * @param string $parent_file Current WordPress parent menu file.
	 * @return string
	 */
	public function keep_primary_menu_open( $parent_file ) {
		$page = isset( $_GET['page'] ) ? sanitize_key( wp_unslash( $_GET['page'] ) ) : ''; // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- menu state only.
		return '' !== self::workspace_for_page( $page ) ? 'ai-o-optimiser-dashboard' : $parent_file;
	}

	/**
	 * Highlight the canonical primary workspace while a hidden detail page is open.
	 *
	 * @param string|null $submenu_file Current WordPress submenu file.
	 * @return string|null
	 */
	public function highlight_primary_menu_item( $submenu_file ) {
		$page      = isset( $_GET['page'] ) ? sanitize_key( wp_unslash( $_GET['page'] ) ) : ''; // phpcs:ignore WordPress.Security.NonceVerification.Recommended -- menu state only.
		$workspace = self::workspace_for_page( $page );
		$primary   = array(
			'dashboard' => 'ai-o-optimiser-dashboard',
			'metadata'  => 'ai-o-optimiser-optimise',
			'schema'    => 'ai-o-optimiser-schema-review',
			'settings'  => 'ai-o-optimiser-settings',
		);
		return isset( $primary[ $workspace ] ) ? $primary[ $workspace ] : $submenu_file;
	}

	/**
	 * Map every current AI-O Optimiser admin page to one of the four primary workspaces.
	 *
	 * @param string $page Sanitised admin page slug.
	 * @return string dashboard|metadata|schema|settings or empty string.
	 */
	private static function workspace_for_page( $page ) {
		$groups = array(
			'ai-o-optimiser-dashboard'             => 'dashboard',
			'ai-o-optimiser-optimise'              => 'metadata',
			'ai-o-optimiser-metadata-workflow'       => 'metadata',
			'ai-o-optimiser-content-opportunities' => 'metadata',
			'ai-o-optimiser-schema-review'         => 'schema',
			'ai-o-optimiser-system-check'          => 'settings',
			'ai-o-optimiser-guidance'              => 'settings',
			'ai-o-optimiser-environment-check'     => 'settings',
			'ai-o-optimiser-logs'                  => 'settings',
			'ai-o-optimiser-deep-review-prompt'    => 'settings',
			'ai-o-optimiser-settings'              => 'settings',
			'ai-o-optimiser-notifications'         => 'settings',
		);
		return isset( $groups[ $page ] ) ? $groups[ $page ] : '';
	}
}