From cfbac90c9cfedc8f894ac507ae03d832170bb5a4 Mon Sep 17 00:00:00 2001
From: Fuwn <contact@fuwn.me>
Date: Wed, 30 Mar 2022 03:26:12 -0700
Subject: feat: custom port

---
 src/main.rs | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'src')

diff --git a/src/main.rs b/src/main.rs
index e88fd9b..9c0a1bc 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -73,6 +73,7 @@ fn hits_from_route(route: &str) -> i32 {
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
   std::env::set_var("RUST_LOG", "windmark,locus=trace");
   pretty_env_logger::init();
+  dotenv::dotenv().ok();
 
   let mut time_mount = Instant::now();
   let mut router = Router::new();
@@ -80,6 +81,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
 
   router.set_private_key_file(".locus/locus_private.pem");
   router.set_certificate_chain_file(".locus/locus_pair.pem");
+  router.set_port(std::env::var("PORT").unwrap().parse().unwrap());
   router.set_pre_route_callback(Box::new(|stream, url, _| {
     info!(
       "accepted connection from {} to {}",
-- 
cgit v1.2.3