feat: reimplement exclude option

1 files changed, 5 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index 365d42a..0559985 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -11,6 +11,10 @@ pub struct Config {
     /// Tmux socket name
     #[arg(short, long, default_value = "ssh")]
     pub socket: String,
+
+    /// Exclude host from output (can be specified multiple times)
+    #[arg(short, long)]
+    pub exclude: Vec<sshr::Host>,
 }
 
 #[tokio::main]
@@ -24,7 +28,7 @@ async fn main() -> Result<(), std::io::Error> {
 
     drop(tx);
 
-    let mut output = HashSet::new();
+    let mut output: HashSet<sshr::Host> = HashSet::from_iter(config.exclude.into_iter());
 
     let mut stdout = stdout();
     while let Some(host) = rx.recv().await {