payjoin / rust-payjoin / 14180251374

async fn main() -> Result<(), Box<dyn std::error::Error>> {

    init_logging();

    let key_dir = std::env::var("PJ_OHTTP_KEY_DIR")

        .map(std::path::PathBuf::from)

        .unwrap_or_else(|_| std::path::PathBuf::from(DEFAULT_KEY_CONFIG_DIR));

    // Check for generate-key command

    if let Some(arg) = std::env::args().nth(1) {

        if arg == "generate-key" {

            let key_id = std::env::args().nth(2).and_then(|s| s.parse::<u8>().ok()).unwrap_or(1);

            let ohttp_config = key_config::gen_ohttp_server_config()?;

            let path = key_config::persist_new_key_config(ohttp_config, &key_dir)?;

            println!("Generated new key configuration with ID {} at {}", key_id, path.display());

            return Ok(());

        }

    }

    let dir_port =

        env::var("PJ_DIR_PORT").map_or(DEFAULT_DIR_PORT, |s| s.parse().expect("Invalid port"));

    let timeout_env = env::var("PJ_DIR_TIMEOUT_SECS")

        .map_or(DEFAULT_TIMEOUT_SECS, |s| s.parse().expect("Invalid timeout"));

    let timeout = std::time::Duration::from_secs(timeout_env);

    let db_host = env::var("PJ_DB_HOST").unwrap_or_else(|_| DEFAULT_DB_HOST.to_string());

    let ohttp = key_config::read_server_config(&key_dir)?;

    payjoin_directory::listen_tcp(dir_port, db_host, timeout, ohttp.into()).await

}

fn init_logging() {

    let env_filter =

        EnvFilter::builder().with_default_directive(LevelFilter::INFO.into()).from_env_lossy();

    tracing_subscriber::fmt().with_target(true).with_level(true).with_env_filter(env_filter).init();

    println!("Logging initialized");

}