69 lines
1.6 KiB
Rust
69 lines
1.6 KiB
Rust
use std::{fs, path::Path};
|
|
use serde::Deserialize;
|
|
use toml;
|
|
use tokio;
|
|
use crate::processes;
|
|
|
|
#[derive(Deserialize, Debug)]
|
|
pub struct Service {
|
|
pub name: String,
|
|
pub start: Start,
|
|
pub stop: Option<Stop>,
|
|
}
|
|
|
|
#[derive(Deserialize, Debug)]
|
|
pub struct Start {
|
|
pub command: String,
|
|
pub args: Vec<String>
|
|
}
|
|
|
|
#[derive(Deserialize, Debug)]
|
|
pub struct Stop {
|
|
pub command: String,
|
|
pub args: Vec<String>
|
|
}
|
|
|
|
pub fn get_services() -> Vec<Service> {
|
|
let mut services = Vec::new();
|
|
let mini_dir = Path::new("/etc/mini/");
|
|
|
|
if mini_dir.exists() && mini_dir.is_dir() {
|
|
|
|
let dir = fs::read_dir(mini_dir).unwrap();
|
|
|
|
for file in dir {
|
|
// TODO: check ext
|
|
let svc_file_contents = fs::read_to_string(file.unwrap().path()).unwrap();
|
|
let svc : Service = toml::from_str(&svc_file_contents).unwrap();
|
|
services.push(svc);
|
|
}
|
|
|
|
}
|
|
|
|
services
|
|
}
|
|
|
|
pub async fn launch_services() {
|
|
let services = get_services();
|
|
|
|
for s in services {
|
|
tokio::spawn(async move {
|
|
println!("starting {}", s.name);
|
|
let args: Vec<&str> = s.start.args.iter().map(|e| e.as_str()).collect();
|
|
processes::run(&s.start.command, &args[..]).await;
|
|
});
|
|
}
|
|
}
|
|
|
|
pub fn stop_services() {
|
|
let services = get_services();
|
|
|
|
for s in services {
|
|
println!("stopping {}", s.name);
|
|
if let Some(stop) = s.stop {
|
|
let args: Vec<&str> = stop.args.iter().map(|e| e.as_str()).collect();
|
|
processes::sync_run_wait(&stop.command, &args[..]);
|
|
}
|
|
}
|
|
}
|