1 files changed, 8 insertions, 7 deletions
diff --git a/src/diplonat.rs b/src/diplonat.rs
index 38b2f8a..c209da5 100644
--- a/src/diplonat.rs
+++ b/src/diplonat.rs
@@ -1,18 +1,19 @@
-use igd::Gateway;
 use anyhow::{Result, Context};
-
+use log::*;
 use crate::*;
 
 pub struct DiplonatContext {
   pub config: config::DiplonatConfig,
-  //pub gateway: igd::Gateway
+  pub gateway: igd::aio::Gateway
 }
 
-pub fn setup() -> Result<DiplonatContext> {
-  return Ok(DiplonatContext {
+pub async fn setup() -> Result<DiplonatContext> {
+  let ctx = DiplonatContext {
     config: config::load_env().context("Unable to read configuration from environment")?,
-    //gateway: search_gateway(Default::default()).await
-  });
+    gateway: gw::get_gateway().await? 
+  };
+
+  return Ok(ctx);
 }
 
 pub fn listen() -> bool {