fix: exclude local session when attached

1 files changed, 25 insertions, 3 deletions

diff --git a/src/tmux.rs b/src/tmux.rs
index b53c172..8a0d581 100644
--- a/src/tmux.rs
+++ b/src/tmux.rs
@@ -1,4 +1,4 @@
-use std::process::Command;
+use std::{path::PathBuf, process::Command};
 
 use crate::Session;
 
@@ -39,9 +39,18 @@ impl Tmux {
     }
 
     pub fn attached(&self) -> Result<Session, Error> {
+        if Self::current_socket()?
+            .file_name()
+            .and_then(|f| f.to_str())
+            .is_some_and(|f| f != self.socket)
+        {
+            return Self::local_session()?
+                .first()
+                .ok_or(Error::NotFound)
+                .cloned();
+        }
+
         let stdout = Command::new("tmux")
-            .arg("-L")
-            .arg(&self.socket)
             .arg("display")
             .arg("-p")
             .arg(Self::SESSION_FORMAT)
@@ -78,6 +87,19 @@ impl Tmux {
 
         Ok(sessions)
     }
+
+    pub fn current_socket() -> Result<PathBuf, Error> {
+        let stdout = Command::new("tmux")
+            .arg("display")
+            .arg("-p")
+            .arg("#{session_path}")
+            .output()?
+            .stdout;
+
+        std::str::from_utf8(&stdout)
+            .map(Into::into)
+            .map_err(Into::into)
+    }
 }
 
 #[cfg(test)]