feat: forward the Agent's ports while run or attach is connected

crates/ally-cli/src/commands/attach.rs

@@ -2,6 +2,7 @@
 //! current project's copy, or in the Agent's home folder when there is no project.
 
 use crate::client::{self, Refused, unexpected};
+use crate::forward::{Forwarding, Shown};
 use crate::project::{self, Local};
 use crate::route;
 use crate::ssh::{self, RemoteCommand};
@@ -54,6 +55,15 @@ pub async fn attach(
         &place,
         style.colored(),
     );
+    let forwarding = Forwarding::start(
+        target,
+        Shown::Session {
+            tmux: session.tmux.clone(),
+            socket: session.socket.clone(),
+            id: session.job.id.clone(),
+            label: label.clone(),
+        },
+    );
     let mut remote = RemoteCommand::to(
         target,
         session.tmux,
@@ -67,6 +77,7 @@ pub async fn attach(
         presentation::row("Return", "ally attach").trim_end()
     );
     let code = super::run::interact(target, &remote, lost).await?;
+    forwarding.stop().await;
     if pulls_after(code)
         && let Err(error) = after_leaving(target, local.as_ref(), &session.job.id).await
     {

crates/ally-cli/src/commands/run.rs

@@ -1,6 +1,7 @@
 //! `ally run <cmd>`: sync the project, then run a command in the Agent copy.
 
 use crate::client::{self, Refused};
+use crate::forward::{Forwarding, Shown};
 use crate::project::{self, Local};
 use crate::route;
 use crate::ssh::{Disconnected, RemoteCommand};
@@ -62,7 +63,9 @@ pub async fn run(agent: Option<String>, cmd: Vec<String>) -> anyhow::Result<i32>
     );
 
     let remote = RemoteCommand::to(target, target.program().to_string(), args);
+    let forwarding = Forwarding::start(target, Shown::Terminal { raw: remote.tty });
     let code = interact(target, &remote, lost_connection(&target.name)).await?;
+    forwarding.stop().await;
     eprintln!("\n{}", finished(code, started.elapsed(), style));
     Ok(code)
 }