<?php

if ( ! defined( 'ABSPATH' ) ) exit;

/**
 * Class ADVAIPBL_Cron_Manager
 * 
 * Handles all WP-Cron scheduling and execution logic.
 */
class ADVAIPBL_Cron_Manager {

    private $plugin;

    public function __construct( $plugin ) {
        $this->plugin = $plugin;
        
        // Register interval schedules
        add_filter('cron_schedules', [$this, 'add_cron_intervals']);
        
        // --- Main Logic Tasks ---
        add_action('advaipbl_threat_score_decay_event', [$this, 'execute_threat_score_decay']);
        add_action('advaipbl_signature_analysis_event', [$this, 'execute_signature_analysis']);
        add_action('advaipbl_scheduled_scan_event', [$this, 'execute_malware_scan']);
        add_action('advaipbl_update_spamhaus_list_event', [$this, 'update_spamhaus_list']);
        
        // Note: The following actions are registered in ADVAIPBL_Main::add_hooks() to avoid duplication:
        // - advaipbl_purge_old_logs_event
        // - advaipbl_send_telemetry_data_event
        // - advaipbl_update_geoip_db_event
        // - advaipbl_clear_expired_blocks_event
        // - advaipbl_cleanup_expired_cache_event
        
        // --- Manager Specific Tasks ---
        add_action('advaipbl_daily_fim_scan', [$this->plugin->file_verifier, 'scan_files']);
        add_action('advaipbl_cloudflare_cleanup_event', [$this->plugin->cloudflare_manager, 'clear_all_aib_rules']);
        add_action('advaipbl_cloudflare_sync_event', [$this->plugin->cloudflare_manager, 'sync_blocked_ips']);
        add_action('advaipbl_update_community_list_event', [$this->plugin->community_manager, 'update_list']);
        add_action('advaipbl_community_report_event_v2', [$this->plugin, 'execute_community_report']); // Activated hook
        add_action('advaipbl_update_ai_bot_lists_event', [$this->plugin->bot_verifier, 'fetch_and_cache_ai_lists']);
    }

    /**
     * Adds custom time intervals to WP Cron.
     */
    public function add_cron_intervals( $schedules ) {
        $schedules['advaipbl_6_hours'] = [
            'interval' => 21600,
            'display'  => __( 'Every 6 Hours (AIB)', 'advanced-ip-blocker' )
        ];
        $schedules['advaipbl_3_days'] = [
            'interval' => 259200, // 3 days * 24 * 60 * 60
            'display'  => __( 'Every 3 Days (AIB)', 'advanced-ip-blocker' )
        ];
        $schedules['advaipbl_weekly'] = [
            'interval' => 604800,
            'display'  => __( 'Once Weekly (AIB)', 'advanced-ip-blocker' )
        ];
        return $schedules;
    }



    /**
     * Schedules all necessary cron jobs based on settings.
     * Called usually during admin init or settings save.
     */
    public function schedule_jobs() {
        // 1. Decaimiento de Puntuación (Threat Score Decay)
        if ( ! wp_next_scheduled( 'advaipbl_threat_score_decay_event' ) ) {
            wp_schedule_event( time(), 'hourly', 'advaipbl_threat_score_decay_event' );
        }

        // 2. Análisis de Firmas (Signature Analysis)
        if ( ! empty( $this->plugin->options['enable_signature_analysis'] ) ) {
            if ( ! wp_next_scheduled( 'advaipbl_signature_analysis_event' ) ) {
                wp_schedule_event( time(), 'hourly', 'advaipbl_signature_analysis_event' );
            }
        } else {
            wp_clear_scheduled_hook( 'advaipbl_signature_analysis_event' );
        }

        // 3. Purga de Logs (Daily)
        if (!wp_next_scheduled('advaipbl_purge_old_logs_event')) {
            wp_schedule_event(time(), 'daily', 'advaipbl_purge_old_logs_event');
        }

        // 4. Standard Summary Email (Daily at 8:00 AM)
        $freq_main = $this->plugin->options['notification_frequency'] ?? 'daily';
        
        if ($freq_main !== 'instant' && $freq_main !== 'disabled') {
            if (!wp_next_scheduled('advaipbl_send_summary_email')) {
                $timestamp = strtotime('08:00:00');
                if ($timestamp < time()) {
                    $timestamp += DAY_IN_SECONDS;
                }
                wp_schedule_event($timestamp, $freq_main, 'advaipbl_send_summary_email');
            }
        } else {
            wp_clear_scheduled_hook('advaipbl_send_summary_email');
        }

        // 4b. Signature Summary Email (Daily at 8:05 AM)
        $freq_sig = $this->plugin->options['signature_notification_frequency'] ?? 'instant';

        if ($freq_sig !== 'instant' && $freq_sig !== 'disabled') {
            if (!wp_next_scheduled('advaipbl_send_signature_summary_email')) {
                // Schedule slightly after standard to avoid burst
                $timestamp = strtotime('08:05:00');
                if ($timestamp < time()) {
                    $timestamp += DAY_IN_SECONDS;
                }
                // Determine valid schedule key
                $sig_schedule = ($freq_sig === 'daily' || $freq_sig === 'weekly') ? $freq_sig : 'daily';
                
                wp_schedule_event($timestamp, $sig_schedule, 'advaipbl_send_signature_summary_email');
            }
        } else {
            wp_clear_scheduled_hook('advaipbl_send_signature_summary_email');
        }

        // 5. Telemetry (Weekly + Jitter)
        if (!empty($this->plugin->options['allow_telemetry']) && $this->plugin->options['allow_telemetry'] === '1') {
             if (!wp_next_scheduled('advaipbl_send_telemetry_data_event')) {
                 $jitter = wp_rand(0, 12 * HOUR_IN_SECONDS); 
                 wp_schedule_event(time() + $jitter, 'weekly', 'advaipbl_send_telemetry_data_event');
             }
        } else {
             wp_clear_scheduled_hook('advaipbl_send_telemetry_data_event');
        }

        // 6. Update GeoIP DB (Every 3 Days + Random)
        // Check for legacy or mismatched schedules
        $schedule = wp_get_schedule('advaipbl_update_geoip_db_event');
        if ($schedule === 'weekly' || $schedule === 'three_days' || $schedule === 'advaipbl_weekly') {
            wp_clear_scheduled_hook('advaipbl_update_geoip_db_event');
        }
        
        if (!wp_next_scheduled('advaipbl_update_geoip_db_event')) {
            // Se programa la primera descarga más rápido (1 a 5 minutos) en lugar de 12 horas.
            wp_schedule_event(time() + wp_rand(60, 300), 'advaipbl_3_days', 'advaipbl_update_geoip_db_event');
        }

        // 7. Clear Expired Blocks (Hourly)
        if (!wp_next_scheduled('advaipbl_clear_expired_blocks_event')) {
            wp_schedule_event(time(), 'hourly', 'advaipbl_clear_expired_blocks_event');
        }

        // 8. Cleanup Cache (Daily)
        if (!wp_next_scheduled('advaipbl_cleanup_expired_cache_event')) {
            wp_schedule_event(time(), 'daily', 'advaipbl_cleanup_expired_cache_event');
        }

        // 9. FIM Daily Scan (Daily + Random)
        if (!empty($this->plugin->options['enable_fim']) && !wp_next_scheduled('advaipbl_daily_fim_scan')) {
             wp_schedule_event(time() + wp_rand(0, 4 * HOUR_IN_SECONDS), 'daily', 'advaipbl_daily_fim_scan');
        } elseif (empty($this->plugin->options['enable_fim'])) {
             wp_clear_scheduled_hook('advaipbl_daily_fim_scan');
        }

        // 10. Malware Scan Scheduled
        $enable_scan = $this->plugin->options['enable_scheduled_scans'] ?? '0'; // Fixed key name from View?
        // Wait, View said 'enable_scheduled_scans' (plural) at line 2640.
        // My previous file write used 'enable_scheduled_scan' (singular).
        // I must ensure key correctness. Main line 2640: $this->options['enable_scheduled_scans']
        $scan_freq   = $this->plugin->options['scheduled_scan_frequency'] ?? 'daily'; // Main 2641 has 'scan_frequency'?
        // Main 2641: $this->options['scan_frequency']
        
        $enable_scan_val = $this->plugin->options['enable_scheduled_scans'] ?? '0';
        $scan_freq_val   = $this->plugin->options['scan_frequency'] ?? 'weekly'; // Default weekly in Main 2641

        if ( $enable_scan_val === '1' ) {
            if ( ! wp_next_scheduled( 'advaipbl_scheduled_scan_event' ) ) {
                 $schedule_time = strtotime('tomorrow 00:00:00') + wp_rand(0, 6 * HOUR_IN_SECONDS);
                 wp_schedule_event($schedule_time, $scan_freq_val, 'advaipbl_scheduled_scan_event');
            }
        } else {
            wp_clear_scheduled_hook('advaipbl_scheduled_scan_event');
        }
        
        // Handle frequency change reschedules
        $next_scan = wp_next_scheduled('advaipbl_scheduled_scan_event');
        if ($enable_scan_val === '1' && $next_scan) {
             $schedule = wp_get_schedule('advaipbl_scheduled_scan_event');
             if ($schedule !== $scan_freq_val) {
                 wp_clear_scheduled_hook('advaipbl_scheduled_scan_event');
                 $schedule_time = strtotime('tomorrow 00:00:00') + wp_rand(0, 6 * HOUR_IN_SECONDS);
                 wp_schedule_event($schedule_time, $scan_freq_val, 'advaipbl_scheduled_scan_event');
             }
        }
		
		// 11. Community Reports
        if ( ! wp_next_scheduled( 'advaipbl_community_report_event_v2' ) ) {
            wp_clear_scheduled_hook('advaipbl_community_report_event');
            wp_schedule_event( time() + HOUR_IN_SECONDS, 'six_hours', 'advaipbl_community_report_event_v2' );
        }
		
		// 12. Spamhaus Update
        if (!empty($this->plugin->options['enable_spamhaus_asn']) && '1' === $this->plugin->options['enable_spamhaus_asn']) {
            if (!wp_next_scheduled('advaipbl_update_spamhaus_list_event')) {
                wp_schedule_event(time(), 'daily', 'advaipbl_update_spamhaus_list_event');
            }
        } else {
            wp_clear_scheduled_hook('advaipbl_update_spamhaus_list_event');
        }

        // 12.5 AI Bot JSON Verification
        if (!empty($this->plugin->options['enable_ai_bot_verification']) && '1' === $this->plugin->options['enable_ai_bot_verification']) {
            if (!wp_next_scheduled('advaipbl_update_ai_bot_lists_event')) {
                wp_schedule_event(time(), 'daily', 'advaipbl_update_ai_bot_lists_event');
            }
        } else {
            wp_clear_scheduled_hook('advaipbl_update_ai_bot_lists_event');
        }

        // 13. Community List Update (Missing logic fixed)
        $schedule = wp_get_schedule('advaipbl_update_community_list_event');
        if ($schedule === 'six_hours') {
            wp_clear_scheduled_hook('advaipbl_update_community_list_event');
        }

        if ( ! wp_next_scheduled( 'advaipbl_update_community_list_event' ) ) {
            wp_schedule_event( time() + HOUR_IN_SECONDS, 'advaipbl_6_hours', 'advaipbl_update_community_list_event' );
        }
        
        // 14. Community Reports (Missing logic fixed)
        $schedule = wp_get_schedule('advaipbl_community_report_event_v2');
        if ($schedule === 'six_hours') {
            wp_clear_scheduled_hook('advaipbl_community_report_event_v2');
        }

        if ( ! wp_next_scheduled( 'advaipbl_community_report_event_v2' ) ) {
             wp_schedule_event( time() + (2 * HOUR_IN_SECONDS), 'advaipbl_6_hours', 'advaipbl_community_report_event_v2' );
        }

        // 15. Cloudflare Cleanup (Daily if enabled)
        if ( !empty($this->plugin->options['enable_cloudflare']) && '1' === $this->plugin->options['enable_cloudflare'] ) {
            if ( ! wp_next_scheduled( 'advaipbl_cloudflare_cleanup_event' ) ) {
                wp_schedule_event( time() + wp_rand(0, DAY_IN_SECONDS), 'daily', 'advaipbl_cloudflare_cleanup_event' );
            }
        } else {
             wp_clear_scheduled_hook('advaipbl_cloudflare_cleanup_event');
        }

        // 17. Cloudflare Sync (Hourly if enabled) - CRITICAL FIX
        if ( !empty($this->plugin->options['enable_cloudflare']) && '1' === $this->plugin->options['enable_cloudflare'] ) {
            if ( ! wp_next_scheduled( 'advaipbl_cloudflare_sync_event' ) ) {
                wp_schedule_event( time(), 'hourly', 'advaipbl_cloudflare_sync_event' );
            }
        } else {
             wp_clear_scheduled_hook('advaipbl_cloudflare_sync_event');
        }
		
		// 16. Community List Sync (Deduplication check)
        $crons = _get_cron_array();
        $count = 0;
        if ( is_array($crons) ) {
            foreach ( $crons as $timestamp => $cronhooks ) {
                if ( isset( $cronhooks['advaipbl_update_community_list_event'] ) ) {
                    $count++;
                }
            }
        }
        
        if ( $count > 1 ) {
            wp_clear_scheduled_hook( 'advaipbl_update_community_list_event' );
            wp_schedule_event( time() + HOUR_IN_SECONDS, 'advaipbl_6_hours', 'advaipbl_update_community_list_event' );
        }

        // Cleanup
        delete_transient('advaipbl_crons_scheduled');
        set_transient('advaipbl_crons_scheduled_v866', true, DAY_IN_SECONDS);
    }

    /**
     * Executes Threat Score Decay.
     */
    public function execute_threat_score_decay() {
        if (!empty($this->plugin->threat_score_manager)) {
            $decay_points = (int) ($this->plugin->options['threat_score_decay_points'] ?? 10);
            $inactive_hours = (int) ($this->plugin->options['threat_score_decay_period'] ?? 1);
            $inactive_seconds = $inactive_hours * HOUR_IN_SECONDS;
            
            $this->plugin->threat_score_manager->decay_scores($decay_points, $inactive_seconds);
        }
    }

    /**
     * Executes Signature Analysis.
     */
    public function execute_signature_analysis() {
        if (empty($this->plugin->options['enable_signature_analysis'])) {
            return;
        }

        $ip_threshold = (int) ($this->plugin->options['signature_ip_threshold'] ?? 5);
        $analysis_window_hours = (int) ($this->plugin->options['signature_analysis_window'] ?? 1);
        $rule_ttl_hours = (int) ($this->plugin->options['signature_rule_ttl'] ?? 24);

        if (!empty($this->plugin->fingerprint_manager)) {
            $this->plugin->fingerprint_manager->analyze_and_flag_signatures(
                $ip_threshold,
                $analysis_window_hours * HOUR_IN_SECONDS,
                $rule_ttl_hours * HOUR_IN_SECONDS
            );
        }
    }

    /**
     * Executes Scheduled Malware Scan.
     */
    public function execute_malware_scan() {
        if (!empty($this->plugin->site_scanner)) {
            // Fix: Call the method that handles email logic
            // Pass null as email to use the configured default in run_full_scan_and_email
            $this->plugin->site_scanner->run_full_scan_and_email(null, false);
        }
    }

    /**
     * Updates Spamhaus List.
     */
    public function update_spamhaus_list() {
         if (empty($this->plugin->options['enable_spamhaus_asn']) || '1' !== $this->plugin->options['enable_spamhaus_asn']) {
            return;
        }

        // Delegate to the main class method which handles the logic correctly (JSON parsing, correct option name)
        // This ensures the cron job behaves exactly like the manual "Refresh Now" button.
        $this->plugin->update_spamhaus_list();
    }
}