Implement Authorization

nats-client/src/main.rs

@@ -17,14 +17,25 @@ fn main() {
                 eprintln!("Connection to server is refused. Please check if it is really running.");
                 exit(1);
             }
-            _ => panic!("{}", e),
+            io::ErrorKind::InvalidData => {
+                if e.to_string().contains("Authorization Violation") {
+                    eprintln!("Invalid login credentials.");
+                    exit(1);
+                } else {
+                    panic!("NATS response was invalid: {}", e);
+                }
+            }
+            _ => panic!("{:?}", e),
         }
     }
 }
 
 fn run_chat_program() -> io::Result<()> {
-    let nc = nats_cli::connect_to_nats()?;
     let username = ask_user_name();
+    // TODO: Ask password from user
+    let password = username.clone();
+    let nc = nats_cli::connect_to_nats(&username, &password)?;
+
     println!(
         "Hello {}, please write your message. Use q to quit:",
         username

nats-client/src/nats_cli.rs

@@ -5,10 +5,12 @@ use nats::{Connection, Handler};
 use crate::chat_message::ChatMessage;
 
 const SERVER: &str = "127.0.0.1";
-const SUBJECT_MESSAGES: &str = "here.happens.messaging";
+const SUBJECT_MESSAGES: &str = "telestion.chat";
 
-pub fn connect_to_nats() -> io::Result<Connection> {
-    nats::connect(SERVER)
+pub fn connect_to_nats(username: &str, password: &str) -> io::Result<Connection> {
+    nats::Options::with_user_pass(username, password)
+        .with_name("Chat Application")
+        .connect(SERVER)
 }
 
 pub fn subscribe_to_chat_messages(username: String, nc: &Connection) -> io::Result<Handler> {
@@ -23,7 +25,7 @@ pub fn subscribe_to_chat_messages(username: String, nc: &Connection) -> io::Resu
                             println!("Received {}", message);
                         }
                     }
-                    Err(e) => eprintln!("{}", e),
+                    Err(e) => eprintln!("Error from NATS: {}", e),
                 };
                 Ok(())
             });