userd 0.2.0

A user daemon, managing services and regular running of jobs, in user space.
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

use std::{fs::File, path::Path};

mod activity;
mod conditions;
mod config;
mod prelude;
mod service;
mod task;
mod triggers;
mod utils;

use prelude::*;

struct Daemon {
    tasks: Vec<Task>,
    services: Vec<Service>,
}

impl Daemon {
    fn parse(path: impl AsRef<Path>) -> Result<Daemon> {
        let path = path.as_ref();
        if path.exists() {
            let config: config::Config = serde_yaml_ng::from_reader(File::open(path)?)?;
            let tasks = config
                .tasks
                .into_iter()
                .map(Task::new)
                .collect::<Result<_>>()?;
            let services = config
                .services
                .into_iter()
                .map(Service::new)
                .collect::<Result<_>>()?;
            Ok(Daemon { tasks, services })
        } else {
            Err(anyhow::anyhow!(
                "Missing configuration file at '{:?}'.",
                path
            ))
        }
    }
}

#[tokio::main]
async fn main() {
    env_logger::builder()
        .filter_level(log::LevelFilter::Info)
        .parse_default_env()
        .init();
    let dirs = directories::ProjectDirs::from("com", "cyloncore", "userd").unwrap();
    let config_path = dirs.config_dir().join("config");

    loop {
        let daemon = Daemon::parse(&config_path);
        match daemon {
            Ok(daemon) => {
                log::info!("Success!");
                let tasks: Vec<_> = daemon
                    .tasks
                    .into_iter()
                    .map(|task| tokio::spawn(task.start()))
                    .collect();
                let services: Vec<_> = daemon
                    .services
                    .into_iter()
                    .map(|service| tokio::spawn(service.start()))
                    .collect();
                utils::async_watch(config_path.clone()).await.unwrap();
                for t in tasks {
                    t.abort();
                }
                for s in services {
                    s.abort();
                }
            }
            Err(e) => {
                log::error!("Failure to start daemon: {:?}", e);
                utils::async_watch(config_path.clone()).await.unwrap();
            }
        }
    }
}