use super::snake::Snake;
use super::variables::{Direction, Position, Size};

use crossterm::{
    style, Color, ColorType, Crossterm, ObjectStyle, Screen, StyledObject, TerminalCursor,
};

use rand::distributions::{IndependentSample, Range};

use std::collections::HashMap;
use std::fmt::Display;
use std::hash::Hash;
use std::ops::Index;

use rand;

pub struct Map {
    pub size: Size,
    pub foot_pos: Position,
}

impl Map {
    pub fn new(size: Size) -> Self {
        Map {
            size: size,
            foot_pos: Position::new(0, 0),
        }
    }

    // render the map on the screen.
    pub fn render_map(&mut self, screen: &Screen, free_positions: &mut HashMap<String, Position>) {
        let crossterm = Crossterm::from_screen(screen);
        let mut cursor = crossterm.cursor();
        let mut terminal = crossterm.terminal();

        for y in 0..self.size.height {
            for x in 0..self.size.height {
                if (y == 0 || y == self.size.height - 1) || (x == 0 || x == self.size.width - 1) {
                    cursor.goto(x as u16, y as u16);
                    terminal.write("█");
                } else {
                    free_positions.insert(format!("{},{}", x, y), Position::new(x, y));
                }
            }
        }
    }

    pub fn is_out_of_bounds(&self, new_pos: Position) -> bool {
        if (new_pos.x == 0 || new_pos.x == self.size.width)
            || (new_pos.y == 0 || new_pos.y == self.size.height)
        {
            return true;
        }

        return false;
    }

    pub fn is_food_eaten(&self, snake_head: Position) -> bool {
        snake_head.x == self.foot_pos.x && snake_head.y == self.foot_pos.y
    }

    pub fn spawn_food(&mut self, free_positions: &HashMap<String, Position>, screen: &Screen) {
        let index = Range::new(0, free_positions.len()).ind_sample(&mut rand::thread_rng());
        self.foot_pos = free_positions.values().skip(index).next().unwrap().clone();
        self.draw_food(screen);
    }

    fn draw_food(&self, screen: &Screen) {
        let cursor = TerminalCursor::from_output(&screen.stdout);
        cursor.goto(self.foot_pos.x as u16, self.foot_pos.y as u16);
        style("$").with(Color::Green).paint(&screen.stdout);
        screen.stdout.flush();
    }
}