Mark current session in the output of list-sessions

1 files changed, 8 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index 122747b65..e4651fd1f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -58,10 +58,17 @@ fn list_sessions() {
     match fs::read_dir(&*ZELLIJ_SOCK_DIR) {
         Ok(files) => {
             let mut is_empty = true;
+            let session_name = std::env::var("ZELLIJ_SESSION_NAME").unwrap_or("".into());
             files.for_each(|file| {
                 let file = file.unwrap();
                 if file.file_type().unwrap().is_socket() {
-                    println!("{}", file.file_name().into_string().unwrap());
+                    let fname = file.file_name().into_string().unwrap();
+                    let suffix = if session_name == fname {
+                        " (current)"
+                    } else {
+                        ""
+                    };
+                    println!("{}{}", fname, suffix);
                     is_empty = false;
                 }
             });