extern crate termion;
extern crate serde_json;

use std::net::TcpStream;
use std::collections::BTreeMap;
use self::termion::async_stdin;
use std::io::{BufRead, BufReader};
use std::io::{stdout, Read, Write};
use self::termion::raw::IntoRawMode;

use math::distance;
use mass::{Mass, MassType};
use modules::navigation::Navigation;

pub fn client_navigation(name : String, mut stream : TcpStream, mut buff_r : BufReader<TcpStream>){
    let stdout = stdout();
    let mut stdout = stdout.lock().into_raw_mode().unwrap();
    let mut stdin = async_stdin().bytes();

    loop {
        let mut recv = String::new();
        buff_r.read_line(&mut recv).unwrap();
        let mut within_range : BTreeMap<String, Mass> = serde_json::from_str(&recv).unwrap();

        write!(stdout, "{}{}Targets:",
               termion::clear::All,
               termion::cursor::Goto(1,1)).unwrap();

        let ship = within_range.remove(&name).unwrap();

        if let MassType::Ship{ref navigation, ..} = ship.mass_type {
            let navigation = navigation.clone().unwrap();
            for (i, (mass_name, mass)) in within_range.iter().enumerate() {
                let target_data = get_target_status(&navigation, &mass_name);
                write!(stdout, "{}{}) {} ({:.2}, {:.2}, {:.2}) Distance : {:.2} {}",
                       termion::cursor::Goto(1, 2 + i as u16),
                       i,
                       mass_name,
                       mass.position.0,
                       mass.position.1,
                       mass.position.2,
                       distance(mass.position, ship.position),
                       target_data
                       ).unwrap();
            }

            if let Some(c) = stdin.next() {
                let c = c.unwrap() as char;
                if c == 'q' {
                    break;
                }
                else {
                    let i = c.to_digit(10).unwrap() as usize;
                    if i < within_range.len() {
                        let mut send = String::new();
                        send.push_str(within_range.iter().nth(i).unwrap().0);
                        send.push_str("\n");
                        stream.write(send.as_bytes()).unwrap();
                    }
                }
            }
        }
        stdout.flush().unwrap();
    }
}

fn get_target_status(navigation : &Navigation, mass_name : &String) -> String {
    match navigation.target_name.clone() {
        Some(name) => {
            match &name == mass_name {
                true => serde_json::to_string(&navigation.status).unwrap(),
                false => String::new()
            }
        },
        _ => String::new(),
    }
}