allow undo with 'u'

1 files changed, 10 insertions, 0 deletions

diff --git a/src/main.rs b/src/main.rs
index 15e12dc..69a10cd 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -86,6 +86,11 @@ fn prompt(anki: &AnkiClient) {
         match event::read().unwrap() {
             Event::Key(e) => match e.code {
                 KeyCode::Enter => break,
+                KeyCode::Char('u') => {
+                    anki.request(GuiUndoRequest {}).unwrap();
+                    event::read().unwrap();
+                    return;
+                }
                 _ => (),
             },
             _ => (),
@@ -104,6 +109,11 @@ fn prompt(anki: &AnkiClient) {
             Event::Key(e) => match e.code {
                 KeyCode::Char(AGAIN) => break Ease::Again,
                 KeyCode::Char(GOOD) => break Ease::Good,
+                KeyCode::Char('u') => {
+                    anki.request(GuiUndoRequest {}).unwrap();
+                    event::read().unwrap();
+                    return;
+                }
                 _ => (),
             },
             _ => (),