78 lines
1.7 KiB
Rust
78 lines
1.7 KiB
Rust
use anyhow::{anyhow, Result};
|
|
use log::debug;
|
|
use tokio::try_join;
|
|
|
|
use crate::acme_actor::AcmeActor;
|
|
use crate::config::ConfigOpts;
|
|
use crate::consul_actor::ConsulActor;
|
|
use crate::fw_actor::FirewallActor;
|
|
use crate::igd_actor::IgdActor;
|
|
|
|
pub struct Diplonat {
|
|
consul: ConsulActor,
|
|
|
|
acme: Option<AcmeActor>,
|
|
firewall: Option<FirewallActor>,
|
|
igd: Option<IgdActor>,
|
|
}
|
|
|
|
impl Diplonat {
|
|
pub async fn new() -> Result<Self> {
|
|
let config = ConfigOpts::from_env()?;
|
|
debug!("{:#?}", config);
|
|
|
|
let consul_actor = ConsulActor::new(config.consul);
|
|
|
|
let acme_actor = AcmeActor::new(config.acme, &consul_actor.rx_open_ports).await?;
|
|
|
|
let firewall_actor = FirewallActor::new(config.firewall, &consul_actor.rx_open_ports).await?;
|
|
|
|
let igd_actor = IgdActor::new(config.igd, &consul_actor.rx_open_ports).await?;
|
|
|
|
if acme_actor.is_none() && firewall_actor.is_none() && igd_actor.is_none() {
|
|
return Err(anyhow!(
|
|
"At least enable *one* module, otherwise it's boring!"
|
|
));
|
|
}
|
|
|
|
let ctx = Self {
|
|
consul: consul_actor,
|
|
acme: acme_actor,
|
|
firewall: firewall_actor,
|
|
igd: igd_actor,
|
|
};
|
|
|
|
return Ok(ctx);
|
|
}
|
|
|
|
pub async fn listen(&mut self) -> Result<()> {
|
|
let acme = &mut self.acme;
|
|
let firewall = &mut self.firewall;
|
|
let igd = &mut self.igd;
|
|
|
|
try_join!(
|
|
self.consul.listen(),
|
|
async {
|
|
match acme {
|
|
Some(x) => x.listen().await,
|
|
None => Ok(()),
|
|
}
|
|
},
|
|
async {
|
|
match firewall {
|
|
Some(x) => x.listen().await,
|
|
None => Ok(()),
|
|
}
|
|
},
|
|
async {
|
|
match igd {
|
|
Some(x) => x.listen().await,
|
|
None => Ok(()),
|
|
}
|
|
},
|
|
)?;
|
|
|
|
return Ok(());
|
|
}
|
|
}
|