use super::Mappable;
use crate::colours::DEFAULT_COLOURS;
use crate::math::{LineSegment, Rect, Vec2};
use crate::scaleable::Scaleable;
use crate::transform::Transform;
use raylib::drawing::{RaylibDraw, RaylibDrawHandle};
use std::ops::{Deref, DerefMut};

pub type WallData = LineSegment<f64>;

pub struct Wall {
    data: WallData,
    selected: bool,
    round_start: bool,
    round_end: bool,
}

impl Wall {
    pub fn from_data(data: WallData, round_start: bool, round_end: bool) -> Self {
        Self {
            data,
            selected: false,
            round_start,
            round_end,
        }
    }

    pub fn data(&self) -> &WallData {
        &self.data
    }
}

fn draw_round_corner(
    rld: &mut RaylibDrawHandle,
    pos_px: Vec2<f64>,
    transform: &Transform,
    selected: bool,
) {
    rld.draw_circle_v(
        pos_px,
        transform.length_m_to_px(0.05) as f32,
        if selected {
            DEFAULT_COLOURS.wall_selected
        } else {
            DEFAULT_COLOURS.wall_normal
        },
    );
}

impl Mappable for Wall {
    fn draw(&self, rld: &mut RaylibDrawHandle, transform: &Transform) {
        let start_px = transform.point_m_to_px(&self.data.start);
        let end_px = transform.point_m_to_px(&self.data.end);
        rld.draw_line_ex(
            start_px,
            end_px,
            transform.length_m_to_px(0.1) as f32,
            if self.selected() {
                DEFAULT_COLOURS.wall_selected
            } else {
                DEFAULT_COLOURS.wall_normal
            },
        );

        if self.round_start {
            draw_round_corner(rld, start_px, transform, self.selected());
        }
        if self.round_end {
            draw_round_corner(rld, end_px, transform, self.selected());
        }
    }

    fn set_selected(&mut self, selected: bool) {
        self.selected = selected;
    }

    fn selected(&self) -> bool {
        self.selected
    }

    fn bounding_rect(&self) -> Rect<f64> {
        Rect::bounding_rect(self.data.start, self.data.end)
    }
}

impl Scaleable for Wall {
    fn scale(&mut self, by: &Vec2<f64>) {
        if by.x <= 0. || by.y <= 0. {
            panic!("Cannot set dimensions with negative size");
        }

        self.data.start.x *= by.x;
        self.data.start.y *= by.y;
        self.data.end.x *= by.x;
        self.data.end.y *= by.y;
    }
}

impl Deref for Wall {
    type Target = WallData;

    fn deref(&self) -> &Self::Target {
        &self.data
    }
}

impl DerefMut for Wall {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.data
    }
}