diff --git a/src/main.rs b/src/main.rs index 3bf321c..9ef7d71 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,12 +1,11 @@ -use std::net::{SocketAddr, IpAddr, ToSocketAddrs}; -use std::net::TcpStream; -use std::{process, thread}; -use std::fmt::{Debug}; -use std::time::Duration; use chrono::Local; -use clap::{Parser}; +use clap::Parser; use dns_lookup::lookup_host; - +use std::fmt::Debug; +use std::net::TcpStream; +use std::net::{IpAddr, SocketAddr, ToSocketAddrs}; +use std::time::Duration; +use std::{process, thread}; /// Simple program to greet a person #[derive(Parser, Debug)] @@ -73,16 +72,16 @@ fn get_target_socket(args: &Args) -> Result { if let Ok(sock_iter) = args.host.to_socket_addrs() { for sock in sock_iter { if !args.v4 && !args.v6 { - return Ok(sock) + return Ok(sock); } else if args.v4 && sock.is_ipv4() { - return Ok(sock) + return Ok(sock); } else if args.v6 && sock.is_ipv6() { return Ok(sock); } } Err(format!("Error: Cannot parse host")) - } else if let Ok(target_result) = get_target_ip(&args.host, &args.v4, &args.v6){ - return Ok(SocketAddr::new(target_result, args.port)) + } else if let Ok(target_result) = get_target_ip(&args.host, &args.v4, &args.v6) { + return Ok(SocketAddr::new(target_result, args.port)); } else { Err(format!("Error: Unable to resolve Host")) } @@ -101,22 +100,26 @@ fn get_target_ip(s: &str, v4: &bool, v6: &bool) -> Result { return Ok(ip as IpAddr); } } - Err(format!( - "Failed to Lookup IP", - )) + Err(format!("Failed to Lookup IP",)) } else { - Err(format!( - "Unable to resolve Address", - )) + Err(format!("Unable to resolve Address",)) } } fn ping(target: &SocketAddr, timeout: &Duration) -> Result<(), ()> { if let Ok(_stream) = TcpStream::connect_timeout(&target, *timeout) { - println!("{} - {} - Connection Successful", Local::now().format("%Y-%m-%d %H:%M:%S"), target); + println!( + "{} - {} - Connection Successful", + Local::now().format("%Y-%m-%d %H:%M:%S"), + target + ); Ok(()) } else { - println!("{} - {} - Connection Failed", Local::now().format("%Y-%m-%d %H:%M:%S"), target); + println!( + "{} - {} - Connection Failed", + Local::now().format("%Y-%m-%d %H:%M:%S"), + target + ); Err(()) } -} \ No newline at end of file +}