src/service/systemd.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

use std::{process::Command, time::Duration};

use serde::Deserialize;
use tokio::sync::watch::Sender;

use crate::{Error, Status};

use super::ServiceSpawner;

#[derive(Debug, Clone, Deserialize)]
pub struct Systemd {
    pub service: String,
}

impl ServiceSpawner for Systemd {
    async fn spawn(self, tx: Sender<Status>) -> Result<(), Error> {
        let mut command = Command::new("systemctl");
        command.arg("is-active").arg(&self.service);

        let mut interval = tokio::time::interval(Duration::from_secs(5));
        loop {
            interval.tick().await;

            let status = match command.output() {
                Ok(output) if output.status.success() => Status::Pass,
                Ok(output) => {
                    let stdout = String::from_utf8_lossy(&output.stdout).trim().to_string();
                    Status::Fail(Some(format!("Service state: {}", stdout)))
                }
                Err(err) => {
                    tracing::error!("Failed to spawn process: {err}");
                    Status::Unknown
                }
            };

            tx.send_if_modified(|s| s.update(status));
        }
    }
}