//! Simulate a broken client.
#[macro_use]
extern crate serde_derive;
extern crate text_io;
extern crate srve;

mod shared;

use std::net::{TcpStream, Shutdown};
use std::io::Write;
use std::error::Error;
use text_io::read;
use shared::{Msg, ADDR};
use srve::Client;

/// Stablish a connection and send an incomplete message.
/// Closes the connection before sending the expected amount of bytes.
fn pk_incomplete() -> Result<(), Box<dyn Error>>{
    let mut s = TcpStream::connect(ADDR)?;

    // server expects 8 bytes for 'len' attribute
    let mut buf0 = Vec::new();
    buf0.push(8u8);
    buf0.resize(8, 0u8);
    s.write_all(buf0.as_slice())?;

    // now the server expects 8 bytes of data, let's send only 5
    let buf1 = [1u8; 5];
    s.write_all(&buf1[..])?;

    // then let's close the connection
    s.shutdown(Shutdown::Both)?;
    Ok(())
}

/// Stablish a connection and send a bad message.
/// Sends a message that will trigger a deserialization error on the server.
fn pk_bad() -> Result<(), Box<dyn Error>>{
    let mut s = TcpStream::connect(ADDR)?;

    // server expects 8 bytes for 'len' attribute
    let mut buf0 = Vec::new();
    buf0.push(8u8);
    buf0.resize(8, 0u8);
    s.write_all(buf0.as_slice())?;

    // now the server expects 8 bytes of data, let's send only 5
    let buf1 = [1u8; 8];
    s.write_all(&buf1[..])?;

    // then let's close the connection
    s.shutdown(Shutdown::Both)?;
    Ok(())
}

/// Stablish a connection, sends a good pk, but then closes before receiving the
/// server response-
fn pk_no_recv() -> Result<(), Box<dyn Error>>{
    let mut c = Client::connect(ADDR)?;
    c.send(Msg::Print)?;
    c.close()?;
    Ok(())
}
/// Allows the user to simulate multiple types of client errors.
fn main() {
    println!(" ...::: COMMANDS :::... ");
    println!();
    println!("> bad /* simulates a bad message */");
    println!("> incomplete /* simulates an incomplete message */");
    println!("> norecv /* simulates a client that closes to fast */");
    println!();

    loop {
        let s: String = read!();

        match s.as_str() {
            "incomplete" => {
                pk_incomplete().unwrap();
            }
            "bad" => {
                pk_bad().unwrap();
            }
            "norecv" => {
                pk_no_recv().unwrap();
            }
            _ => {}
        }
    }
}