From c4d5507e9c0b16b0d34df9e871be1d76d296ecea Mon Sep 17 00:00:00 2001
From: Keuin <keuinx@gmail.com>
Date: Sun, 27 Mar 2022 20:17:23 +0800
Subject: Configurable HTTP listen endpoint. Handle Ctrl-C correctly.

---
 src/main.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index fffdb65..65e01b2 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,4 +1,5 @@
 use std::convert::Infallible;
+use std::net::SocketAddr;
 use std::sync::Arc;
 
 use log::{debug, error, info, warn};
@@ -51,5 +52,12 @@ async fn main() {
 
     tokio::spawn(bot::repl(bot, Arc::new(db)));
 
-    warp::serve(send_message).run(([127, 0, 0, 1], 3030)).await;
+    let endpoint: SocketAddr = config.listen.parse()
+        .expect("Cannot parse `listen` as endpoint.");
+
+    println!("Listen on {}", endpoint);
+
+    tokio::spawn(warp::serve(send_message).run(endpoint));
+
+    tokio::signal::ctrl_c().await.unwrap();
 }
-- 
cgit v1.2.3