From afc776d2df0c0edff0f77ba7aecd22b8bc032c53 Mon Sep 17 00:00:00 2001
From: Raphael Nestler <raphael.nestler@gmail.com>
Date: Sun, 25 Sep 2022 22:49:06 +0200
Subject: [PATCH] Allow to specify reboot packages on command line

---
 src/main.rs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index c930231..d76179c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -32,8 +32,12 @@ fn parse_xdpyinfo_output(xdpyinfo_output: &str) -> Option<&str> {
 )]
 struct Args {
     /// Disable desktop notification
-    #[structopt(long)]
+    #[clap(long)]
     disable_notification: bool,
+
+    /// Comma separated list of packages were we should reboot after an upgrade.
+    #[clap(long, use_delimiter = true)]
+    reboot_packages: Vec<String>,
 }
 
 fn main() {
@@ -51,7 +55,7 @@ fn main() {
     let session_info = session::SessionInfo::from_utmp();
     if let Ok(session_info) = session_info {
         let critical_packages_checker =
-            CriticalPackagesCheck::new(vec!["systemd".to_owned()], session_info, db);
+            CriticalPackagesCheck::new(args.reboot_packages, session_info, db);
         checkers.push(Box::new(critical_packages_checker));
     }