2023-06-21 11:47:04 +00:00
|
|
|
#![feature(trait_upcasting)]
|
|
|
|
|
2023-05-11 12:19:36 +00:00
|
|
|
use clap::Parser;
|
|
|
|
use futures::stream::StreamExt;
|
|
|
|
use signal_hook_tokio::Signals;
|
|
|
|
|
2023-05-12 09:17:15 +00:00
|
|
|
use std::net::SocketAddr;
|
2023-06-21 12:02:42 +00:00
|
|
|
use std::path;
|
2023-05-12 09:17:15 +00:00
|
|
|
use std::str::FromStr;
|
2023-06-18 12:28:46 +00:00
|
|
|
|
2023-05-11 12:19:36 +00:00
|
|
|
#[derive(Parser, Debug)]
|
|
|
|
#[command(version)]
|
2023-06-25 11:35:59 +00:00
|
|
|
#[command(about = "A domani to another dimension")]
|
2023-05-11 12:19:36 +00:00
|
|
|
struct Cli {
|
2023-06-13 19:33:43 +00:00
|
|
|
#[arg(
|
|
|
|
long,
|
|
|
|
help = "OFF, ERROR, WARN, INFO, DEBUG, or TRACE",
|
|
|
|
default_value_t = log::LevelFilter::Info,
|
2023-06-25 11:35:59 +00:00
|
|
|
env = "DOMANI_LOG_LEVEL"
|
2023-06-13 19:33:43 +00:00
|
|
|
)]
|
|
|
|
log_level: log::LevelFilter,
|
|
|
|
|
2023-06-25 11:35:59 +00:00
|
|
|
#[arg(long, default_value_t = false, env = "DOMANI_LOG_TIMESTAMP")]
|
2023-06-13 19:33:43 +00:00
|
|
|
log_timestamp: bool,
|
|
|
|
|
2023-06-25 11:35:59 +00:00
|
|
|
#[arg(long, required = true, env = "DOMANI_HTTP_DOMAIN")]
|
|
|
|
http_domain: domani::domain::Name,
|
2023-05-18 20:02:57 +00:00
|
|
|
|
2023-06-25 11:35:59 +00:00
|
|
|
#[arg(long, default_value_t = SocketAddr::from_str("[::]:3030").unwrap(), env = "DOMANI_HTTP_LISTEN_ADDR")]
|
2023-05-12 09:17:15 +00:00
|
|
|
http_listen_addr: SocketAddr,
|
|
|
|
|
2023-05-18 20:02:57 +00:00
|
|
|
#[arg(
|
|
|
|
long,
|
|
|
|
help = "E.g. '[::]:443', if given then SSL certs will automatically be retrieved for all domains using LetsEncrypt",
|
2023-06-25 11:35:59 +00:00
|
|
|
env = "DOMANI_HTTPS_LISTEN_ADDR",
|
2023-05-20 12:51:36 +00:00
|
|
|
requires = "domain_acme_contact_email",
|
|
|
|
requires = "domain_acme_store_dir_path"
|
2023-05-18 20:02:57 +00:00
|
|
|
)]
|
|
|
|
https_listen_addr: Option<SocketAddr>,
|
2023-05-15 18:25:07 +00:00
|
|
|
|
2023-06-25 11:35:59 +00:00
|
|
|
#[arg(long, required = true, env = "DOMANI_PASSPHRASE")]
|
2023-05-12 16:17:23 +00:00
|
|
|
passphrase: String,
|
|
|
|
|
2023-06-25 11:35:59 +00:00
|
|
|
#[arg(long, required = true, env = "DOMANI_ORIGIN_STORE_GIT_DIR_PATH")]
|
2023-05-12 13:19:24 +00:00
|
|
|
origin_store_git_dir_path: path::PathBuf,
|
2023-05-11 12:19:36 +00:00
|
|
|
|
2023-06-25 11:35:59 +00:00
|
|
|
#[arg(long, required = true, env = "DOMANI_DOMAIN_CHECKER_TARGET_A")]
|
2023-05-15 20:16:29 +00:00
|
|
|
domain_checker_target_a: std::net::Ipv4Addr,
|
2023-05-11 12:19:36 +00:00
|
|
|
|
2023-06-25 11:35:59 +00:00
|
|
|
#[arg(long, default_value_t = String::from("1.1.1.1:53"), env = "DOMANI_DOMAIN_CHECKER_RESOLVER_ADDR")]
|
2023-05-11 12:19:36 +00:00
|
|
|
domain_checker_resolver_addr: String,
|
|
|
|
|
2023-06-25 11:35:59 +00:00
|
|
|
#[arg(long, required = true, env = "DOMANI_DOMAIN_CONFIG_STORE_DIR_PATH")]
|
2023-05-12 13:19:24 +00:00
|
|
|
domain_config_store_dir_path: path::PathBuf,
|
2023-05-18 20:02:57 +00:00
|
|
|
|
2023-06-25 11:35:59 +00:00
|
|
|
#[arg(long, env = "DOMANI_DOMAIN_ACME_STORE_DIR_PATH")]
|
2023-05-20 12:51:36 +00:00
|
|
|
domain_acme_store_dir_path: Option<path::PathBuf>,
|
2023-05-18 20:02:57 +00:00
|
|
|
|
2023-06-25 11:35:59 +00:00
|
|
|
#[arg(long, env = "DOMANI_DOMAIN_ACME_CONTACT_EMAIL")]
|
2023-05-19 10:09:41 +00:00
|
|
|
domain_acme_contact_email: Option<String>,
|
2023-05-11 12:19:36 +00:00
|
|
|
}
|
|
|
|
|
2023-05-20 12:28:02 +00:00
|
|
|
#[tokio::main]
|
|
|
|
async fn main() {
|
2023-05-11 12:19:36 +00:00
|
|
|
let config = Cli::parse();
|
|
|
|
|
2023-06-13 19:33:43 +00:00
|
|
|
env_logger::Builder::new()
|
|
|
|
.filter_level(config.log_level)
|
|
|
|
.format_timestamp(
|
|
|
|
config
|
|
|
|
.log_timestamp
|
|
|
|
.then_some(env_logger::TimestampPrecision::Micros),
|
|
|
|
)
|
|
|
|
.init();
|
|
|
|
|
2023-06-29 14:54:55 +00:00
|
|
|
let origin_store = domani::origin::store::git::FSStore::new(config.origin_store_git_dir_path)
|
2023-06-14 18:22:10 +00:00
|
|
|
.expect("git origin store initialization failed");
|
2023-05-11 12:19:36 +00:00
|
|
|
|
2023-06-29 14:54:55 +00:00
|
|
|
let domain_checker = domani::domain::checker::DNSChecker::new(
|
2023-05-17 12:37:23 +00:00
|
|
|
config.domain_checker_target_a,
|
|
|
|
&config.domain_checker_resolver_addr,
|
|
|
|
)
|
2023-05-20 12:28:02 +00:00
|
|
|
.await
|
2023-06-14 18:22:10 +00:00
|
|
|
.expect("domain checker initialization failed");
|
2023-05-17 12:37:23 +00:00
|
|
|
|
2023-06-29 14:54:55 +00:00
|
|
|
let domain_config_store =
|
|
|
|
domani::domain::config::FSStore::new(&config.domain_config_store_dir_path)
|
|
|
|
.expect("domain config store initialization failed");
|
2023-05-17 12:37:23 +00:00
|
|
|
|
2023-06-21 12:02:42 +00:00
|
|
|
let domain_acme_manager = if config.https_listen_addr.is_some() {
|
2023-05-20 12:51:36 +00:00
|
|
|
let domain_acme_store_dir_path = config.domain_acme_store_dir_path.unwrap();
|
|
|
|
|
2023-06-29 14:54:55 +00:00
|
|
|
let domain_acme_store =
|
|
|
|
domani::domain::acme::store::FSStore::new(&domain_acme_store_dir_path)
|
|
|
|
.expect("domain acme store initialization failed");
|
2023-05-18 20:02:57 +00:00
|
|
|
|
2023-05-19 10:09:41 +00:00
|
|
|
// if https_listen_addr is set then domain_acme_contact_email is required, see the Cli/clap
|
|
|
|
// settings.
|
|
|
|
let domain_acme_contact_email = config.domain_acme_contact_email.unwrap();
|
|
|
|
|
2023-06-21 12:02:42 +00:00
|
|
|
Some(
|
2023-06-29 14:54:55 +00:00
|
|
|
domani::domain::acme::manager::ManagerImpl::new(
|
|
|
|
domain_acme_store,
|
|
|
|
&domain_acme_contact_email,
|
|
|
|
)
|
|
|
|
.await
|
|
|
|
.expect("domain acme manager initialization failed"),
|
2023-06-21 12:02:42 +00:00
|
|
|
)
|
2023-05-20 12:28:02 +00:00
|
|
|
} else {
|
2023-05-20 12:51:36 +00:00
|
|
|
None
|
2023-05-20 12:28:02 +00:00
|
|
|
};
|
2023-05-18 20:02:57 +00:00
|
|
|
|
2023-06-25 11:35:59 +00:00
|
|
|
let mut task_stack = domani::util::TaskStack::new();
|
2023-06-21 11:15:42 +00:00
|
|
|
|
2023-07-03 11:39:44 +00:00
|
|
|
let domain_manager = domani::domain::manager::ManagerImpl::new(
|
2023-06-21 11:15:42 +00:00
|
|
|
&mut task_stack,
|
2023-05-18 20:02:57 +00:00
|
|
|
origin_store,
|
|
|
|
domain_config_store,
|
|
|
|
domain_checker,
|
2023-06-21 12:02:42 +00:00
|
|
|
domain_acme_manager,
|
2023-05-18 20:02:57 +00:00
|
|
|
);
|
2023-05-17 12:37:23 +00:00
|
|
|
|
2023-06-25 11:35:59 +00:00
|
|
|
let _ = domani::service::http::new(
|
2023-06-21 11:15:42 +00:00
|
|
|
&mut task_stack,
|
|
|
|
domain_manager.clone(),
|
|
|
|
config.domain_checker_target_a,
|
|
|
|
config.passphrase,
|
|
|
|
config.http_listen_addr.clone(),
|
|
|
|
config.http_domain.clone(),
|
2023-06-21 12:02:42 +00:00
|
|
|
config
|
|
|
|
.https_listen_addr
|
2023-06-25 11:35:59 +00:00
|
|
|
.map(|listen_addr| domani::service::http::HTTPSParams {
|
2023-06-21 12:02:42 +00:00
|
|
|
listen_addr,
|
|
|
|
cert_resolver: domain_manager.clone(),
|
|
|
|
}),
|
2023-06-21 11:15:42 +00:00
|
|
|
);
|
2023-06-18 13:57:51 +00:00
|
|
|
|
2023-06-21 11:15:42 +00:00
|
|
|
let mut signals =
|
|
|
|
Signals::new(signal_hook::consts::TERM_SIGNALS).expect("initializing signals failed");
|
2023-06-18 13:57:51 +00:00
|
|
|
|
2023-06-21 11:15:42 +00:00
|
|
|
if (signals.next().await).is_some() {
|
|
|
|
log::info!("Gracefully shutting down...");
|
2023-05-18 20:02:57 +00:00
|
|
|
}
|
|
|
|
|
2023-06-21 11:15:42 +00:00
|
|
|
tokio::spawn(async move {
|
|
|
|
if (signals.next().await).is_some() {
|
|
|
|
log::warn!("Forcefully shutting down");
|
|
|
|
std::process::exit(1);
|
|
|
|
};
|
|
|
|
});
|
|
|
|
|
|
|
|
task_stack
|
2023-06-18 12:28:46 +00:00
|
|
|
.stop()
|
|
|
|
.await
|
2023-06-21 11:15:42 +00:00
|
|
|
.expect("failed to stop all background tasks");
|
2023-06-18 12:28:46 +00:00
|
|
|
|
2023-06-13 19:33:43 +00:00
|
|
|
log::info!("Graceful shutdown complete");
|
2023-05-11 12:19:36 +00:00
|
|
|
}
|