feat: mask/unmask

Author: matheus-git

src/domain/service_repository.rs

@@ -13,6 +13,8 @@ pub trait ServiceRepository {
     fn restart_service(&self, name: &str) -> Result<Service, Box<dyn Error>>;
     fn enable_service(&self, name: &str) -> Result<Service, Box<dyn Error>>;
     fn disable_service(&self, name: &str) -> Result<Service, Box<dyn Error>>;
+    fn mask_service(&self, name: &str) -> Result<Service, Box<dyn Error>>;
+    fn unmask_service(&self, name: &str) -> Result<Service, Box<dyn Error>>;
     fn reload_daemon(&self) -> Result<(), Box<dyn std::error::Error>>;
     fn change_connection(&mut self, connection_type: ConnectionType) -> Result<(), zbus::Error>;
     fn systemctl_cat(&self, name: &str) -> Result<String, Box<dyn Error>>;

src/infrastructure/systemd_service_adapter.rs

@@ -194,6 +194,7 @@ impl ServiceRepository for SystemdServiceAdapter {
         let mut service = self.get_unit(name)?;
         while service.state().active().ends_with("ing") {
             service = self.get_unit(name)?;
+            thread::sleep(Duration::from_millis(100));
         }
         Ok(service)
     }
@@ -205,6 +206,7 @@ impl ServiceRepository for SystemdServiceAdapter {
         let mut service = self.get_unit(name)?;
         while service.state().active().ends_with("ing") {
             service = self.get_unit(name)?;
+            thread::sleep(Duration::from_millis(100));
         }
         Ok(service)
     }
@@ -216,6 +218,7 @@ impl ServiceRepository for SystemdServiceAdapter {
         let mut service = self.get_unit(name)?;
         while service.state().active().ends_with("ing") {
             service = self.get_unit(name)?;
+            thread::sleep(Duration::from_millis(100));
         }
         Ok(service)
     }
@@ -236,6 +239,22 @@ impl ServiceRepository for SystemdServiceAdapter {
         self.get_unit(name)
     }
 
+    fn mask_service(&self, name: &str) -> Result<Service, Box<dyn std::error::Error>> {
+        let proxy = self.manager_proxy()?;
+        let _output: Vec<(String, String, String)> =
+        proxy.call("MaskUnitFiles", &(vec![name], false, true))?;
+        thread::sleep(Duration::from_millis(SLEEP_DURATION));
+        self.get_unit(name)
+    }
+
+    fn unmask_service(&self, name: &str) -> Result<Service, Box<dyn std::error::Error>> {
+        let proxy = self.manager_proxy()?;
+        let _output: Vec<(String, String, String)> =
+        proxy.call("UnmaskUnitFiles", &(vec![name], false))?;
+        thread::sleep(Duration::from_millis(SLEEP_DURATION));
+        self.get_unit(name)
+    }
+
     fn reload_daemon(&self) -> Result<(), Box<dyn std::error::Error>> {
         let proxy = self.manager_proxy()?;
         proxy.call::<&str, (), ()>("Reload", &())?;

src/terminal/components/list.rs

@@ -58,7 +58,8 @@ pub enum ServiceAction {
     Enable,
     Disable,
     RefreshAll,
-    ToggleFilter
+    ToggleFilter,
+    ToggleMask
 }
 
 pub struct TableServices {
@@ -258,6 +259,10 @@ impl TableServices {
                 self.sender.send(AppEvent::Action(Actions::GoDetails)).unwrap();
                 return;
             }
+            KeyCode::Char('m') => {
+                self.sender.send(AppEvent::Action(Actions::ServiceAction(ServiceAction::ToggleMask))).unwrap();
+                return;
+            }
             _ => {}
         }
 
@@ -336,6 +341,15 @@ impl TableServices {
             let binding_usecase = self.usecase.clone();
             let usecase = binding_usecase.borrow();
             match action {
+                ServiceAction::ToggleMask => {
+                    match service.state().file() {
+                        "masked" => self.handle_service_result(usecase.unmask_service(service)),
+                        "masked-runtime" => self.handle_service_result(usecase.unmask_service(service)),
+                        _ => self.handle_service_result(usecase.mask_service(service)),
+                    }
+                    self.fetch_services();
+                    self.fetch_and_refresh(self.old_filter_text.clone());
+                },
                 ServiceAction::Start => self.handle_service_result(usecase.start_service(service)),
                 ServiceAction::Stop => self.handle_service_result(usecase.stop_service(service)),
                 ServiceAction::Restart => self.handle_service_result(usecase.restart_service(service)),
@@ -382,7 +396,7 @@ impl TableServices {
             )));
 
             help_text.push(Line::from(
-                "Navigate: ↑/↓ | Switch tab: ←/→ | List all: f | Start: s | Stop: x | Restart: r | Enable: e | Disable: d | Refresh all: u | Log: v | Unit File: c"
+                "Navigate: ↑/↓ | Switch tab: ←/→ | List all: f | Start: s | Stop: x | Restart: r | Enable: e | Disable: d | Mask/Unmask: m | Refresh list: u | Log: v | Unit File: c"
             ));
         }
 

src/usecases/services_manager.rs

@@ -39,6 +39,16 @@ impl ServicesManager {
         self.repository.reload_daemon()?;
         Ok(service)
     }
+    
+    pub fn mask_service(&self, service: &Service) -> Result<Service, Box<dyn Error>> {
+        let service = self.repository.mask_service(service.name())?;
+        Ok(service)
+    }
+
+    pub fn unmask_service(&self, service: &Service) -> Result<Service, Box<dyn Error>> {
+        let service = self.repository.unmask_service(service.name())?;
+        Ok(service)
+    }
 
     pub fn list_services(&self, filter: bool) -> Result<Vec<Service>, Box<dyn Error>> {
         let mut all = Vec::new();