use std::{collections::VecDeque, time::Instant};

use dropbear_engine::WGPU_BACKEND;
use dropbear_engine::input::{Controller, Keyboard, Mouse};
use dropbear_engine::scene::Scene;
use egui::{Color32, RichText, Ui};
use egui_plot::{Legend, Line, Plot, PlotPoints};

use dropbear_engine::gilrs;
use winit::dpi::PhysicalPosition;
use winit::event::MouseButton;
use winit::event_loop::ActiveEventLoop;
use winit::keyboard::KeyCode;

pub const EGUI_VERSION: &str = "0.33";
pub const WGPU_VERSION: &str = "27";

/// Statistics for checking performance of the editor.
pub struct NerdStats {
    pub show_window: bool,

    fps_history: VecDeque<[f64; 2]>,
    frame_times: VecDeque<f64>,
    last_fps_update: Instant,
    current_fps: f32,

    frame_time_history: VecDeque<[f64; 2]>,

    memory_history: VecDeque<[f64; 2]>,

    start_time: Instant,
    total_frames: u64,

    min_fps: f32,
    max_fps: f32,
    avg_fps: f32,
    entity_count: u32,
}

impl Default for NerdStats {
    fn default() -> Self {
        Self {
            fps_history: VecDeque::with_capacity(300),
            frame_times: VecDeque::with_capacity(60),
            last_fps_update: Instant::now(),
            current_fps: 0.0,
            frame_time_history: VecDeque::with_capacity(300),
            memory_history: VecDeque::with_capacity(300),
            start_time: Instant::now(),
            total_frames: 0,
            min_fps: f32::MAX,
            max_fps: 0.0,
            avg_fps: 0.0,
            show_window: false,
            entity_count: 0,
        }
    }
}

impl NerdStats {
    /// Updates all information in [`NerdStats`] with the deltatime provided by the scene
    pub fn record_stats(&mut self, dt: f32, entity_count: u32) {
        self.total_frames += 1;
        let elapsed = self.start_time.elapsed().as_secs_f64();

        let frame_time_ms = (dt * 1000.0) as f64;
        self.frame_times.push_back(dt as f64);
        if self.frame_times.len() > 60 {
            self.frame_times.pop_front();
        }

        if self.last_fps_update.elapsed().as_secs_f32() >= 0.1 && dt > 0.0 {
            self.current_fps = 1.0 / dt;

            self.min_fps = self.min_fps.min(self.current_fps);
            self.max_fps = self.max_fps.max(self.current_fps);

            if !self.frame_times.is_empty() {
                let avg_frame_time: f64 =
                    self.frame_times.iter().sum::<f64>() / self.frame_times.len() as f64;
                self.avg_fps = (1.0 / avg_frame_time) as f32;
            }

            self.fps_history
                .push_back([elapsed, self.current_fps as f64]);
            if self.fps_history.len() > 300 {
                self.fps_history.pop_front();
            }

            self.frame_time_history.push_back([elapsed, frame_time_ms]);
            if self.frame_time_history.len() > 300 {
                self.frame_time_history.pop_front();
            }

            self.last_fps_update = Instant::now();
        }

        if self.total_frames.is_multiple_of(30) {
            let memory_mb = if let Some(usage) = memory_stats::memory_stats() {
                (usage.physical_mem / 1024 / 1024) as f64
            } else {
                0.0
            };

            self.memory_history.push_back([elapsed, memory_mb]);
            if self.memory_history.len() > 300 {
                self.memory_history.pop_front();
            }
        }
        self.entity_count = entity_count;
    }

    /// Resets statistics to their defaults
    pub fn reset_stats(&mut self) {
        self.min_fps = self.current_fps;
        self.max_fps = self.current_fps;
        self.fps_history.clear();
        self.frame_time_history.clear();
        self.memory_history.clear();
        self.start_time = Instant::now();
        self.total_frames = 0;
    }

    /// Shows the egui content with a passable [egui::Ui]
    pub fn content(&mut self, ui: &mut Ui) {
        ui.vertical(|ui| {
            ui.horizontal(|ui| {
                ui.heading("Performance Monitor");
                ui.with_layout(egui::Layout::right_to_left(egui::Align::Center), |ui| {
                    if ui.button("Reset Stats").clicked() {
                        self.reset_stats();
                    }
                });
            });

            ui.separator();

            ui.horizontal(|ui| {
                ui.vertical(|ui| {
                    ui.label(RichText::new("Current FPS").strong());
                    let fps_color = if self.current_fps >= 60.0 {
                        Color32::GREEN
                    } else if self.current_fps >= 30.0 {
                        Color32::YELLOW
                    } else {
                        Color32::RED
                    };
                    ui.label(
                        RichText::new(format!("{:.1}", self.current_fps))
                            .size(24.0)
                            .color(fps_color),
                    );
                });

                ui.separator();

                ui.vertical(|ui| {
                    ui.label(RichText::new("Frame Time").strong());
                    ui.label(
                        RichText::new(format!("{:.2} ms", 1000.0 / self.current_fps.max(1.0)))
                            .size(24.0),
                    );
                });

                ui.separator();

                ui.vertical(|ui| {
                    ui.label(RichText::new("Avg FPS").strong());
                    ui.label(RichText::new(format!("{:.1}", self.avg_fps)).size(24.0));
                });

                ui.separator();

                ui.vertical(|ui| {
                    ui.label(RichText::new("Entity Count").strong());
                    ui.label(RichText::new(format!("{} entities", self.entity_count)).size(24.0));
                });
            });

            ui.add_space(5.0);

            ui.horizontal(|ui| {
                ui.label(format!("Min: {:.1} fps", self.min_fps));
                ui.separator();
                ui.label(format!("Max: {:.1} fps", self.max_fps));
                ui.separator();
                ui.label(format!("Total Frames: {}", self.total_frames));
                ui.separator();
                ui.label(format!(
                    "Uptime: {:.1}s",
                    self.start_time.elapsed().as_secs_f32()
                ));
            });

            ui.separator();

            ui.label(RichText::new("FPS Over Time").strong());
            Plot::new("fps_plot")
                .height(150.0)
                .show_axes([false, true])
                .show_grid([false, true])
                .legend(Legend::default())
                .show(ui, |plot_ui| {
                    if !self.fps_history.is_empty() {
                        let points: Vec<[f64; 2]> = self.fps_history.iter().cloned().collect();
                        plot_ui.line(
                            Line::new("fps", PlotPoints::from(points))
                                .color(Color32::from_rgb(100, 200, 100))
                                .name("FPS"),
                        );

                        let first = self.fps_history.front().copied();
                        let last = self.fps_history.back().copied();

                        if let Some(first) = first
                            && let Some(last) = last
                        {
                            plot_ui.line(
                                Line::new(
                                    "60 fps target",
                                    PlotPoints::from(vec![
                                        [*first.get(0).unwrap(), 60.0],
                                        [*last.get(0).unwrap(), 60.0],
                                    ]),
                                )
                                .color(Color32::from_rgba_unmultiplied(255, 255, 0, 100))
                                .style(egui_plot::LineStyle::Dashed { length: 5.0 })
                                .name("60 FPS Target"),
                            );
                        }
                    }
                });

            ui.add_space(5.0);

            ui.label(RichText::new("Frame Time").strong());
            Plot::new("frame_time_plot")
                .height(150.0)
                .show_axes([false, true])
                .show_grid([false, true])
                .legend(Legend::default())
                .show(ui, |plot_ui| {
                    if !self.frame_time_history.is_empty() {
                        let points: Vec<[f64; 2]> =
                            self.frame_time_history.iter().cloned().collect();
                        plot_ui.line(
                            Line::new("frametime", PlotPoints::from(points))
                                .color(Color32::from_rgb(100, 150, 255))
                                .name("Frame Time (ms)"),
                        );

                        let first = self.frame_time_history.front().copied();
                        let last = self.frame_time_history.back().copied();

                        if let Some(first) = first
                            && let Some(last) = last
                        {
                            plot_ui.line(
                                Line::new(
                                    "frametime_base",
                                    PlotPoints::from(vec![
                                        [*first.get(0).unwrap(), 16.67],
                                        [*last.get(0).unwrap(), 16.67],
                                    ]),
                                )
                                .color(Color32::from_rgba_unmultiplied(255, 255, 0, 100))
                                .style(egui_plot::LineStyle::Dashed { length: 5.0 })
                                .name("16.67ms (60 FPS)"),
                            );
                        }
                    }
                });

            ui.add_space(5.0);

            ui.label(RichText::new("Memory Usage").strong());
            Plot::new("memory_plot")
                .height(120.0)
                .show_axes([false, true])
                .show_grid([false, true])
                .legend(Legend::default())
                .show(ui, |plot_ui| {
                    if !self.memory_history.is_empty() {
                        let points: Vec<[f64; 2]> = self.memory_history.iter().cloned().collect();
                        plot_ui.line(
                            Line::new("memory", PlotPoints::from(points))
                                .color(Color32::from_rgb(255, 150, 100))
                                .name("Memory (MB)"),
                        );
                    }
                });

            ui.separator();
            ui.collapsing("System Information", |ui| {
                ui.horizontal(|ui| {
                    ui.label("egui version:");
                    ui.label(EGUI_VERSION);
                });
                ui.horizontal(|ui| {
                    ui.label("Backend:");
                    ui.label(WGPU_BACKEND.get().unwrap());
                });
                ui.horizontal(|ui| {
                    ui.label("OS:");
                    ui.label(std::env::consts::OS);
                });
                ui.horizontal(|ui| {
                    ui.label("Architecture:");
                    ui.label(std::env::consts::ARCH);
                });
            });
        });
    }

    /// Shows the egui window as a CentralPanel, typically used for another window.
    pub fn show_window(&mut self, ui: &mut Ui) {
        egui::CentralPanel::default().show_inside(ui, |ui| {
            self.content(ui);
        });
    }
}

impl Scene for NerdStats {
    fn load(
        &mut self,
        _graphics: std::sync::Arc<dropbear_engine::graphics::SharedGraphicsContext>,
        _ui: &mut Ui,
    ) {
    }
    fn physics_update(
        &mut self,
        _dt: f32,
        _graphics: std::sync::Arc<dropbear_engine::graphics::SharedGraphicsContext>,
        _ui: &mut Ui,
    ) {
    }
    fn update(
        &mut self,
        dt: f32,
        _graphics: std::sync::Arc<dropbear_engine::graphics::SharedGraphicsContext>,
        _ui: &mut Ui,
    ) {
        self.record_stats(dt, self.entity_count);
    }
    
    fn render<'a>(
        &mut self,
        _graphics: std::sync::Arc<dropbear_engine::graphics::SharedGraphicsContext>,
        ui: &mut Ui,
    ) {
        self.show_window(ui);
    }
    fn exit(&mut self, _event_loop: &ActiveEventLoop) {}
}

impl Keyboard for NerdStats {
    fn key_down(&mut self, _key: KeyCode, _event_loop: &ActiveEventLoop) {}
    fn key_up(&mut self, _key: KeyCode, _event_loop: &ActiveEventLoop) {}
}

impl Mouse for NerdStats {
    fn mouse_move(&mut self, _position: PhysicalPosition<f64>, _delta: Option<(f64, f64)>) {}
    fn mouse_down(&mut self, _button: MouseButton) {}
    fn mouse_up(&mut self, _button: MouseButton) {}
}

impl Controller for NerdStats {
    fn button_down(&mut self, _button: gilrs::Button, _id: gilrs::GamepadId) {}
    fn button_up(&mut self, _button: gilrs::Button, _id: gilrs::GamepadId) {}
    fn left_stick_changed(&mut self, _x: f32, _y: f32, _id: gilrs::GamepadId) {}
    fn right_stick_changed(&mut self, _x: f32, _y: f32, _id: gilrs::GamepadId) {}
    fn on_connect(&mut self, _id: gilrs::GamepadId) {}
    fn on_disconnect(&mut self, _id: gilrs::GamepadId) {}
}