extern crate a03;
extern crate serde;
extern crate serde_json;
#[macro_use]
extern crate serde_derive;

use a03::*;
use std::net::{TcpStream, Shutdown};
use std::io::Write;
use std::net::TcpListener;

fn main() {
    let endpoint = parse_endpoint_from_cli(0);
    let listener = TcpListener::bind(endpoint).unwrap();
    register_with_meta_server();

    for stream in listener.incoming() {
        let mut stream = stream.unwrap();
//        let mut buf = Vec::new();
//        match stream.read_to_end(&mut buf) {
//            Ok(size) => {
//                println!("Total bytes: {}", size);
//                let mut copy = File::create("new_version").unwrap();
//                copy.write_all(&buf[..]).unwrap();
//            },
//            Err(e) => println!("{}", e),
//        }
        match serde_json::from_reader(&mut stream) {
            Ok(packet @ Packet { .. }) => match packet.p_type {
//                PacketType::GetFiles => shutdown(&mut stream),
//                PacketType::PutFile => put(&mut stream, &packet.json.unwrap(), &mut Vec::new()),
                PacketType::ShutdownDataNode => shutdown(&mut stream),
                _ => (),
            },
            Err(e) => println!("Error parsing json: {}", e.to_string()),
        };
    }
}

fn register_with_meta_server() {
    let mut stream = TcpStream::connect("localhost:6770").unwrap();
    serde_json::to_writer(
        &mut stream,
        &Packet {
            p_type: PacketType::NodeRegistration,
            json: Some(serde_json::to_string(
                &NodeRegistration { register: true, ip: String::from("localhost"), port: 6771 }).unwrap()),
        })
        .unwrap();
    println!("Registered myself");
    stream.flush().unwrap();
    stream.shutdown(Shutdown::Write).unwrap();
    let result: Packet = serde_json::from_reader(&mut stream).unwrap();
    println!("{:?}", result);
}

//fn put(stream: &mut TcpStream, json: &String, files: &mut Vec<String>) {
//    let files: PutFiles = serde_json::from_str(json).unwrap();
//}

fn shutdown(stream: &mut TcpStream) {
    let mut stream = TcpStream::connect("localhost:6770").unwrap();
    serde_json::to_writer(
        &mut stream,
        &Packet {
            p_type: PacketType::NodeRegistration,
            json: Some(serde_json::to_string(
                &NodeRegistration { register: false, ip: String::from("localhost"), port: 6771 }).unwrap()),
        })
        .unwrap();
    println!("Unregistered myself");
    stream.flush().unwrap();
    stream.shutdown(Shutdown::Write).unwrap();
    let result: Packet = serde_json::from_reader(&mut stream).unwrap();
    println!("{:?}", result);
    std::process::exit(0);
}