<?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']);
        add_action('advaipbl_purge_old_logs_event', [$this->plugin, 'purge_old_logs']);
        add_action('advaipbl_send_summary_email', [$this->plugin, 'process_and_send_summary']);
        add_action('advaipbl_send_telemetry_data_event', [$this->plugin, 'send_telemetry_data']);
        add_action('advaipbl_update_geoip_db_event', [$this->plugin, 'execute_geoip_db_update']);
        add_action('advaipbl_clear_expired_blocks_event', [$this->plugin, 'limpiar_ips_expiradas']);
        add_action('advaipbl_cleanup_expired_cache_event', [$this->plugin, 'cleanup_expired_cache_entries']); 
        
        // --- 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_update_community_list_event', [$this->plugin->community_manager, 'update_list']);
        add_action('advaipbl_community_report_event_v2', [$this->plugin, 'execute_community_report']); // Activated hook
    }

    /**
     * Adds custom time intervals to WP Cron.
     */
    public function add_cron_intervals( $schedules ) {
        $schedules['six_hours'] = [
            'interval' => 21600,
            'display'  => __( 'Every 6 Hours', 'advanced-ip-blocker' )
        ];
        $schedules['weekly'] = [
            'interval' => 604800,
            'display'  => __( 'Once Weekly', '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. Summary Email (Daily at 8:00 AM)
        if (!wp_next_scheduled('advaipbl_send_summary_email') && ($this->plugin->options['notification_frequency'] ?? 'daily') !== 'instant' && ($this->plugin->options['notification_frequency'] ?? 'daily') !== 'disabled') {
            $frequency = $this->plugin->options['notification_frequency'] ?? 'daily';
            $timestamp = strtotime('08:00:00');
            if ($timestamp < time()) {
                $timestamp += DAY_IN_SECONDS;
            }
            wp_schedule_event($timestamp, $frequency, 'advaipbl_send_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 (Weekly + Random)
        if (!wp_next_scheduled('advaipbl_update_geoip_db_event')) {
            wp_schedule_event(time() + wp_rand(0, DAY_IN_SECONDS), 'weekly', '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');
        }

        // 13. Community List Update (Missing logic fixed)
        if ( ! wp_next_scheduled( 'advaipbl_update_community_list_event' ) ) {
            wp_schedule_event( time() + HOUR_IN_SECONDS, 'six_hours', 'advaipbl_update_community_list_event' );
        }
        
        // 14. Community Reports (Missing logic fixed)
        if ( ! wp_next_scheduled( 'advaipbl_community_report_event_v2' ) ) {
             wp_schedule_event( time() + (2 * HOUR_IN_SECONDS), 'six_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');
        }
		
		// 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, 'six_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;
        }

        $url = 'https://www.spamhaus.org/drop/asndrop.txt';
        $response = wp_remote_get($url, ['timeout' => 15]);

        if (is_wp_error($response)) {
            $this->plugin->log_event('Failed to download Spamhaus ASN DROP list: ' . $response->get_error_message(), 'error');
            return;
        }

        $body = wp_remote_retrieve_body($response);
        if (empty($body)) {
             $this->plugin->log_event('Spamhaus ASN DROP list is empty.', 'warning');
             return;
        }

        $lines = explode("\n", $body);
        $asns = [];

        foreach ($lines as $line) {
            $line = trim($line);
            if (empty($line) || strpos($line, ';') === 0) continue;

            $parts = explode(';', $line);
            $asn_part = trim($parts[0]);
            
            if (preg_match('/^AS(\d+)$/i', $asn_part, $matches)) {
                $asns[] = $matches[1];
            }
        }

        if (!empty($asns)) {
            update_option('advaipbl_spamhaus_drop_list', $asns, false);
            update_option('advaipbl_spamhaus_last_update', time()); // Fix: Update the last update timestamp
            $count = count($asns);
            $this->plugin->log_event("Spamhaus ASN DROP list updated. {$count} ASNs imported.", 'info');
        }
    }
}