fix: restore double prefix passthrough in client mode

2 files changed, 32 insertions, 4 deletions

diff --git a/src/app/input/mod.rs b/src/app/input/mod.rs
index e2454861..3beeac6b 100644
--- a/src/app/input/mod.rs
+++ b/src/app/input/mod.rs
@@ -36,7 +36,7 @@ pub(crate) use self::{
         handle_confirm_close_key, handle_context_menu_key, handle_global_menu_key,
         handle_keybind_help_key, handle_rename_key, handle_resize_key,
     },
-    navigate::{handle_navigate_key, terminal_direct_navigation_action},
+    navigate::terminal_direct_navigation_action,
     settings::open_settings,
 };
 use self::{
@@ -56,12 +56,13 @@ impl App {
     pub(super) async fn handle_key(&mut self, key: TerminalKey) {
         match self.state.mode {
             Mode::Terminal => self.handle_terminal_key(key).await,
+            Mode::Navigate => self.handle_navigate_key(key),
             _ => {
                 let key = key.as_key_event();
                 match self.state.mode {
                     Mode::Onboarding => self.handle_onboarding_key(key),
                     Mode::ReleaseNotes => self.handle_release_notes_key(key),
-                    Mode::Navigate => self.handle_navigate_key(key),
+                    Mode::Navigate => unreachable!(),
                     Mode::RenameWorkspace | Mode::RenameTab => {
                         handle_rename_key(&mut self.state, key)
                     }
diff --git a/src/app/input/navigate.rs b/src/app/input/navigate.rs
index b961c6d3..d55eedea 100644
--- a/src/app/input/navigate.rs
+++ b/src/app/input/navigate.rs
@@ -1,5 +1,6 @@
 use std::process::{Command, Stdio};
 
+use bytes::Bytes;
 use crossterm::event::{KeyCode, KeyEvent};
 use ratatui::layout::Direction;
 
@@ -8,6 +9,7 @@ use crate::{
         state::{key_matches, AppState, Mode},
         App,
     },
+    input::TerminalKey,
     layout::NavDirection,
 };
 
@@ -68,10 +70,18 @@ pub(crate) fn terminal_direct_navigation_action(
 }
 
 impl App {
-    pub(super) fn handle_navigate_key(&mut self, key: KeyEvent) {
+    pub(crate) fn handle_navigate_key(&mut self, raw_key: TerminalKey) {
+        let key = raw_key.as_key_event();
         self.state.update_dismissed = true;
 
-        if self.state.is_prefix(&key) || key.code == KeyCode::Esc {
+        if self.state.is_prefix(&key) {
+            if !self.pass_through_key_to_focused_pane(raw_key) {
+                leave_navigate_mode(&mut self.state);
+            }
+            return;
+        }
+
+        if key.code == KeyCode::Esc {
             leave_navigate_mode(&mut self.state);
             return;
         }
@@ -90,6 +100,23 @@ impl App {
         }
     }
 
+    fn pass_through_key_to_focused_pane(&mut self, key: TerminalKey) -> bool {
+        let Some(ws) = self.state.active.and_then(|i| self.state.workspaces.get(i)) else {
+            return false;
+        };
+        let Some(rt) = ws.focused_runtime() else {
+            return false;
+        };
+
+        let bytes = rt.encode_terminal_key(key);
+        if bytes.is_empty() || rt.try_send_bytes(Bytes::from(bytes)).is_err() {
+            return false;
+        }
+
+        self.state.mode = Mode::Terminal;
+        true
+    }
+
     fn launch_custom_command(&mut self, binding: crate::config::CustomCommandKeybind) {
         let previous_toast = self.state.toast.clone();
         let result = match binding.action {