reorganize

git-svn-id: svn+ssh://diminuette.aengel.lesbianunix.dev/salaryman/trunk@11 b9215c17-b818-4693-b096-d1e41a411fef

1 files changed, 95 insertions, 0 deletions

diff --git a/src/smd/main.rs b/src/smd/main.rs
new file mode 100644
index 0000000..231701b
--- /dev/null
+++ b/src/smd/main.rs
@@ -0,0 +1,95 @@
+mod endpoints;
+
+use clap::Parser;
+use serde::{Deserialize, Serialize};
+use tokio::fs::read_to_string;
+use salaryman::service::{Service, ServiceConf};
+
+use std::{net::IpAddr, path::PathBuf};
+
+#[derive(Parser, Debug)]
+#[command(version, about, long_about = None)]
+struct Args {
+    #[arg(
+        short,
+        long,
+        value_name = "FILE",
+        help = "config file override",
+        default_value = "salaryman.toml"
+    )]
+    config: PathBuf,
+    #[arg(
+        short,
+        long,
+        value_name = "ADDR",
+        help = "IP address to bind API to",
+        default_value = "127.0.0.1"
+    )]
+    address: IpAddr,
+    #[arg(
+        short,
+        long,
+        value_name = "PORT",
+        help = "TCP Port to bind API to",
+        default_value = "3080"
+    )]
+    port: u16,
+}
+
+#[derive(Serialize, Deserialize, Debug)]
+struct Config {
+    address: Option<IpAddr>,
+    port: Option<u16>,
+    service: Vec<ServiceConf>,
+}
+/*
+impl Config {
+    fn new() -> Self {
+        Self {
+            address: None,
+            port: None,
+            service: Vec::new(),
+        }
+    }
+}
+*/
+
+async fn load_config(file: &PathBuf) -> Result<Config, Box<dyn std::error::Error>> {
+    let s: String = match read_to_string(file).await {
+        Ok(s) => s,
+        Err(_) => return Err(Box::new(std::io::Error::new(std::io::ErrorKind::NotFound, "cannot find config file"))),
+    };
+    match toml::from_str(s.as_str()) {
+        Ok(c) => Ok(c),
+        Err(_) => Err(Box::new(std::io::Error::new(std::io::ErrorKind::Other, "unable to parse config file"))),
+    }
+}
+
+#[tokio::main]
+async fn main() -> Result<(), Box<dyn std::error::Error>> {
+    let args = Args::parse();
+    let conf: Config = load_config(&args.config).await?;
+    let mut services: Vec<Service> = Vec::new();
+    for i in 0..conf.service.len() {
+        services.push(Service::from_conf(&conf.service[i]));
+        if conf.service[i].autostart {
+            services[i].start().await?;
+            services[i].scan_stdout().await?;
+            services[i].scan_stderr().await?;
+        }
+    }
+    tokio::time::sleep(std::time::Duration::from_secs(60)).await;
+    println!("trying to write to stdin!");
+    for i in 0..services.len() {
+        services[i].write_stdin("stop\n".into()).await?;
+    }
+    tokio::time::sleep(std::time::Duration::from_secs(30)).await;
+    for mut service in services {
+        match service.stop().await {
+            Ok(_) => println!("lol it was killed"),
+            Err(_) => println!("it either didn't exist, or failed to kill"),
+        }
+    }
+    Ok(())
+}
+