1 files changed, 53 insertions, 8 deletions

diff --git a/src/main.rs b/src/main.rs
index 487d095..dad98c6 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -2,8 +2,8 @@ use backup::Backup;
 use clap::Parser;
 use cli::Subcommands;
 use config::Config;
-use error::Error;
-use tracing::{debug, level_filters::LevelFilter};
+use error::{Error, Result};
+use tracing::{debug, error, level_filters::LevelFilter};
 use tracing_subscriber::{fmt, layer::SubscriberExt, util::SubscriberInitExt, EnvFilter};
 
 mod backup;
@@ -13,9 +13,7 @@ mod error;
 mod packages;
 mod pathinfo;
 
-fn main() -> color_eyre::Result<()> {
-    color_eyre::install()?;
-
+fn main() -> Result<()> {
     let file_appender = tracing_appender::rolling::never("./", "arbs.log");
     let (non_blocking, _guard) = tracing_appender::non_blocking(file_appender);
 
@@ -35,15 +33,27 @@ fn main() -> color_eyre::Result<()> {
         )
         .init();
     debug!("logging initialized");
+    
+    match run_cli() {
+        Ok(()) => { println!("OK") ; Ok(())},
+        Err(err) => {
+            error!(?err);
+            error!("{:?}", std::error::Error::source(&err));
+            send_notification("Backup Error", &err.to_string(), Urgency::Critical)?;
+            Err(err)
+        }
+    }
+}
 
+fn run_cli() -> Result<()> {
     let cli = cli::Cli::parse();
 
     let config = Config::load(cli.config)?;
 
     match cli.subcommand {
         Subcommands::GenerateConfig => Config::generate()?,
-        Subcommands::Save { package_manager } => {
-            let backup = Backup::create(&config, package_manager)?;
+        Subcommands::Save => {
+            let backup = Backup::create(&config)?;
             backup.save(&config)?;
         }
         Subcommands::Restore { package_install } => {
@@ -56,7 +66,42 @@ fn main() -> color_eyre::Result<()> {
             }
 
             last_backup.restore(&config)?;
-        }
+        },
     };
     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")]
+    {
+        let Some(mut icon) = dirs::data_dir() else {
+            return Err(Error::NoSysDir);
+        };
+        icon.push(env!("CARGO_PKG_NAME"));
+        icon.push("icon.png");
+
+    notify_rust::Notification::new()
+        .summary(summary)
+        .body(body)
+        .icon(&icon.to_string_lossy())
+        .timeout(0)
+        .urgency(urgency.into())
+        .show()?;
+    }
+    Ok(())
+}