comandare 0.2.0

TCP Command Execution
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

use std::net::{TcpStream, Shutdown};
use std::io::{Read, Write};

use crate::Logger;
use crate::format_address;

pub fn connect(log: Logger, host: &str, port: u16, command: &str) {
    let address = format_address(host, port);
    match TcpStream::connect(&address) {
        Ok(mut stream) => {
            log.out(format!("Successfully connected to server in port {}", port));

            let mut buffer = Vec::new();

            stream.write_all(command.as_bytes()).unwrap();
            stream.shutdown(Shutdown::Write).unwrap();

            log.out(format!("Sent command to {}", &address));

            match stream.read_to_end(&mut buffer) {
                Ok(_) => {
                    let response = String::from_utf8( buffer ).unwrap();
                    // todo - add always out feature
                    println!("{}", response);
                },
                Err(e) => {
                    log.out(format!("Failed to receive data: {}", e));
                }
            }
        },
        Err(e) => {
            log.out(format!("Failed to connect: {}", e));
        }
    }
    log.out(format!("Terminated."));
}