use super::Tool;
use crate::map_data::MapData;
use crate::math::{Rect, Vec2};
use crate::transform::Transform;
use raylib::core::drawing::{RaylibDraw, RaylibDrawHandle};
use raylib::ffi::{Color, MouseButton};
use raylib::RaylibHandle;

pub struct DeletionTool {
    deletion_rect: Option<(Vec2<f32>, Vec2<f32>)>,
}

impl DeletionTool {
    pub fn new() -> Self {
        Self {
            deletion_rect: None,
        }
    }

    /// Delete all map-data that is contained inside the provided rectangular space.
    pub fn delete_rect(map_data: &mut MapData, rect: Rect<f32>) {
        map_data
            .rooms_mut()
            .retain(|&room| !rect.contains_rect(room));
        map_data
            .walls_mut()
            .retain(|&(pos1, pos2)| !rect.contains(pos1) || !rect.contains(pos2));
        map_data
            .icons_mut()
            .retain(|icon| !rect.contains(icon.position));
    }
}

impl Tool for DeletionTool {
    fn active_update(&mut self, map_data: &mut MapData, rl: &RaylibHandle, transform: &Transform) {
        let mouse_pos_m = transform.point_px_to_m(rl.get_mouse_position().into());
        if let Some((_, ref mut pos2)) = &mut self.deletion_rect {
            *pos2 = mouse_pos_m;
        }

        if rl.is_mouse_button_pressed(MouseButton::MOUSE_LEFT_BUTTON) {
            if let Some((pos1, pos2)) = self.deletion_rect {
                Self::delete_rect(map_data, Rect::bounding_rect(pos1, pos2));
                self.deletion_rect = None;
            } else {
                self.deletion_rect = Some((mouse_pos_m, mouse_pos_m))
            }
        }

        // Abort deletion.
        if rl.is_mouse_button_pressed(MouseButton::MOUSE_RIGHT_BUTTON) {
            self.deletion_rect = None;
        }
    }

    fn draw(&self, _map_data: &MapData, rld: &mut RaylibDrawHandle, transform: &Transform) {
        if let Some((pos1, pos2)) = self.deletion_rect {
            let rect_px = transform.rect_m_to_px(Rect::bounding_rect(pos1, pos2));
            rld.draw_rectangle_rec(
                rect_px,
                Color {
                    r: 200,
                    g: 150,
                    b: 150,
                    a: 50,
                },
            );
            rld.draw_rectangle_lines_ex(
                rect_px,
                4,
                Color {
                    r: 200,
                    g: 150,
                    b: 150,
                    a: 150,
                },
            );
        }
    }
}