fix: refresh forwarded ssh agents after reconnect (#4443)

refs #1931

2 files changed, 10 insertions, 0 deletions

diff --git a/src/api/schema/server.rs b/src/api/schema/server.rs
index b274bc95..1adb62f5 100644
--- a/src/api/schema/server.rs
+++ b/src/api/schema/server.rs
@@ -14,6 +14,12 @@ pub struct ServerLiveHandoffParams {
 }
 
 #[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, schemars::JsonSchema)]
+pub struct ServerSshAgentRegisterParams {
+    /// Absolute remote-host agent socket. Registration lasts until this API connection closes.
+    pub socket_path: String,
+}
+
+#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, schemars::JsonSchema)]
 pub struct ServerCapabilities {
     pub live_handoff: bool,
     #[serde(default)]
@@ -27,4 +33,7 @@ pub struct ServerCapabilities {
     /// Whether this server supports endpoint health probes.
     #[serde(default)]
     pub health_check: bool,
+    /// Supports connection-scoped `server.ssh_agent.register` on the local JSON API.
+    #[serde(default)]
+    pub ssh_agent_registration: bool,
 }
diff --git a/src/api/schema/tests.rs b/src/api/schema/tests.rs
index 4f15b322..3b43133b 100644
--- a/src/api/schema/tests.rs
+++ b/src/api/schema/tests.rs
@@ -726,6 +726,7 @@ fn success_response_round_trips() {
                 endpoint_protocol_generation: Some(1),
                 surface_interest: true,
                 health_check: true,
+                ssh_agent_registration: false,
             }),
         },
     };