From 078f117916b1eafbc28318c914733a01c5ae3867 Mon Sep 17 00:00:00 2001
From: Dheepak Krishnamurthy <me@kdheepak.com>
Date: Thu, 18 Mar 2021 07:02:56 -0600
Subject: [PATCH 1/4] Add shortcuts

---
 src/app.rs       | 123 +++++++++++++++++++++++++++++++++++++++++++++++
 src/config.rs    |  12 +++++
 src/keyconfig.rs |  20 ++++++++
 3 files changed, 155 insertions(+)

diff --git a/src/app.rs b/src/app.rs
index 22fa489..087afb7 100644
--- a/src/app.rs
+++ b/src/app.rs
@@ -1083,6 +1083,49 @@ impl TTApp {
         }
     }
 
+    pub fn task_shortcut(&mut self, s: usize) -> Result<(), String> {
+        if self.tasks.lock().unwrap().is_empty() {
+            return Ok(());
+        }
+        let selected = self.task_table_state.selected().unwrap_or_default();
+        let task_id = self.tasks.lock().unwrap()[selected].id().unwrap_or_default();
+        let task_uuid = *self.tasks.lock().unwrap()[selected].uuid();
+        let shell = &self.config.uda_shortcuts[s];
+
+        if shell.is_empty() {
+            return Err("Trying to run empty shortcut.".to_string());
+        }
+
+        let shell = format!("{} {}", shell, task_uuid);
+        match shlex::split(&shell) {
+            Some(cmd) => {
+                let mut command = Command::new(&cmd[0]);
+                for s in cmd.iter().skip(1) {
+                    command.arg(&s);
+                }
+                let output = command.output();
+                match output {
+                    Ok(o) => {
+                        if o.status.success() {
+                            Ok(())
+                        } else {
+                            Err(format!(
+                                "Unable to run shortcut {}. Failed with status code {}",
+                                s,
+                                o.status.code().unwrap()
+                            ))
+                        }
+                    }
+                    Err(_) => Err(format!(
+                        "Cannot run `{}`. Check documentation for more information",
+                        shell,
+                    )),
+                }
+            }
+            None => Err(format!("Unable to run `{}`. Cannot shlex split `{}`", shell, shell)),
+        }
+    }
+
     pub fn task_modify(&mut self) -> Result<(), String> {
         if self.tasks.lock().unwrap().is_empty() {
             return Ok(());
@@ -1562,6 +1605,86 @@ impl TTApp {
                     self.mode = AppMode::TaskHelpPopup;
                 } else if input == self.keyconfig.filter {
                     self.mode = AppMode::TaskFilter;
+                } else if input == self.keyconfig.shortcut0 {
+                    match self.task_shortcut(0) {
+                        Ok(_) => self.update(true)?,
+                        Err(e) => {
+                            self.mode = AppMode::TaskError;
+                            self.error = e;
+                        }
+                    }
+                } else if input == self.keyconfig.shortcut1 {
+                    match self.task_shortcut(1) {
+                        Ok(_) => self.update(true)?,
+                        Err(e) => {
+                            self.mode = AppMode::TaskError;
+                            self.error = e;
+                        }
+                    }
+                } else if input == self.keyconfig.shortcut2 {
+                    match self.task_shortcut(2) {
+                        Ok(_) => self.update(true)?,
+                        Err(e) => {
+                            self.mode = AppMode::TaskError;
+                            self.error = e;
+                        }
+                    }
+                } else if input == self.keyconfig.shortcut3 {
+                    match self.task_shortcut(3) {
+                        Ok(_) => self.update(true)?,
+                        Err(e) => {
+                            self.mode = AppMode::TaskError;
+                            self.error = e;
+                        }
+                    }
+                } else if input == self.keyconfig.shortcut4 {
+                    match self.task_shortcut(4) {
+                        Ok(_) => self.update(true)?,
+                        Err(e) => {
+                            self.mode = AppMode::TaskError;
+                            self.error = e;
+                        }
+                    }
+                } else if input == self.keyconfig.shortcut5 {
+                    match self.task_shortcut(5) {
+                        Ok(_) => self.update(true)?,
+                        Err(e) => {
+                            self.mode = AppMode::TaskError;
+                            self.error = e;
+                        }
+                    }
+                } else if input == self.keyconfig.shortcut6 {
+                    match self.task_shortcut(6) {
+                        Ok(_) => self.update(true)?,
+                        Err(e) => {
+                            self.mode = AppMode::TaskError;
+                            self.error = e;
+                        }
+                    }
+                } else if input == self.keyconfig.shortcut7 {
+                    match self.task_shortcut(7) {
+                        Ok(_) => self.update(true)?,
+                        Err(e) => {
+                            self.mode = AppMode::TaskError;
+                            self.error = e;
+                        }
+                    }
+                } else if input == self.keyconfig.shortcut8 {
+                    match self.task_shortcut(8) {
+                        Ok(_) => self.update(true)?,
+                        Err(e) => {
+                            self.mode = AppMode::TaskError;
+                            self.error = e;
+                        }
+                    }
+                } else if input == self.keyconfig.shortcut9 {
+                    match self.task_shortcut(9) {
+                        Ok(_) => self.update(true)?,
+                        Err(e) => {
+                            self.mode = AppMode::TaskError;
+                            self.error = e;
+                        }
+                    }
                 } else if input == self.keyconfig.zoom {
                     self.task_report_show_info = !self.task_report_show_info;
                 } else if input == self.keyconfig.context_menu {
diff --git a/src/config.rs b/src/config.rs
index c44d735..9912b69 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -52,6 +52,7 @@ pub struct Config {
     pub uda_calendar_months_per_row: usize,
     pub uda_style_context_active: Style,
     pub uda_style_calendar_title: Style,
+    pub uda_shortcuts: Vec<String>,
 }
 
 impl Config {
@@ -76,6 +77,7 @@ impl Config {
             uda_calendar_months_per_row: Self::get_uda_months_per_row(),
             uda_style_calendar_title: Self::get_uda_style("calendar.title").unwrap_or_default(),
             uda_style_context_active: Self::get_uda_style("context.active").unwrap_or_default(),
+            uda_shortcuts: Self::get_uda_shortcuts(),
         })
     }
 
@@ -83,6 +85,16 @@ impl Config {
         HashMap::new()
     }
 
+    fn get_uda_shortcuts() -> Vec<String> {
+        let mut v = vec![];
+        for s in 0..9 {
+            let c = format!("uda.taskwarrior-tui.shortcuts.{}", s);
+            let s = Self::get_config(&c);
+            v.push(s);
+        }
+        return v;
+    }
+
     fn get_uda_style(config: &str) -> Option<Style> {
         let c = format!("uda.taskwarrior-tui.style.{}", config);
         let s = Self::get_config(&c);
diff --git a/src/keyconfig.rs b/src/keyconfig.rs
index 1eaa337..b92ffba 100644
--- a/src/keyconfig.rs
+++ b/src/keyconfig.rs
@@ -31,6 +31,16 @@ pub struct KeyConfig {
     pub context_menu: Key,
     pub next_tab: Key,
     pub previous_tab: Key,
+    pub shortcut0: Key,
+    pub shortcut1: Key,
+    pub shortcut2: Key,
+    pub shortcut3: Key,
+    pub shortcut4: Key,
+    pub shortcut5: Key,
+    pub shortcut6: Key,
+    pub shortcut7: Key,
+    pub shortcut8: Key,
+    pub shortcut9: Key,
 }
 
 impl Default for KeyConfig {
@@ -60,6 +70,16 @@ impl Default for KeyConfig {
             context_menu: Key::Char('c'),
             next_tab: Key::Char(']'),
             previous_tab: Key::Char('['),
+            shortcut0: Key::Char('0'),
+            shortcut1: Key::Char('1'),
+            shortcut2: Key::Char('2'),
+            shortcut3: Key::Char('3'),
+            shortcut4: Key::Char('4'),
+            shortcut5: Key::Char('5'),
+            shortcut6: Key::Char('6'),
+            shortcut7: Key::Char('7'),
+            shortcut8: Key::Char('8'),
+            shortcut9: Key::Char('9'),
         }
     }
 }

From d351b86c30ca64a333bc450eb2a4e802e06f6e86 Mon Sep 17 00:00:00 2001
From: Dheepak Krishnamurthy <me@kdheepak.com>
Date: Thu, 18 Mar 2021 07:05:13 -0600
Subject: [PATCH 2/4] Update documentation

---
 KEYBINDINGS.md            | 2 ++
 docs/taskwarrior-tui.1.md | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/KEYBINDINGS.md b/KEYBINDINGS.md
index ab6838a..877eb08 100644
--- a/KEYBINDINGS.md
+++ b/KEYBINDINGS.md
@@ -47,6 +47,8 @@ Keybindings for task report:
 
     !: {string}                          - Custom shell command
 
+    0-9: {string}                        - Run user defined shortcuts
+
     c: context switcher menu             - Open context switcher menu
 
     ?: help                              - Help menu
diff --git a/docs/taskwarrior-tui.1.md b/docs/taskwarrior-tui.1.md
index 078bf8d..7a4f1f5 100644
--- a/docs/taskwarrior-tui.1.md
+++ b/docs/taskwarrior-tui.1.md
@@ -103,6 +103,9 @@ Keybindings for task report:
 `!`
 : {string}                          - Custom shell command
 
+`0-9`
+: {string}                          - Run user defined shortcuts
+
 `c`
 : context switcher menu             - Open context switcher menu
 

From d52699abe7f27fce1d732aafad58fa801fd4b5db Mon Sep 17 00:00:00 2001
From: Dheepak Krishnamurthy <me@kdheepak.com>
Date: Thu, 18 Mar 2021 07:25:58 -0600
Subject: [PATCH 3/4] Expand tilde and show error message

---
 src/app.rs | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/app.rs b/src/app.rs
index 087afb7..9254178 100644
--- a/src/app.rs
+++ b/src/app.rs
@@ -1097,6 +1097,7 @@ impl TTApp {
         }
 
         let shell = format!("{} {}", shell, task_uuid);
+        let shell = shellexpand::tilde(&shell).into_owned();
         match shlex::split(&shell) {
             Some(cmd) => {
                 let mut command = Command::new(&cmd[0]);
@@ -1116,10 +1117,7 @@ impl TTApp {
                             ))
                         }
                     }
-                    Err(_) => Err(format!(
-                        "Cannot run `{}`. Check documentation for more information",
-                        shell,
-                    )),
+                    Err(s) => Err(format!("`{}` failed: {}", shell, s,)),
                 }
             }
             None => Err(format!("Unable to run `{}`. Cannot shlex split `{}`", shell, shell)),

From bde9346cafaab9cc76768e4758128a7496e8d1c5 Mon Sep 17 00:00:00 2001
From: Dheepak Krishnamurthy <me@kdheepak.com>
Date: Thu, 18 Mar 2021 07:26:55 -0600
Subject: [PATCH 4/4] Run cargo clippy

---
 src/config.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/config.rs b/src/config.rs
index 9912b69..49ffa6c 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -92,7 +92,7 @@ impl Config {
             let s = Self::get_config(&c);
             v.push(s);
         }
-        return v;
+        v
     }
 
     fn get_uda_style(config: &str) -> Option<Style> {