From 36400afab4d63120ea795a2612b80a2bbfba5091 Mon Sep 17 00:00:00 2001
From: Federico Terzi <federico-terzi@users.noreply.github.com>
Date: Fri, 6 Aug 2021 19:25:58 +0200
Subject: [PATCH] feat(core): add heartbeat handling

---
 espanso/src/cli/worker/engine/funnel/ui.rs | 1 +
 espanso/src/cli/worker/mod.rs              | 1 +
 espanso/src/engine/event/mod.rs            | 1 +
 3 files changed, 3 insertions(+)

diff --git a/espanso/src/cli/worker/engine/funnel/ui.rs b/espanso/src/cli/worker/engine/funnel/ui.rs
index 1ebd3a1..4601f70 100644
--- a/espanso/src/cli/worker/engine/funnel/ui.rs
+++ b/espanso/src/cli/worker/engine/funnel/ui.rs
@@ -58,6 +58,7 @@ impl<'a> funnel::Source<'a> for UISource<'a> {
         UIEvent::ContextMenuClick(context_item_id) => {
           EventType::ContextMenuClicked(ContextMenuClickedEvent { context_item_id })
         }
+        UIEvent::Heartbeat => EventType::Heartbeat,
       },
     }
   }
diff --git a/espanso/src/cli/worker/mod.rs b/espanso/src/cli/worker/mod.rs
index efee5d9..0420479 100644
--- a/espanso/src/cli/worker/mod.rs
+++ b/espanso/src/cli/worker/mod.rs
@@ -157,6 +157,7 @@ fn worker_main(args: CliModuleArgs) -> i32 {
     .run(Box::new(move |event| {
       if let Err(error) = engine_ui_event_sender.send(event) {
         error!("unable to send UIEvent to engine: {}", error);
+        panic!("broken UI->Engine channel");
       }
     }))
     .expect("unable to run main eventloop");
diff --git a/espanso/src/engine/event/mod.rs b/espanso/src/engine/event/mod.rs
index 36827cf..8b980a2 100644
--- a/espanso/src/engine/event/mod.rs
+++ b/espanso/src/engine/event/mod.rs
@@ -50,6 +50,7 @@ pub enum EventType {
   ProcessingError(String),
   ExitRequested(ExitMode),
   Exit(ExitMode),
+  Heartbeat,
   
   // Inputs
   Keyboard(input::KeyboardEvent),