dufs/src/main.rs

mod args;
mod auth;
mod server;

#[macro_use]
extern crate log;

pub type BoxResult<T> = Result<T, Box<dyn std::error::Error>>;

use std::env;
use std::io::Write;

use crate::args::{matches, Args};
use crate::server::serve;

#[tokio::main]
async fn main() {
    run().await.unwrap_or_else(handle_err)
}

async fn run() -> BoxResult<()> {
    if env::var("RUST_LOG").is_err() {
        env::set_var("RUST_LOG", "info")
    }
    env_logger::builder()
        .format(|buf, record| {
            let timestamp = buf.timestamp();
            writeln!(buf, "[{} {}] {}", timestamp, record.level(), record.args())
        })
        .init();

    let args = Args::parse(matches())?;
    tokio::select! {
        ret = serve(args) => {
            ret
        },
        _ = shutdown_signal() => {
            Ok(())
        },
    }
}

fn handle_err<T>(err: Box<dyn std::error::Error>) -> T {
    eprintln!("error: {}", err);
    std::process::exit(1);
}

async fn shutdown_signal() {
    tokio::signal::ctrl_c()
        .await
        .expect("Failed to install CTRL+C signal handler")
}