From 005d5d3b19738728023f639c5d1837757bc64c4e Mon Sep 17 00:00:00 2001
From: Phuntsok Drak-pa <drakpa@drakpa.fr>
Date: Wed, 21 Mar 2018 12:55:10 +0100
Subject: [PATCH] minor changes

---
 Lucien/Rust/src/client.rs | 2 --
 Lucien/Rust/src/main.rs   | 6 +++++-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/Lucien/Rust/src/client.rs b/Lucien/Rust/src/client.rs
index 6da8106..694ba30 100644
--- a/Lucien/Rust/src/client.rs
+++ b/Lucien/Rust/src/client.rs
@@ -19,14 +19,12 @@ fn write_to_server(stream: TcpStream) {
         match &*get_entry() {
             "/quit" => {
                 println!("Disconnecting...");
-                // send!("BYE");
                 writeln!(writer, "BYE").unwrap();
                 writer.flush().unwrap();
                 println!("Disconnected!");
                 return ();
             }
             line => {
-                // send!(line);
                 writeln!(writer, "{}", line).unwrap();
                 writer.flush().unwrap();
             }
diff --git a/Lucien/Rust/src/main.rs b/Lucien/Rust/src/main.rs
index 0b90c76..df85bcb 100644
--- a/Lucien/Rust/src/main.rs
+++ b/Lucien/Rust/src/main.rs
@@ -21,7 +21,11 @@ fn main() {
         //                           Client opened                           //
         ///////////////////////////////////////////////////////////////////////
         println!("Client connecting on server {}:{}", args[1], args[2]);
-        let mut serv = args[1].clone();
+        let mut serv: String = if args[1] == "localhost" {
+            String::from("127.0.0.1")
+        } else {
+            args[1].clone()
+        };
         serv.push(':');
         serv.push_str(&args[2]);
         client::client(serv);