From 733d7e2ff4d98643c7fb56dc42e7d0e22bbc80d1 Mon Sep 17 00:00:00 2001
From: Federico Terzi <federicoterzi96@gmail.com>
Date: Fri, 6 Aug 2021 19:56:36 +0200
Subject: [PATCH] feat(ui): implement Heartbeat on Linux

---
 Cargo.lock                  |  1 +
 espanso-ui/Cargo.toml       |  1 +
 espanso-ui/src/linux/mod.rs | 45 ++++++++++++++++++++++++++-----------
 3 files changed, 34 insertions(+), 13 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index 9a1a388..33aec5e 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -759,6 +759,7 @@ version = "0.1.0"
 dependencies = [
  "anyhow",
  "cc",
+ "crossbeam",
  "lazy_static",
  "lazycell",
  "log",
diff --git a/espanso-ui/Cargo.toml b/espanso-ui/Cargo.toml
index da2f630..1194758 100644
--- a/espanso-ui/Cargo.toml
+++ b/espanso-ui/Cargo.toml
@@ -28,6 +28,7 @@ lazycell = "1.3.0"
 
 [target.'cfg(target_os="linux")'.dependencies]
 notify-rust = "4.2.2"
+crossbeam = "0.8.0"
 
 [build-dependencies]
 cc = "1.0.66"
\ No newline at end of file
diff --git a/espanso-ui/src/linux/mod.rs b/espanso-ui/src/linux/mod.rs
index cfdebbe..f51d319 100644
--- a/espanso-ui/src/linux/mod.rs
+++ b/espanso-ui/src/linux/mod.rs
@@ -17,20 +17,24 @@
  * along with espanso.  If not, see <https://www.gnu.org/licenses/>.
  */
 
+use std::time::Duration;
+
 use anyhow::Result;
+use crossbeam::{
+  channel::{unbounded, Receiver, Sender},
+  select,
+};
 use log::error;
 use notify_rust::Notification;
-use std::sync::mpsc;
-use std::sync::mpsc::{Receiver, Sender};
 
-use crate::{UIEventLoop, UIRemote};
+use crate::{event::UIEvent, UIEventLoop, UIRemote};
 
 pub struct LinuxUIOptions {
   pub notification_icon_path: String,
 }
 
 pub fn create(options: LinuxUIOptions) -> (LinuxRemote, LinuxEventLoop) {
-  let (tx, rx) = mpsc::channel();
+  let (tx, rx) = unbounded();
   let remote = LinuxRemote::new(tx, options.notification_icon_path);
   let eventloop = LinuxEventLoop::new(rx);
   (remote, eventloop)
@@ -75,7 +79,9 @@ impl UIRemote for LinuxRemote {
   }
 
   fn exit(&self) {
-    self.stop().expect("unable to send termination signal to ui eventloop");
+    self
+      .stop()
+      .expect("unable to send termination signal to ui eventloop");
   }
 }
 
@@ -95,14 +101,27 @@ impl UIEventLoop for LinuxEventLoop {
     Ok(())
   }
 
-  fn run(&self, _: crate::UIEventCallback) -> Result<()> {
-    // We don't run an event loop on Linux as there is no tray icon or application window needed.
-    // Thad said, we still need a way to block this method, and thus we use a channel
-    if let Err(error) = self.rx.recv() {
-      error!("Unable to block the LinuxEventLoop: {}", error);
-      return Err(error.into());
+  fn run(&self, callback: crate::UIEventCallback) -> Result<()> {
+    loop {
+      select! {
+        recv(self.rx) -> result => {
+          // We don't run an event loop on Linux as there is no tray icon or application window needed.
+          // Thad said, we still need a way to block this method, and thus we use a channel
+          match result {
+            Ok(_) => {
+              // remote.exit() called
+              return Ok(());
+            }
+            Err(error) => {
+              error!("Unable to block the LinuxEventLoop: {}", error);
+              return Err(error.into());
+            }
+          }
+        },
+        default(Duration::from_millis(1000)) => {
+          (*callback)(UIEvent::Heartbeat);
+        }
+      }
     }
-
-    Ok(())
   }
 }