From 8a628d51a653d6f34e5f4c89de1829d72dea8c54 Mon Sep 17 00:00:00 2001
From: Dreaded_X <tim@huizinga.dev>
Date: Thu, 10 Apr 2025 17:05:58 +0200
Subject: [PATCH] Allow protected to be set from cli

---
 src/ssh.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/ssh.rs b/src/ssh.rs
index 7a5f150..d6aa3de 100644
--- a/src/ssh.rs
+++ b/src/ssh.rs
@@ -167,8 +167,11 @@ impl Handler {
 #[command(version, about, long_about = None)]
 struct Args {
     /// Make all tunnels public by default instead of private
-    #[arg(short, long)]
+    #[arg(long, group = "access")]
     public: bool,
+
+    #[arg(long, group = "access")]
+    protected: bool,
 }
 
 impl russh::server::Handler for Handler {
@@ -244,6 +247,11 @@ impl russh::server::Handler for Handler {
                     self.set_access_all(TunnelAccess::Public).await;
                     self.redraw().await?;
                 }
+                if args.protected {
+                    trace!("Making tunnels protected");
+                    self.set_access_all(TunnelAccess::Protected).await;
+                    self.redraw().await?;
+                }
             }
             Err(err) => {
                 trace!("Sending help message and disconnecting");