From 72b472e54e570086ad26fc589f3afd3086c97084 Mon Sep 17 00:00:00 2001
From: Joshua Barretto <joshua.s.barretto@gmail.com>
Date: Tue, 2 Jul 2019 21:06:44 +0100
Subject: [PATCH] Added ability to specify server in chat-cli

---
 chat-cli/src/main.rs | 26 ++++++++++++++++++++------
 1 file changed, 20 insertions(+), 6 deletions(-)

diff --git a/chat-cli/src/main.rs b/chat-cli/src/main.rs
index 9f355718fa..98361736b6 100644
--- a/chat-cli/src/main.rs
+++ b/chat-cli/src/main.rs
@@ -1,10 +1,14 @@
 use client::{Client, Event};
 use common::{clock::Clock, comp};
 use log::{error, info};
-use std::io;
-use std::sync::mpsc;
-use std::thread;
-use std::time::Duration;
+use std::{
+    io,
+    net::{SocketAddr, ToSocketAddrs},
+    str::FromStr,
+    sync::mpsc,
+    thread,
+    time::Duration,
+};
 
 const TPS: u64 = 10; // Low value is okay, just reading messages.
 
@@ -30,9 +34,19 @@ fn main() {
     println!("Enter your username");
     let username = read_input();
 
+    println!("Enter the server address");
+    let server_addr = read_input();
+
     // Create a client.
-    let mut client =
-        Client::new(([127, 0, 0, 1], 59003), None).expect("Failed to create client instance");
+    let mut client = Client::new(
+        server_addr
+            .to_socket_addrs()
+            .expect("Invalid server address")
+            .next()
+            .unwrap(),
+        None,
+    )
+    .expect("Failed to create client instance");
 
     println!("Server info: {:?}", client.server_info);