Fix set webhook bug

src/bots_manager/axum_server.rs

@@ -48,9 +48,7 @@ pub async fn start_axum_server(stop_signal: Arc<AtomicBool>) {
                         break 'creator;
                     }
 
-                    if !start_bot(&bot_data.unwrap(), SERVER_PORT).await {
-                        return StatusCode::SERVICE_UNAVAILABLE;
-                    }
+                    start_bot(&bot_data.unwrap()).await
                 }
 
                 BOTS_ROUTES.get(&token).await.unwrap()

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;
+use super::{BotData, SERVER_PORT};
 
 type UpdateSender = mpsc::UnboundedSender<Result<Update, std::convert::Infallible>>;
 
@@ -46,7 +46,23 @@ pub fn get_listener() -> (
     (stop_token, stop_flag, tx, listener)
 }
 
-pub async fn start_bot(bot_data: &BotData, port: u16) -> bool {
+pub async fn set_webhook(bot_data: &BotData) -> bool {
+    let bot = Bot::new(bot_data.token.clone());
+
+    let token = &bot_data.token;
+
+    let host = format!("{}:{}", &config::CONFIG.webhook_base_url, SERVER_PORT);
+    let url = Url::parse(&format!("{host}/{token}/"))
+        .unwrap_or_else(|_| panic!("Can't parse webhook url!"));
+
+    if bot.set_webhook(url.clone()).await.is_err() {
+        return false;
+    }
+
+    true
+}
+
+pub async fn start_bot(bot_data: &BotData) {
     let bot = Bot::new(bot_data.token.clone())
         .set_api_url(config::CONFIG.telegram_bot_api.clone())
         .throttle(Limits::default())
@@ -73,14 +89,6 @@ pub async fn start_bot(bot_data: &BotData, port: u16) -> bool {
 
     let (stop_token, _stop_flag, tx, listener) = get_listener();
 
-    let host = format!("{}:{}", &config::CONFIG.webhook_base_url, port);
-    let url = Url::parse(&format!("{host}/{token}/"))
-        .unwrap_or_else(|_| panic!("Can't parse webhook url!"));
-
-    if bot.set_webhook(url.clone()).await.is_err() {
-        return false;
-    }
-
     tokio::spawn(async move {
         dispatcher
             .dispatch_with_listener(
@@ -93,6 +101,4 @@ pub async fn start_bot(bot_data: &BotData, port: u16) -> bool {
     BOTS_ROUTES
         .insert(token.to_string(), (stop_token, ClosableSender::new(tx)))
         .await;
-
-    true
 }

src/bots_manager/mod.rs

@@ -23,6 +23,7 @@ use self::axum_server::start_axum_server;
 use self::bot_manager_client::get_bots;
 pub use self::bot_manager_client::{BotCache, BotData};
 use self::closable_sender::ClosableSender;
+use self::internal::set_webhook;
 
 pub static USER_ACTIVITY_CACHE: Lazy<Cache<UserId, ()>> = Lazy::new(|| {
     Cache::builder()
@@ -76,9 +77,15 @@ impl BotsManager {
         match bots_data {
             Ok(v) => {
                 for bot_data in v.iter() {
+                    if BOTS_DATA.contains_key(&bot_data.token) {
+                        continue;
+                    }
+
                     BOTS_DATA
                         .insert(bot_data.token.clone(), bot_data.clone())
                         .await;
+
+                    set_webhook(bot_data).await;
                 }
             }
             Err(err) => {