aboutsummaryrefslogtreecommitdiff
path: root/src/main.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs55
1 files changed, 8 insertions, 47 deletions
diff --git a/src/main.rs b/src/main.rs
index 09e9ecd..4a08ecc 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,58 +1,18 @@
-use std::env;
use std::net::SocketAddrV4;
+//use std::collections::HashMap;
+
+use log::*;
use igd::aio::search_gateway;
use igd::PortMappingProtocol;
-use std::collections::HashMap;
-
-struct DiplonatConfig {
- private_ip: String,
- consul_node_name: String,
- refresh_time: u32,
- expiration_time: u32
-}
-
-fn fetch_configuration() -> Result<DiplonatConfig, String> {
- let env_private_ip = "DIPLONAT_PRIVATE_IP";
- let private_ip = match env::var(env_private_ip) {
- Ok(val) => val,
- Err(e) => return Err(format!("unable to fetch environment variable {}: {}", env_private_ip, e)),
- };
-
- let env_refresh_time = "DIPLONAT_REFRESH_TIME";
- let refresh_time: u32 = match env::var(env_refresh_time) {
- Ok(val) => val.parse().unwrap(),
- Err(e) => return Err(format!("unable to fetch environment variable {}: {}", env_refresh_time,e))
- };
-
- let env_expiration_time = "DIPLONAT_EXPIRATION_TIME";
- let expiration_time: u32 = match env::var(env_expiration_time) {
- Ok(val) => val.parse().unwrap(),
- Err(e) => return Err(format!("unable to fetch environment variable {}: {}", env_expiration_time,e))
- };
-
- let env_consul_node_name = "DIPLONAT_CONSUL_NODE_NAME";
- let consul_node_name = match env::var(env_consul_node_name) {
- Ok(val) => val,
- Err(e) => return Err(format!("unable to fetch environment variable {}: {}", env_consul_node_name,e))
- };
-
- if refresh_time * 2 > expiration_time {
- return Err(format!("Expiration time (currently: {}s) must be twice bigger than refresh time (currently: {}s)", expiration_time, refresh_time))
- }
-
- let config = DiplonatConfig { private_ip: private_ip, refresh_time: refresh_time, expiration_time: expiration_time, consul_node_name: consul_node_name };
- println!("\tConsul node name: {}", config.consul_node_name);
- println!("\tPrivate IP address: {}", config.private_ip);
- println!("\tRefresh time: {} seconds", config.refresh_time);
- println!("\tExpiration time: {} seconds", config.expiration_time);
- return Ok(config);
-}
+mod config;
#[tokio::main]
async fn main() {
- let config = match fetch_configuration() {
+ pretty_env_logger::init();
+
+ let config = match config::load_env() {
Ok(val) => val,
Err(e) => return println!("unable to build configuration: {}", e),
};
@@ -69,6 +29,7 @@ async fn main() {
Ok(g) => g,
Err(err) => return println!("Faild to find IGD: {}", err),
};
+ info!("Gateway: {}", gateway);
let service = format!("{}:{}", config.private_ip, 1234);
let service: SocketAddrV4 = service.parse().expect("Invalid socket address");