1 files changed, 20 insertions, 1 deletions

diff --git a/src/bin/client/repl.rs b/src/bin/client/repl.rs
index 40b267e..dd55acc 100644
--- a/src/bin/client/repl.rs
+++ b/src/bin/client/repl.rs
@@ -273,11 +273,15 @@ impl ClientState {
                                             }
                                         }
                                         "connect" => {
+                                            if self.stream.is_some() {
+                                                println!("\nAlready connected. Disconnect first.");
+                                                continue 'outer;
+                                            }
                                             let addr_str = if parts.len() < 2 {
                                                 if let Some(h) = &self.addr {
                                                     h.clone()
                                                 } else {
-                                                    println!("\nNo host specified and no previous connection. Usage: connect <ip>:<port>");
+                                                    println!("\nNo host specified. Usage: connect <ip>:<port>");
                                                     continue 'outer;
                                                 }
                                             } else {
@@ -591,6 +595,20 @@ impl ClientState {
                                                 Err(e) => println!("\nFailed to forget host: {}", e),
                                             }
                                         }
+                                                                                "hosts" => {
+                                            let hosts = database::get_all_hosts(&self.conn);
+                                            if hosts.is_empty() {
+                                                println!("\nNo remembered hosts.");
+                                            } else {
+                                                println!("\nRemembered hosts:");
+                                                for (host, fp) in &hosts {
+                                                    match fp {
+                                                        Some(fp) => println!("  {} - {}", host, fp),
+                                                        None => println!("  {} - (no fingerprint)", host),
+                                                    }
+                                                }
+                                            }
+                                        }
                                         _ => help(),
                                     }
                                     break;
@@ -752,6 +770,7 @@ fn get_commands() -> Vec<String> {
         "help".to_string(),
         "quit".to_string(),
         "forget-host".to_string(),
+        "hosts".to_string(),
     ]
 }