system-hook 0.2.2

shook: webhook server to automatically update production servers
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

mod config;
mod daemon;
mod error;
mod init;
mod server;

use std::fs::OpenOptions;

use config::{Action, AppArgs};
use std::fs::File;
use tracing_subscriber::prelude::*;

#[cfg(not(unix))]
compile_error!("this program only runs on linux");

#[tokio::main]
async fn main() -> color_eyre::Result<()> {
    color_eyre::install()?;
    let _ = dotenvy::dotenv().map_err(|_| {
        tracing::warn!("no '.env' file found");
    });
    let args: AppArgs = argh::from_env();

    let log_writer = match args.log_file {
        Some(path) => OpenOptions::new().create(true).append(true).open(path)?,
        None => File::create("/dev/stdout")?,
    };
    tracing_subscriber::registry()
        .with(args.log_level.map(Into::into).unwrap_or_else(|| {
            tracing_subscriber::EnvFilter::try_from_default_env()
                .unwrap_or_else(|_| "shook=INFO".into())
        }))
        .with(tracing_subscriber::fmt::layer().with_writer(log_writer))
        .init();

    match args.action {
        Action::Init(init) => init::init_project(init),
        Action::Serve(serve) => server::serve(serve).await,
        Action::Daemon(daemon) => daemon::daemon_message(daemon),
        Action::Version(_) => {
            println!("shook version: {}", env!("CARGO_PKG_VERSION"));
            Ok(())
        }
    }?;

    Ok(())
}