1 files changed, 16 insertions, 2 deletions

diff --git a/src/main.rs b/src/main.rs
index ab23ab7..dad98c6 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -3,7 +3,6 @@ use clap::Parser;
 use cli::Subcommands;
 use config::Config;
 use error::{Error, Result};
-use notify_rust::Urgency;
 use tracing::{debug, error, level_filters::LevelFilter};
 use tracing_subscriber::{fmt, layer::SubscriberExt, util::SubscriberInitExt, EnvFilter};
 
@@ -72,6 +71,21 @@ fn run_cli() -> Result<()> {
     Ok(())
 }
 
+enum Urgency {
+    Normal,
+    Critical
+}
+
+#[cfg(feature = "notifications")]
+impl From<Urgency> for notify_rust::Urgency {
+    fn from(value: Urgency) -> Self {
+        match value {
+            Urgency::Normal => Self::Normal,
+            Urgency::Critical => Self::Critical,
+        }
+    }
+}
+
 fn send_notification(summary: &str, body: &str, urgency: Urgency) -> Result<()> {
     #[cfg(feature = "notifications")]
     {
@@ -86,7 +100,7 @@ fn send_notification(summary: &str, body: &str, urgency: Urgency) -> Result<()>
         .body(body)
         .icon(&icon.to_string_lossy())
         .timeout(0)
-        .urgency(urgency)
+        .urgency(urgency.into())
         .show()?;
     }
     Ok(())