<?php
/**
 * Dashboard module – aggregates security data for the main admin view.
 *
 * @package AipatchSecurityScanner
 */

if ( ! defined( 'ABSPATH' ) ) {
    exit;
}

/**
 * Class AIPSC_Dashboard
 */
class AIPSC_Dashboard {

    /**
     * @var AIPSC_Scanner|null
     */
    private $scanner;

    /**
     * @var AIPSC_Vulnerabilities|null
     */
    private $vulnerabilities;

    /**
     * @var AIPSC_Logger|null
     */
    private $logger;

    /**
     * Constructor.
     *
     * @param AIPSC_Scanner|null         $scanner         Scanner module.
     * @param AIPSC_Vulnerabilities|null $vulnerabilities Vulnerabilities module.
     * @param AIPSC_Logger|null          $logger          Logger instance.
     */
    public function __construct( $scanner, $vulnerabilities, $logger = null ) {
        $this->scanner         = $scanner;
        $this->vulnerabilities = $vulnerabilities;
        $this->logger          = $logger;
    }

    /**
     * Get complete dashboard data.
     *
     * @return array
     */
    public function get_dashboard_data() {
        $scan_results = $this->scanner ? $this->scanner->get_last_results() : false;
        $summary      = $this->scanner ? $this->scanner->get_summary() : array();
        $dismissed    = AIPSC_Utils::get_option( 'dismissed', array() );

        $score  = $scan_results ? $scan_results['score'] : 0;
        $issues = $scan_results ? $scan_results['issues'] : array();

        // Separate active and dismissed issues.
        $active_issues    = array();
        $dismissed_issues = array();

        foreach ( $issues as $issue ) {
            if ( isset( $dismissed[ $issue['id'] ] ) ) {
                $dismissed_issues[] = $issue;
            } else {
                $active_issues[] = $issue;
            }
        }

        // Sort by severity weight (critical first).
        usort( $active_issues, function ( $a, $b ) {
            return AIPSC_Utils::severity_weight( $b['severity'] ) - AIPSC_Utils::severity_weight( $a['severity'] );
        } );

        // Prioritized recommendations from active issues.
        $recommendations = array_map( function ( $issue ) {
            return array(
                'id'             => $issue['id'],
                'title'          => $issue['title'],
                'severity'       => $issue['severity'],
                'recommendation' => $issue['recommendation'],
                'dismissible'    => $issue['dismissible'],
            );
        }, $active_issues );

        return array(
            'score'            => $score,
            'score_label'      => $this->get_score_label( $score ),
            'score_class'      => $this->get_score_class( $score ),
            'summary'          => $summary,
            'issues'           => $active_issues,
            'dismissed_issues' => $dismissed_issues,
            'recommendations'  => $recommendations,
            'last_scan'        => AIPSC_Utils::get_option( 'last_scan', 0 ),
            'next_scan'        => AIPSC_Cron::get_next_scan(),
            'has_scan'         => ! empty( $scan_results ),
        );
    }

    /**
     * Get human-readable score label.
     *
     * @param int $score Security score.
     * @return string
     */
    private function get_score_label( $score ) {
        if ( $score >= 90 ) {
            return __( 'Excellent', 'aipatch-security-scanner' );
        }
        if ( $score >= 70 ) {
            return __( 'Good', 'aipatch-security-scanner' );
        }
        if ( $score >= 50 ) {
            return __( 'Needs Attention', 'aipatch-security-scanner' );
        }
        if ( $score >= 30 ) {
            return __( 'At Risk', 'aipatch-security-scanner' );
        }
        return __( 'Critical', 'aipatch-security-scanner' );
    }

    /**
     * Get CSS class for score display.
     *
     * @param int $score Security score.
     * @return string
     */
    private function get_score_class( $score ) {
        if ( $score >= 90 ) {
            return 'aipatch-score-excellent';
        }
        if ( $score >= 70 ) {
            return 'aipatch-score-good';
        }
        if ( $score >= 50 ) {
            return 'aipatch-score-warning';
        }
        return 'aipatch-score-danger';
    }

    /**
     * Dismiss a risk issue.
     *
     * @param string $issue_id Issue identifier.
     * @return bool
     */
    public function dismiss_issue( $issue_id ) {
        $dismissed = AIPSC_Utils::get_option( 'dismissed', array() );
        $dismissed[ sanitize_key( $issue_id ) ] = array(
            'dismissed_at' => time(),
            'dismissed_by' => get_current_user_id(),
        );
        $result = AIPSC_Utils::update_option( 'dismissed', $dismissed );

        if ( $result && $this->logger ) {
            $this->logger->info(
                'issue_dismissed',
                sprintf(
                    /* translators: %s: Issue ID. */
                    __( 'Issue "%s" dismissed.', 'aipatch-security-scanner' ),
                    sanitize_key( $issue_id )
                )
            );
        }

        return $result;
    }

    /**
     * Restore a dismissed issue.
     *
     * @param string $issue_id Issue identifier.
     * @return bool
     */
    public function restore_issue( $issue_id ) {
        $dismissed = AIPSC_Utils::get_option( 'dismissed', array() );
        unset( $dismissed[ sanitize_key( $issue_id ) ] );
        $result = AIPSC_Utils::update_option( 'dismissed', $dismissed );

        if ( $result && $this->logger ) {
            $this->logger->info(
                'issue_restored',
                sprintf(
                    /* translators: %s: Issue ID. */
                    __( 'Issue "%s" restored.', 'aipatch-security-scanner' ),
                    sanitize_key( $issue_id )
                )
            );
        }

        return $result;
    }
}