From b7b3e942fa23157ec832e2bb07c22f0573b158f6 Mon Sep 17 00:00:00 2001
From: Rajas Paranjpe <52586855+ChocolateLoverRaj@users.noreply.github.com>
Date: Fri, 3 May 2024 14:37:01 -0700
Subject: [PATCH] Add CopyOrSigint shortcut for Ctrl+C

---
 src/key_bind.rs |  1 +
 src/main.rs     | 23 +++++++++++++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/src/key_bind.rs b/src/key_bind.rs
index 04cff75..c2a234e 100644
--- a/src/key_bind.rs
+++ b/src/key_bind.rs
@@ -23,6 +23,7 @@ pub fn key_binds() -> HashMap<KeyBind, Action> {
     // Standard key bindings
     bind!([Ctrl, Shift], Key::Character("A".into()), SelectAll);
     bind!([Ctrl, Shift], Key::Character("C".into()), Copy);
+    bind!([Ctrl], Key::Character("c".into()), CopyOrSigint);
     bind!([Ctrl, Shift], Key::Character("F".into()), Find);
     bind!([Ctrl, Shift], Key::Character("N".into()), WindowNew);
     bind!([Ctrl, Shift], Key::Character("Q".into()), WindowClose);
diff --git a/src/main.rs b/src/main.rs
index f8a36e0..c34acd6 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -168,6 +168,7 @@ pub enum Action {
     About,
     ColorSchemes(ColorSchemeKind),
     Copy,
+    CopyOrSigint,
     CopyPrimary,
     Find,
     PaneFocusDown,
@@ -214,6 +215,7 @@ impl MenuAction for Action {
                 Message::ToggleContextPage(ContextPage::ColorSchemes(*color_scheme_kind))
             }
             Self::Copy => Message::Copy(entity_opt),
+            Self::CopyOrSigint => Message::CopyOrSigint(entity_opt),
             Self::CopyPrimary => Message::CopyPrimary(entity_opt),
             Self::Find => Message::Find(true),
             Self::PaneFocusDown => Message::PaneFocusAdjacent(pane_grid::Direction::Down),
@@ -268,6 +270,7 @@ pub enum Message {
     ColorSchemeTabActivate(widget::segmented_button::Entity),
     Config(Config),
     Copy(Option<segmented_button::Entity>),
+    CopyOrSigint(Option<segmented_button::Entity>),
     CopyPrimary(Option<segmented_button::Entity>),
     DefaultBoldFontWeight(usize),
     DefaultDimFontWeight(usize),
@@ -1676,6 +1679,26 @@ impl Application for App {
                 }
                 return self.update_focus();
             }
+            Message::CopyOrSigint(entity_opt) => {
+                if let Some(tab_model) = self.pane_model.active() {
+                    let entity = entity_opt.unwrap_or_else(|| tab_model.active());
+                    if let Some(terminal) = tab_model.data::<Mutex<Terminal>>(entity) {
+                        let terminal = terminal.lock().unwrap();
+                        let term = terminal.term.lock();
+                        if let Some(text) = term.selection_to_string() {
+                            return Command::batch([clipboard::write(text), self.update_focus()]);
+                        } else {
+                            // Drop the lock for term so that input_scroll doesn't block forever
+                            drop(term);
+                            // 0x03 is ^C
+                            terminal.input_scroll(&[0x03]);
+                        }
+                    }
+                } else {
+                    log::warn!("Failed to get focused pane");
+                }
+                return self.update_focus();
+            }
             Message::CopyPrimary(entity_opt) => {
                 if let Some(tab_model) = self.pane_model.active() {
                     let entity = entity_opt.unwrap_or_else(|| tab_model.active());