use ggez::filesystem::File;
use ggez::graphics::Rect;
use std::collections::HashMap;

use crate::constants;
use crate::property::Property;
use crate::xmlelements::XMLElements;

pub struct Tileset {
    tiles: HashMap<usize, Rect>,
    properties: Vec<Property>,
}

impl Tileset {
    pub fn new(file: File) -> Tileset {
        let elements = XMLElements::new(file);

        let height = elements
            .get_element_attribute("image", "height")
            .unwrap()
            .parse::<usize>()
            .unwrap();
        let columns = elements
            .get_element_attribute("tileset", "columns")
            .unwrap()
            .parse::<usize>()
            .unwrap();

        let rows = height / (constants::TILE_HEIGHT as usize);

        let mut tiles = HashMap::new();
        tiles.insert(0, Rect::zero());

        let w = 1.0 / columns as f32;
        let h = 1.0 / rows as f32;
        let mut key = 1;
        for r in 0..rows {
            for c in 0..columns {
                let x = c as f32 / columns as f32;
                let y = r as f32 / rows as f32;
                tiles.insert(key, Rect::new(x, y, w, h));
                key += 1;
            }
        }

        let mut properties = Vec::new();

        for tile_element in elements.get_elements("tile") {
            let tile_id = XMLElements::get_attribute(&tile_element, "id")
                .unwrap()
                .parse::<usize>()
                .unwrap()
                + 1;

            let property_elements = elements.get_children(&tile_element, "property");

            properties.push(Property::new(tile_id, property_elements));
        }

        Tileset { tiles, properties }
    }

    pub fn get(&self, id: usize) -> Rect {
        *self.tiles.get(&id).unwrap()
    }

    pub fn get_animation(&self, tile_id: usize) -> Vec<(usize, Rect)> {
        if let Some(property) = self.properties.iter().find(|p| p.tile_id == tile_id) {
            self.properties
                .clone()
                .into_iter()
                .filter(|p| p.entity == property.entity && p.entity.is_some())
                .map(|p| (p.delay.unwrap(), self.get(p.tile_id)))
                .collect()
        } else {
            Vec::new()
        }
    }

    pub fn get_tile_by_entity_keyframe(&self, entity: &str, keyframe: usize) -> Rect {
        *self
            .tiles
            .get(
                &self
                    .properties
                    .iter()
                    .find(|p| p.entity == Some(entity.to_string()) && Some(keyframe) == p.keyframe)
                    .unwrap()
                    .tile_id,
            )
            .unwrap()
    }
}