From 477f6c5238955e60eefb422d7d7d4fd9a73e65b7 Mon Sep 17 00:00:00 2001
From: Bulat Kurbanov <kurbanovbul@gmail.com>
Date: Sat, 29 Mar 2025 21:20:08 +0100
Subject: [PATCH] Fix

---
 src/bots_manager/axum_server.rs | 21 ++++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

diff --git a/src/bots_manager/axum_server.rs b/src/bots_manager/axum_server.rs
index 9c7bd21..de83f76 100644
--- a/src/bots_manager/axum_server.rs
+++ b/src/bots_manager/axum_server.rs
@@ -36,11 +36,12 @@ pub async fn start_axum_server(stop_signal: Arc<AtomicBool>) {
         let (_, r_tx) = match BOTS_ROUTES.get(&token).await {
             Some(tx) => tx,
             None => {
-                let bot_data = BOTS_DATA.get(&token).await;
-
-                if bot_data.is_none() {
-                    return StatusCode::NOT_FOUND;
-                }
+                let bot_data = match BOTS_DATA.get(&token).await {
+                    Some(v) => v,
+                    None => {
+                        return StatusCode::NOT_FOUND;
+                    }
+                };
 
                 'creator: {
                     let _guard = start_bot_mutex.lock().await;
@@ -49,10 +50,16 @@ pub async fn start_axum_server(stop_signal: Arc<AtomicBool>) {
                         break 'creator;
                     }
 
-                    start_bot(&bot_data.unwrap()).await
+                    start_bot(&bot_data).await
                 }
 
-                BOTS_ROUTES.get(&token).await.unwrap()
+                match BOTS_ROUTES.get(&token).await {
+                    None => {
+                        log::error!("Cannot get a bot with token: {}", token);
+                        return StatusCode::SERVICE_UNAVAILABLE;
+                    }
+                    Some(v) => v,
+                }
             }
         };