<?php
/**
 * AI Search Admin Manager
 * 
 * Handles all admin interface functionality
 */

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

class AI_Search_Admin_Manager {
    
    /**
     * @var AI_Search_Settings_Pages
     */
    private $settings_pages;
    
    /**
     * @var AI_Search_Setup_Wizard
     */
    private $setup_wizard;

    /**
     * @var AI_Search_Post_Meta_Box
     */
    private $post_meta_box;

    /**
     * Initialize admin manager
     */
    public function __construct() {
        $this->load_dependencies();
        $this->init_components();
        $this->register_hooks();
    }

    /**
     * Load required files
     */
    private function load_dependencies() {
        require_once plugin_dir_path( __FILE__ ) . 'class-settings-pages.php';
        require_once plugin_dir_path( __FILE__ ) . 'class-setup-wizard.php';
        require_once plugin_dir_path( __FILE__ ) . 'class-post-meta-box.php';
        require_once plugin_dir_path( __FILE__ ) . 'class-deactivation-feedback.php';
    }

    /**
     * Initialize components
     */
    private function init_components() {
        $this->settings_pages = new AI_Search_Settings_Pages();
        $this->setup_wizard = new AI_Search_Setup_Wizard();
        $this->post_meta_box = new AI_Search_Post_Meta_Box();
        new AI_Search_Deactivation_Feedback();
    }
    
    /**
     * Register admin hooks
     */
    private function register_hooks() {
        add_action( 'admin_menu', [ $this, 'register_settings_menu' ] );
        add_action( 'admin_notices', [ $this->setup_wizard, 'show_setup_notice' ] );
        add_action( 'admin_notices', [ $this->setup_wizard, 'show_setup_success_notice' ] );
        add_action( 'admin_notices', [ $this, 'show_domain_mismatch_notice' ] );
        add_action( 'admin_notices', [ $this, 'show_woocommerce_welcome_notice' ] );
        add_action( 'admin_notices', [ $this, 'show_quota_warning_banner' ] );
        add_action( 'admin_notices', [ $this, 'show_pro_upgrade_banner' ] );
        add_action( 'admin_post_ai_search_snooze_quota_banner', [ $this, 'snooze_quota_banner' ] );
        add_action( 'admin_post_ai_search_dismiss_quota_banner_forever', [ $this, 'dismiss_quota_banner_forever' ] );
        add_action( 'admin_post_ai_search_dismiss_pro_banner_forever', [ $this, 'dismiss_pro_banner_forever' ] );
        add_action( 'admin_enqueue_scripts', [ $this, 'enqueue_admin_assets' ] );
        add_action( 'admin_enqueue_scripts', [ $this->setup_wizard, 'enqueue_scripts' ] );
        add_action( 'admin_head', [ $this, 'add_menu_icon_styles' ] );
        add_filter( 'parent_file', [ $this, 'fix_admin_parent_file' ] );
        add_filter( 'submenu_file', [ $this, 'fix_admin_submenu_file' ] );

        // Admin post actions
        add_action( 'admin_post_ai_search_generate_embeddings', [ $this->settings_pages, 'generate_selected_embeddings' ] );
        add_action( 'admin_post_ai_search_generate_term_embeddings', [ $this->settings_pages, 'generate_term_embeddings_bulk' ] );
        add_action( 'admin_post_ai_search_clear_cache', [ $this->settings_pages, 'clear_embeddings_cache' ] );
        add_action( 'admin_post_ai_search_clear_cache_filtered', [ $this->settings_pages, 'clear_embeddings_cache_filtered' ] );
        add_action( 'admin_post_ai_search_validate_token', [ $this->settings_pages, 'validate_service_token' ] );
        add_action( 'admin_post_ai_search_recover_token', [ $this->settings_pages, 'recover_service_token' ] );
        add_action( 'admin_post_ai_search_save_custom_fields', [ $this->settings_pages, 'save_custom_fields_settings' ] );
        add_action( 'admin_post_ai_search_save_woocommerce_fields', [ $this->settings_pages, 'save_woocommerce_fields_settings' ] );
        add_action( 'admin_post_ai_search_refresh_quota', [ $this->settings_pages, 'refresh_quota_data' ] );
        add_action( 'admin_post_ai_search_claim_subscription', [ $this->settings_pages, 'handle_claim_subscription' ] );
        add_action( 'admin_post_ai_search_release_slot', [ $this->settings_pages, 'handle_release_slot' ] );
        add_action( 'admin_post_ai_search_dismiss_reindex_notice', [ $this->settings_pages, 'handle_dismiss_reindex_notice' ] );
        add_action( 'admin_post_ai_search_complete_setup', [ $this->setup_wizard, 'complete_setup' ] );
        add_action( 'admin_post_ai_search_reconnect_service', [ $this, 'reconnect_service' ] );

        // AJAX actions
        add_action( 'wp_ajax_ai_search_validate_token', [ $this->settings_pages, 'validate_service_token' ] );
        add_action( 'wp_ajax_ai_search_get_taxonomies', [ $this, 'ajax_get_taxonomies' ] );
        add_action( 'wp_ajax_ai_search_get_terms', [ $this, 'ajax_get_terms' ] );
        add_action( 'wp_ajax_ai_search_dismiss_woo_welcome', [ $this, 'ajax_dismiss_woo_welcome' ] );
        add_action( 'admin_post_ai_search_dismiss_blocks_notice', [ $this, 'dismiss_blocks_notice' ] );
        add_action( 'admin_post_ai_search_snooze_pro_banner', [ $this, 'snooze_pro_banner' ] );
        add_action( 'wp_ajax_ai_search_wizard_generate_batch', [ $this->setup_wizard, 'ajax_generate_batch' ] );
        add_action( 'wp_ajax_ai_search_wizard_prepare_sync', [ $this->setup_wizard, 'ajax_prepare_sync' ] );
        add_action( 'wp_ajax_ai_search_dismiss_wizard', [ $this->setup_wizard, 'ajax_dismiss_wizard' ] );
        add_action( 'wp_ajax_ai_search_dismiss_setup_success', [ $this->setup_wizard, 'ajax_dismiss_setup_success' ] );
        add_action( 'wp_ajax_ai_search_test_query', [ $this->settings_pages, 'ajax_test_query' ] );
    }

    /**
     * AJAX handler to get taxonomies for a post type.
     */
    public function ajax_get_taxonomies() {
        check_ajax_referer( 'ai_search_ajax' );

        if ( ! current_user_can( 'manage_options' ) ) {
            wp_send_json_error( 'Unauthorized' );
        }

        $post_type = isset( $_POST['post_type'] ) ? sanitize_text_field( $_POST['post_type'] ) : 'post';
        $taxonomies = get_object_taxonomies( $post_type, 'objects' );

        $result = [];
        foreach ( $taxonomies as $taxonomy ) {
            if ( $taxonomy->public && $taxonomy->show_ui ) {
                $result[ $taxonomy->name ] = [
                    'label' => $taxonomy->label,
                    'name'  => $taxonomy->name,
                ];
            }
        }

        wp_send_json_success( $result );
    }

    /**
     * AJAX handler to get terms for a taxonomy.
     */
    public function ajax_get_terms() {
        check_ajax_referer( 'ai_search_ajax' );

        if ( ! current_user_can( 'manage_options' ) ) {
            wp_send_json_error( 'Unauthorized' );
        }

        $taxonomy = isset( $_POST['taxonomy'] ) ? sanitize_text_field( $_POST['taxonomy'] ) : '';

        if ( empty( $taxonomy ) || ! taxonomy_exists( $taxonomy ) ) {
            wp_send_json_error( 'Invalid taxonomy' );
        }

        $terms = get_terms( [
            'taxonomy'   => $taxonomy,
            'hide_empty' => false,
            'number'     => 100,
        ] );

        if ( is_wp_error( $terms ) ) {
            wp_send_json_error( $terms->get_error_message() );
        }

        $result = [];
        foreach ( $terms as $term ) {
            $result[] = [
                'id'    => $term->term_id,
                'name'  => $term->name,
                'slug'  => $term->slug,
                'count' => $term->count,
            ];
        }

        wp_send_json_success( $result );
    }
    
    /**
     * Enqueue admin CSS and JS
     */
    public function enqueue_admin_assets( $hook ) {
        // Only load on our settings page (now it's a top-level menu)
        if ( 'toplevel_page_ai-search' !== $hook ) {
            return;
        }

        // Enqueue WordPress pointer for compatibility with other plugins
        wp_enqueue_script( 'wp-pointer' );
        wp_enqueue_style( 'wp-pointer' );

        wp_enqueue_style(
            'ai-search-admin-settings',
            plugins_url( 'css/admin-settings.css', __FILE__ ),
            [],
            AI_SEARCH_VERSION
        );
    }

    /**
     * Register settings menu in admin
     */
    public function register_settings_menu() {
        // Main menu page
        add_menu_page(
            'AI Search',                                    // Page title
            'AI Search',                                    // Menu title
            'manage_options',                               // Capability
            'ai-search',                                    // Menu slug
            [ $this->settings_pages, 'display_settings_page' ], // Callback
            AI_SEARCH_URL . 'assets/icon.svg',             // Icon
            58                                              // Position (before Settings)
        );

        // Submenu pages
        add_submenu_page(
            'ai-search',                                    // Parent slug
            'General Settings',                             // Page title
            'General Settings',                             // Menu title
            'manage_options',                               // Capability
            'ai-search',                                    // Menu slug (same as parent for first item)
            [ $this->settings_pages, 'display_settings_page' ] // Callback
        );

        // The Setup Wizard submenu only makes sense until the user has
        // walked through it. After that it's noise: the same settings live
        // in the tabs, and a permanent "Setup Wizard" item next to the
        // General/Search/Indexing surface reads like the plugin still
        // isn't configured. The wizard URL keeps working for anyone who
        // navigates to it manually.
        if ( ! get_option( 'ai_search_setup_completed', false ) ) {
            add_submenu_page(
                'ai-search',
                'Setup Wizard',
                'Setup Wizard',
                'manage_options',
                'ai-search&setup=wizard',
                [ $this->settings_pages, 'display_settings_page' ]
            );
        }

        // Upgrade submenu — Free plan only, AI Service provider only.
        //   * Paid customers already got what they were pitched, a permanent
        //     "Upgrade" item after checkout reads as unresolved status.
        //   * OpenAI (BYO key) users pay OpenAI directly and have no plan
        //     concept in our service — showing them "Upgrade — 7-day trial"
        //     would be misleading, so hide it.
        if ( 'ai_service' === get_option( 'ai_search_provider', 'ai_service' )
            && $this->is_on_free_plan() ) {
            add_submenu_page(
                'ai-search',
                __( 'Upgrade', 'ai-search' ),
                __( 'Upgrade: 7-day trial', 'ai-search' ),
                'manage_options',
                'ai-search-upgrade',
                [ $this, 'display_upgrade_page' ]
            );
        }

        add_submenu_page(
            'ai-search',
            'Support Forum',
            'Support Forum',
            'manage_options',
            'https://wordpress.org/support/plugin/ai-search/',
            ''
        );

        add_submenu_page(
            'ai-search',
            'Contact the Author',
            'Contact the Author',
            'manage_options',
            'https://samuelsilva.pt/lets-talk',
            ''
        );
    }
    
    /**
     * Whether this site is currently on the Free plan.
     *
     * Reads from the quota transient the Quota Manager keeps (6h cache) —
     * NEVER makes an HTTP call from here, because admin_menu fires on
     * every wp-admin page load and blocking the sidebar on a 20s request
     * to Railway would tank the whole dashboard. If the transient is
     * empty, default to Free: worst case the paid customer sees an
     * unnecessary "Upgrade" item until the next quota refresh — mildly
     * annoying, not broken.
     *
     * @return bool
     */
    private function is_on_free_plan() {
        $quota = get_transient( 'ai_search_quota_data' );
        if ( ! is_array( $quota ) ) {
            return true;
        }
        $plan = isset( $quota['plan'] )         ? $quota['plan']         : null;
        $sub  = isset( $quota['subscription'] ) ? $quota['subscription'] : null;
        $key  = is_array( $plan ) && isset( $plan['key'] ) ? (string) $plan['key'] : 'free';
        return ! ( 'free' !== $key && is_array( $sub ) );
    }

    /**
     * Render the Upgrade page — a full-height iframe of the pricing page
     * in embed mode so the customer never leaves wp-admin.
     */
    public function display_upgrade_page() {
        // `?minimal=1` triggers our own embed-mode CSS on wp-search.ai (hides
        // header/footer). Do NOT use `?embed=1` here: WordPress reserves that
        // query var for its oEmbed handler and would serve the minimal oEmbed
        // preview (title + excerpt + site name) instead of the actual page.
        $pricing_url = 'https://wp-search.ai/pricing/?minimal=1&utm_source=wp-admin&utm_medium=upgrade-menu';
        ?>
        <div class="wrap" style="margin:0;padding:0;max-width:none;">
            <style>
                #wpcontent, #wpbody-content { padding-left: 0 !important; }
                #wpbody-content { padding-bottom: 0 !important; }
                .ai-search-upgrade-frame {
                    display: block;
                    width: 100%;
                    height: calc(100vh - 32px);
                    border: 0;
                    background: #ffffff;
                }
                @media (max-width: 782px) {
                    .ai-search-upgrade-frame { height: calc(100vh - 46px); }
                }
            </style>
            <iframe
                class="ai-search-upgrade-frame"
                src="<?php echo esc_url( $pricing_url ); ?>"
                title="<?php esc_attr_e( 'AI Search pricing', 'ai-search' ); ?>"
                loading="eager"
                referrerpolicy="no-referrer-when-downgrade"></iframe>
        </div>
        <?php
    }

    /**
     * Get settings pages instance
     */
    public function get_settings_pages() {
        return $this->settings_pages;
    }
    
    /**
     * Get setup wizard instance
     */
    public function get_setup_wizard() {
        return $this->setup_wizard;
    }

    /**
     * Add custom CSS for menu icon sizing
     */
    public function add_menu_icon_styles() {
        ?>
        <style>
            #adminmenu #toplevel_page_ai-search img {
                width: 18px;
                height: 18px;
            }
        </style>
        <?php
    }

    /**
     * Fix parent file highlight for AI Search pages
     *
     * @param string $parent_file The parent file.
     * @return string
     */
    public function fix_admin_parent_file( $parent_file ) {
        global $pagenow;

        // Ensure AI Search menu is highlighted when on our pages
        if ( $pagenow === 'admin.php' && isset( $_GET['page'] ) && strpos( $_GET['page'], 'ai-search' ) === 0 ) {
            return 'ai-search';
        }

        return $parent_file;
    }

    /**
     * Fix submenu file highlight for AI Search pages
     *
     * @param string $submenu_file The submenu file.
     * @return string
     */
    public function fix_admin_submenu_file( $submenu_file ) {
        global $pagenow;

        if ( $pagenow === 'admin.php' && isset( $_GET['page'] ) && $_GET['page'] === 'ai-search' ) {
            // If we're on the wizard, highlight the wizard submenu
            if ( isset( $_GET['setup'] ) && $_GET['setup'] === 'wizard' ) {
                return 'ai-search&setup=wizard';
            }
            // Otherwise highlight the main settings
            return 'ai-search';
        }

        return $submenu_file;
    }

    /**
     * Show admin notice when domain doesn't match registered origin.
     */
    public function show_domain_mismatch_notice() {
        // Only show to admins
        if ( ! current_user_can( 'manage_options' ) ) {
            return;
        }

        // Only check if using AI Search Service (not OpenAI)
        $provider = get_option( 'ai_search_provider', 'ai_service' );
        if ( $provider !== 'ai_service' ) {
            return;
        }

        // Check if token exists
        $service_token = get_option( 'ai_search_service_token' );
        if ( empty( $service_token ) ) {
            return;
        }

        // Check if reconnection was successful
        if ( isset( $_GET['ai_search_reconnected'] ) && $_GET['ai_search_reconnected'] === 'success' ) {
            ?>
            <div class="notice notice-success is-dismissible">
                <p>
                    <strong><?php esc_html_e( 'AI Search', 'ai-search' ); ?>:</strong>
                    <?php esc_html_e( 'Successfully reconnected to AI Search Service with the new domain.', 'ai-search' ); ?>
                </p>
            </div>
            <?php
            return;
        }

        if ( isset( $_GET['ai_search_reconnected'] ) && $_GET['ai_search_reconnected'] === 'error' ) {
            ?>
            <div class="notice notice-error is-dismissible">
                <p>
                    <strong><?php esc_html_e( 'AI Search', 'ai-search' ); ?>:</strong>
                    <?php esc_html_e( 'Failed to reconnect to AI Search Service. Please try again or contact support.', 'ai-search' ); ?>
                </p>
            </div>
            <?php
            return;
        }

        $service = new AI_Search_Service();
        if ( $service->check_origin_match() ) {
            return;
        }

        $registered_origin = $service->get_registered_origin();
        $current_origin = get_site_url();
        $reconnect_url = wp_nonce_url(
            admin_url( 'admin-post.php?action=ai_search_reconnect_service' ),
            'ai_search_reconnect_service'
        );
        ?>
        <div class="notice notice-warning">
            <p>
                <strong><?php esc_html_e( 'AI Search - Domain Mismatch Detected', 'ai-search' ); ?></strong>
            </p>
            <p>
                <?php
                printf(
                    /* translators: 1: registered domain, 2: current domain */
                    esc_html__( 'Your AI Search Service token was registered for %1$s but your site is now running on %2$s. This may cause the search to fail.', 'ai-search' ),
                    '<code>' . esc_html( $registered_origin ) . '</code>',
                    '<code>' . esc_html( $current_origin ) . '</code>'
                );
                ?>
            </p>
            <p>
                <a href="<?php echo esc_url( $reconnect_url ); ?>" class="button button-primary">
                    <?php esc_html_e( 'Reconnect with Current Domain', 'ai-search' ); ?>
                </a>
                <span class="description" style="margin-left: 10px;">
                    <?php esc_html_e( 'This will register a new token for the current domain.', 'ai-search' ); ?>
                </span>
            </p>
        </div>
        <?php
    }

    /**
     * Handle service reconnection.
     */
    public function reconnect_service() {
        if ( ! current_user_can( 'manage_options' ) ) {
            wp_die( __( 'Unauthorized access', 'ai-search' ) );
        }

        check_admin_referer( 'ai_search_reconnect_service' );

        // Clear old token and origin
        delete_option( 'ai_search_service_token' );
        delete_option( 'ai_search_registered_origin' );

        // Register new token
        $service = new AI_Search_Service();
        $new_token = $service->register_origin();

        if ( $new_token ) {
            wp_redirect( admin_url( 'admin.php?page=ai-search&ai_search_reconnected=success' ) );
        } else {
            wp_redirect( admin_url( 'admin.php?page=ai-search&ai_search_reconnected=error' ) );
        }
        exit;
    }

    /**
     * Show a one-time welcome notice when WooCommerce is active.
     */
    public function show_woocommerce_welcome_notice() {
        if ( ! current_user_can( 'manage_options' ) ) {
            return;
        }

        if ( ! class_exists( 'WooCommerce' ) ) {
            return;
        }

        if ( ! get_option( 'ai_search_setup_completed', false ) ) {
            return;
        }

        if ( get_option( 'ai_search_woo_welcome_dismissed', false ) ) {
            return;
        }

        $settings_url = admin_url( 'admin.php?page=ai-search&tab=woocommerce' );
        $nonce        = wp_create_nonce( 'ai_search_dismiss_woo_welcome' );
        ?>
        <div class="notice notice-info is-dismissible ai-search-woo-welcome-notice" data-nonce="<?php echo esc_attr( $nonce ); ?>">
            <p>
                <strong><?php esc_html_e( 'AI Search', 'ai-search' ); ?>:</strong>
                <?php esc_html_e( 'WooCommerce detected. The plugin has been pre-configured with product-friendly defaults — short descriptions, categories, and attributes are already included in your store\'s search index.', 'ai-search' ); ?>
            </p>
            <p>
                <a href="<?php echo esc_url( $settings_url ); ?>" class="button button-primary">
                    <?php esc_html_e( 'Review product search settings', 'ai-search' ); ?>
                </a>
            </p>
        </div>
        <script>
        (function($){
            $(document).on('click', '.ai-search-woo-welcome-notice .notice-dismiss', function(){
                var nonce = $('.ai-search-woo-welcome-notice').data('nonce');
                $.post(ajaxurl, {
                    action: 'ai_search_dismiss_woo_welcome',
                    nonce: nonce
                });
            });
        })(jQuery);
        </script>
        <?php
    }

    /**
     * AJAX handler to permanently dismiss the WooCommerce welcome notice.
     */
    public function ajax_dismiss_woo_welcome() {
        check_ajax_referer( 'ai_search_dismiss_woo_welcome', 'nonce' );

        if ( ! current_user_can( 'manage_options' ) ) {
            wp_send_json_error( 'Unauthorized' );
        }

        update_option( 'ai_search_woo_welcome_dismissed', true );
        wp_send_json_success();
    }

    /**
     * Dismiss the "search as you type" announcement on the General tab.
     *
     * A plain admin-post link rather than AJAX: the notice lives inside the
     * settings form area, and a redirect back to the tab is the simplest
     * thing that works without JavaScript.
     *
     * @since 1.33.0
     */
    public function dismiss_blocks_notice() {
        check_admin_referer( 'ai_search_dismiss_blocks_notice' );

        if ( ! current_user_can( 'manage_options' ) ) {
            wp_die( esc_html__( 'You do not have permission to do this.', 'ai-search' ) );
        }

        update_option( 'ai_search_blocks_notice_dismissed', true );

        wp_safe_redirect( admin_url( 'admin.php?page=ai-search' ) );
        exit;
    }

    /**
     * Render a small "consider upgrading to a paid plan" banner on the
     * plugin's own admin pages.
     *
     * Guardrails:
     *   - never shown outside the plugin's own screens (wp.org guidelines
     *     scold plugins that nag on Dashboard, Plugins, Users, etc.)
     *   - never to paid customers (Standard/Pro users already have plan
     *     info in the Subscription panel)
     *   - never during the first-run experience (setup wizard not done
     *     yet - the user is still figuring out the free tier)
     *   - after the user dismisses it we back off for 60 days, then it
     *     may return once
     *
     * @since 1.34.1
     */
    public function show_pro_upgrade_banner() {
        if ( ! current_user_can( 'manage_options' ) ) {
            return;
        }
        $screen = get_current_screen();
        if ( ! $screen || false === strpos( (string) $screen->id, 'ai-search' ) ) {
            return;
        }
        // Do not stack on top of the Subscription hero after a fresh claim,
        // or the reindex reminder card - both live on the General tab and
        // deserve to be the star for that first load.
        if ( get_transient( 'ai_search_claim_success' ) ) {
            return;
        }
        if ( ! get_option( 'ai_search_setup_completed', false ) ) {
            return;
        }
        // v1.35.0: if the quota banner is going to render this request, don't
        // stack another purple upsell above/below it - the quota story is
        // more urgent and specific, let it stand alone.
        if ( $this->resolve_quota_banner_variant() ) {
            return;
        }
        // v1.35.0: permanent opt-out. The 60-day snooze is fine when the
        // user is undecided; anyone who clicks "Don't show again" has
        // decided, and we owe them silence.
        if ( get_option( 'ai_search_pro_banner_dismissed_forever', false ) ) {
            return;
        }
        $snoozed_until = (int) get_option( 'ai_search_pro_banner_snoozed_until', 0 );
        if ( $snoozed_until > time() ) {
            return;
        }
        // Skip paid customers.
        $quota = get_transient( 'ai_search_quota_data' );
        if ( is_array( $quota ) && isset( $quota['plan']['key'] ) && 'free' !== $quota['plan']['key'] ) {
            return;
        }

        $pricing_url = 'https://wp-search.ai/pricing/?utm_source=wp-admin&utm_medium=pro-banner';
        $dismiss_url = wp_nonce_url(
            admin_url( 'admin-post.php?action=ai_search_snooze_pro_banner' ),
            'ai_search_snooze_pro_banner'
        );
        $forever_url = wp_nonce_url(
            admin_url( 'admin-post.php?action=ai_search_dismiss_pro_banner_forever' ),
            'ai_search_dismiss_pro_banner_forever'
        );
        ?>
        <div class="ai-search-pro-banner">
            <div class="ai-search-pro-banner__icon" aria-hidden="true">
                <svg width="22" height="22" viewBox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg">
                    <path d="M12 3 L14 10 L21 12 L14 14 L12 21 L10 14 L3 12 L10 10 Z" stroke="currentColor" stroke-width="1.6" stroke-linejoin="round"/>
                </svg>
            </div>
            <div class="ai-search-pro-banner__body">
                <strong class="ai-search-pro-banner__title"><?php esc_html_e( 'Get materially more precise results', 'ai-search' ); ?></strong>
                <span class="ai-search-pro-banner__lede">
                    <?php esc_html_e( 'Standard upgrades this site to text-embedding-3-large and raises the monthly cap to 25,000. 7-day free trial - launch offer 35% off for six months.', 'ai-search' ); ?>
                </span>
            </div>
            <div class="ai-search-pro-banner__actions">
                <a href="<?php echo esc_url( $pricing_url ); ?>" class="ai-search-pro-banner__cta" target="_blank" rel="noopener"><?php esc_html_e( 'See plans', 'ai-search' ); ?></a>
                <span class="ai-search-pro-banner__dismisses">
                    <a href="<?php echo esc_url( $dismiss_url ); ?>" class="ai-search-pro-banner__dismiss" aria-label="<?php esc_attr_e( 'Snooze upgrade banner for 60 days', 'ai-search' ); ?>"><?php esc_html_e( 'Not now', 'ai-search' ); ?></a>
                    <span class="ai-search-pro-banner__dismiss-sep" aria-hidden="true">&middot;</span>
                    <a href="<?php echo esc_url( $forever_url ); ?>" class="ai-search-pro-banner__dismiss ai-search-pro-banner__dismiss--forever" aria-label="<?php esc_attr_e( 'Dismiss upgrade banner permanently', 'ai-search' ); ?>"><?php esc_html_e( 'Don\'t show again', 'ai-search' ); ?></a>
                </span>
            </div>
        </div>
        <?php
    }

    /**
     * Decide which quota-warning variant, if any, should render this
     * request. Returns 'quota_warning' (>= 80% used), 'quota_exhausted'
     * (>= 100% used), or null (do not show). Also honours the preview
     * query var so the design can be reviewed on any account state.
     *
     * The 80/100 thresholds compare the WORSE of the daily and monthly
     * fractions - a store that has burned 100% of today's throttle wants
     * the same red banner as one that used up its monthly cap.
     *
     * @since 1.35.0
     *
     * @return string|null 'quota_warning' | 'quota_exhausted' | null
     */
    private function resolve_quota_banner_variant() {
        // Preview mode: any admin can force a variant with ?ai_search_preview=...
        // on a plugin page to review the design. This bypasses plan/quota
        // checks so paid customers or fresh installs can see it too.
        $preview = isset( $_GET['ai_search_preview'] ) ? sanitize_key( (string) $_GET['ai_search_preview'] ) : '';
        if ( in_array( $preview, [ 'quota_warning', 'quota_exhausted' ], true ) ) {
            return $preview;
        }

        $quota = get_transient( 'ai_search_quota_data' );
        if ( ! is_array( $quota ) ) {
            return null;
        }
        // Only nag Free plans. Paid customers already have room.
        $plan_key = isset( $quota['plan']['key'] ) ? (string) $quota['plan']['key'] : 'free';
        if ( 'free' !== $plan_key ) {
            return null;
        }

        $used_daily  = (int) ( $quota['daily_usage'] ?? 0 );
        $limit_daily = (int) ( $quota['daily_limit'] ?? 0 );
        $used_total  = (int) ( $quota['total_embeddings_used'] ?? 0 );
        $limit_total = (int) ( $quota['total_embeddings_limit'] ?? 0 );

        $daily_pct = $limit_daily > 0 ? ( $used_daily / $limit_daily ) : 0;
        $total_pct = $limit_total > 0 ? ( $used_total / $limit_total ) : 0;
        $worst     = max( $daily_pct, $total_pct );

        if ( $worst >= 1.0 ) {
            return 'quota_exhausted';
        }
        if ( $worst >= 0.8 ) {
            return 'quota_warning';
        }
        return null;
    }

    /**
     * Render the quota-warning banner when Free-plan usage approaches or
     * hits the cap. Two variants (see resolve_quota_banner_variant()),
     * both dismissable via a short (3-day) snooze because the underlying
     * situation is dynamic - a purchase reset or a fresh month can flip
     * the state on its own.
     *
     * @since 1.35.0
     */
    public function show_quota_warning_banner() {
        if ( ! current_user_can( 'manage_options' ) ) {
            return;
        }
        $screen = get_current_screen();
        if ( ! $screen || false === strpos( (string) $screen->id, 'ai-search' ) ) {
            return;
        }
        // Do not stack on the claim-success hero or a fresh install without
        // setup. Same reasoning as show_pro_upgrade_banner().
        if ( get_transient( 'ai_search_claim_success' ) ) {
            return;
        }
        if ( ! get_option( 'ai_search_setup_completed', false ) ) {
            return;
        }
        // Snooze / permanent-dismiss respect - both ignored when previewing
        // so the design can be reviewed even after the user opted out.
        $preview = isset( $_GET['ai_search_preview'] ) ? sanitize_key( (string) $_GET['ai_search_preview'] ) : '';
        if ( ! $preview && get_option( 'ai_search_quota_banner_dismissed_forever', false ) ) {
            return;
        }
        $snoozed_until = (int) get_option( 'ai_search_quota_banner_snoozed_until', 0 );
        if ( ! $preview && $snoozed_until > time() ) {
            return;
        }

        $variant = $this->resolve_quota_banner_variant();
        if ( ! $variant ) {
            return;
        }

        $pricing_url = 'https://wp-search.ai/pricing/?utm_source=wp-admin&utm_medium=quota-banner&utm_content=' . rawurlencode( $variant );
        $dismiss_url = wp_nonce_url(
            admin_url( 'admin-post.php?action=ai_search_snooze_quota_banner' ),
            'ai_search_snooze_quota_banner'
        );
        $forever_url = wp_nonce_url(
            admin_url( 'admin-post.php?action=ai_search_dismiss_quota_banner_forever' ),
            'ai_search_dismiss_quota_banner_forever'
        );

        if ( 'quota_exhausted' === $variant ) {
            $title = __( 'Monthly cap reached - new content is no longer being indexed', 'ai-search' );
            $lede  = __( 'You have used every embedding on the Free plan. Existing search keeps working, but new posts and updates will not be searchable until you upgrade or the monthly counter resets.', 'ai-search' );
            $cta   = __( 'Upgrade to keep indexing', 'ai-search' );
        } else {
            $title = __( 'You are running low on Free-plan embeddings', 'ai-search' );
            $lede  = __( 'At the current pace you will hit the monthly cap soon. Standard raises it to 25,000/month, runs on the more precise text-embedding-3-large model, and starts with a 7-day free trial.', 'ai-search' );
            $cta   = __( 'See Standard', 'ai-search' );
        }
        ?>
        <div class="ai-search-quota-banner ai-search-quota-banner--<?php echo esc_attr( $variant ); ?>">
            <div class="ai-search-quota-banner__icon" aria-hidden="true">
                <svg width="22" height="22" viewBox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg">
                    <path d="M12 3 L22 20 H2 Z" stroke="currentColor" stroke-width="1.6" stroke-linejoin="round"/>
                    <path d="M12 10 V14" stroke="currentColor" stroke-width="1.8" stroke-linecap="round"/>
                    <circle cx="12" cy="17" r="1" fill="currentColor"/>
                </svg>
            </div>
            <div class="ai-search-quota-banner__body">
                <strong class="ai-search-quota-banner__title"><?php echo esc_html( $title ); ?></strong>
                <span class="ai-search-quota-banner__lede"><?php echo esc_html( $lede ); ?></span>
            </div>
            <div class="ai-search-quota-banner__actions">
                <a href="<?php echo esc_url( $pricing_url ); ?>" class="ai-search-quota-banner__cta" target="_blank" rel="noopener"><?php echo esc_html( $cta ); ?></a>
                <span class="ai-search-quota-banner__dismisses">
                    <a href="<?php echo esc_url( $dismiss_url ); ?>" class="ai-search-quota-banner__dismiss" aria-label="<?php esc_attr_e( 'Snooze quota banner for 3 days', 'ai-search' ); ?>"><?php esc_html_e( 'Not now', 'ai-search' ); ?></a>
                    <span class="ai-search-quota-banner__dismiss-sep" aria-hidden="true">&middot;</span>
                    <a href="<?php echo esc_url( $forever_url ); ?>" class="ai-search-quota-banner__dismiss ai-search-quota-banner__dismiss--forever" aria-label="<?php esc_attr_e( 'Dismiss quota banner permanently', 'ai-search' ); ?>"><?php esc_html_e( 'Don\'t show again', 'ai-search' ); ?></a>
                </span>
            </div>
        </div>
        <?php
    }

    /**
     * Snooze the quota banner for 3 days. Short window on purpose: at 80%
     * or 100% usage, the underlying situation changes fast (a monthly
     * reset, a claim, a spike), and hiding it for 60 days like the Pro
     * banner would risk them missing a real block later.
     *
     * @since 1.35.0
     */
    public function snooze_quota_banner() {
        check_admin_referer( 'ai_search_snooze_quota_banner' );

        if ( ! current_user_can( 'manage_options' ) ) {
            wp_die( esc_html__( 'You do not have permission to do this.', 'ai-search' ) );
        }

        update_option( 'ai_search_quota_banner_snoozed_until', time() + 3 * DAY_IN_SECONDS );

        $referer = wp_get_referer();
        wp_safe_redirect( $referer ? $referer : admin_url( 'admin.php?page=ai-search' ) );
        exit;
    }

    /**
     * Permanently dismiss the quota banner. Fired from the "Don't show
     * again" link. The user has decided; respect it.
     *
     * @since 1.35.0
     */
    public function dismiss_quota_banner_forever() {
        check_admin_referer( 'ai_search_dismiss_quota_banner_forever' );

        if ( ! current_user_can( 'manage_options' ) ) {
            wp_die( esc_html__( 'You do not have permission to do this.', 'ai-search' ) );
        }

        update_option( 'ai_search_quota_banner_dismissed_forever', true );

        $referer = wp_get_referer();
        wp_safe_redirect( $referer ? $referer : admin_url( 'admin.php?page=ai-search' ) );
        exit;
    }

    /**
     * Permanently dismiss the Pro upgrade banner. Fired from the "Don't
     * show again" link that sits next to the 60-day snooze.
     *
     * @since 1.35.0
     */
    public function dismiss_pro_banner_forever() {
        check_admin_referer( 'ai_search_dismiss_pro_banner_forever' );

        if ( ! current_user_can( 'manage_options' ) ) {
            wp_die( esc_html__( 'You do not have permission to do this.', 'ai-search' ) );
        }

        update_option( 'ai_search_pro_banner_dismissed_forever', true );

        $referer = wp_get_referer();
        wp_safe_redirect( $referer ? $referer : admin_url( 'admin.php?page=ai-search' ) );
        exit;
    }

    /**
     * Snooze the Pro banner for 60 days. Not "dismiss forever" on purpose:
     * a user who declined two months ago might be in a very different
     * place now, and one gentle reappearance in that window is a fair
     * trade for the noise it would take to build a full "reminders" UI.
     *
     * @since 1.34.1
     */
    public function snooze_pro_banner() {
        check_admin_referer( 'ai_search_snooze_pro_banner' );

        if ( ! current_user_can( 'manage_options' ) ) {
            wp_die( esc_html__( 'You do not have permission to do this.', 'ai-search' ) );
        }

        update_option( 'ai_search_pro_banner_snoozed_until', time() + 60 * DAY_IN_SECONDS );

        $referer = wp_get_referer();
        wp_safe_redirect( $referer ? $referer : admin_url( 'admin.php?page=ai-search' ) );
        exit;
    }
}