From 8d9181dc0e254a0873f5c47a32bffad928e7b4bb Mon Sep 17 00:00:00 2001
From: Bulat Kurbanov <kurbanovbul@gmail.com>
Date: Mon, 8 Jan 2024 19:54:37 +0100
Subject: [PATCH] Get webhook port from env

---
 src/bots_manager/axum_server.rs | 5 +++--
 src/bots_manager/internal.rs    | 8 ++++++--
 src/bots_manager/mod.rs         | 2 --
 src/config.rs                   | 2 ++
 4 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/src/bots_manager/axum_server.rs b/src/bots_manager/axum_server.rs
index d455e1b..e42735a 100644
--- a/src/bots_manager/axum_server.rs
+++ b/src/bots_manager/axum_server.rs
@@ -24,7 +24,8 @@ use tower_http::trace::{self, TraceLayer};
 use tracing::log;
 use tracing::Level;
 
-use crate::bots_manager::{internal::start_bot, BOTS_DATA, BOTS_ROUTES, SERVER_PORT};
+use crate::bots_manager::{internal::start_bot, BOTS_DATA, BOTS_ROUTES};
+use crate::config;
 
 pub async fn start_axum_server(stop_signal: Arc<AtomicBool>) {
     async fn telegram_request(
@@ -113,7 +114,7 @@ pub async fn start_axum_server(stop_signal: Arc<AtomicBool>) {
     tokio::spawn(async move {
         log::info!("Start webserver...");
 
-        let addr = SocketAddr::from(([0, 0, 0, 0], SERVER_PORT));
+        let addr = SocketAddr::from(([0, 0, 0, 0], config::CONFIG.webhook_port));
 
         axum::Server::bind(&addr)
             .serve(router.into_make_service())
diff --git a/src/bots_manager/internal.rs b/src/bots_manager/internal.rs
index d8b5f36..9e21feb 100644
--- a/src/bots_manager/internal.rs
+++ b/src/bots_manager/internal.rs
@@ -21,7 +21,7 @@ use crate::config;
 
 use super::closable_sender::ClosableSender;
 use super::utils::tuple_first_mut;
-use super::{BotData, SERVER_PORT};
+use super::BotData;
 
 type UpdateSender = mpsc::UnboundedSender<Result<Update, std::convert::Infallible>>;
 
@@ -53,7 +53,11 @@ pub async fn set_webhook(bot_data: &BotData) -> bool {
 
     let token = &bot_data.token;
 
-    let host = format!("{}:{}", &config::CONFIG.webhook_base_url, SERVER_PORT);
+    let host = format!(
+        "{}:{}",
+        &config::CONFIG.webhook_base_url,
+        config::CONFIG.webhook_port
+    );
     let url = Url::parse(&format!("{host}/{token}/"))
         .unwrap_or_else(|_| panic!("Can't parse webhook url!"));
 
diff --git a/src/bots_manager/mod.rs b/src/bots_manager/mod.rs
index a4c7bd6..c68f235 100644
--- a/src/bots_manager/mod.rs
+++ b/src/bots_manager/mod.rs
@@ -46,8 +46,6 @@ pub static CHAT_DONATION_NOTIFICATIONS_CACHE: Lazy<Cache<ChatId, ()>> = Lazy::ne
         .build()
 });
 
-pub static SERVER_PORT: u16 = 8080;
-
 type StopTokenWithSender = (
     StopToken,
     ClosableSender<Result<Update, std::convert::Infallible>>,
diff --git a/src/config.rs b/src/config.rs
index abc0d94..9bad793 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -4,6 +4,7 @@ pub struct Config {
     pub telegram_bot_api: reqwest::Url,
 
     pub webhook_base_url: String,
+    pub webhook_port: u16,
 
     pub admin_id: String,
     pub bot_token: String,
@@ -38,6 +39,7 @@ impl Config {
                 }),
 
             webhook_base_url: get_env("WEBHOOK_BASE_URL"),
+            webhook_port: get_env("WEBHOOK_PORT").parse().unwrap(),
 
             admin_id: get_env("ADMIN_ID"),
             bot_token: get_env("BOT_TOKEN"),