From af85c57c67559de41fa3e02cc61dfebff31e1462 Mon Sep 17 00:00:00 2001
From: Akulij <aakulij@gmail.com>
Date: Sun, 1 Jun 2025 11:38:04 +0500
Subject: [PATCH] change order of drop for BotController

---
 src/bot_manager.rs | 29 +++++++++++++++++------------
 src/main.rs        |  6 ++++++
 2 files changed, 23 insertions(+), 12 deletions(-)

diff --git a/src/bot_manager.rs b/src/bot_manager.rs
index be324ec..829551a 100644
--- a/src/bot_manager.rs
+++ b/src/bot_manager.rs
@@ -118,9 +118,13 @@ where
                             script_handler_gen(bot_runner.controller.clone(), handlers.collect())
                                 .await;
                         Some(
-                            spawn_bot_thread(bot_runner.controller.clone(), db, handler)
-                                .await
-                                .unwrap(),
+                            spawn_bot_thread(
+                                bot_runner.controller.bot.clone(),
+                                bot_runner.controller.db.clone(),
+                                handler,
+                            )
+                            .await
+                            .unwrap(),
                         )
                     }
                 };
@@ -136,12 +140,13 @@ where
         db: &mut DB,
         plug_handlers: Vec<BotHandler>,
     ) -> BotResult<BotInfo> {
-        let mut db = db.clone().with_name(bi.name.clone());
+        let db = db.clone().with_name(bi.name.clone());
         let controller = BotController::with_db(db.clone(), &bi.token, &bi.script).await?;
 
         let handler = script_handler_gen(controller.clone(), plug_handlers).await;
 
-        let thread = spawn_bot_thread(controller.clone(), &mut db, handler).await?;
+        let thread =
+            spawn_bot_thread(controller.bot.clone(), controller.db.clone(), handler).await?;
 
         let info = BotInfo {
             name: bi.name.clone(),
@@ -174,7 +179,7 @@ pub async fn start_bot(
     db: &mut DB,
     plug_handlers: Vec<BotHandler>,
 ) -> BotResult<BotInfo> {
-    let mut db = db.clone().with_name(bi.name.clone());
+    let db = db.clone().with_name(bi.name.clone());
     let controller = BotController::with_db(db.clone(), &bi.token, &bi.script).await?;
 
     let handler = script_handler(controller.rc.clone());
@@ -185,7 +190,7 @@ pub async fn start_bot(
         .chain(std::iter::once(handler))
         .fold(dptree::entry(), |h, plug| h.branch(plug));
 
-    let thread = spawn_bot_thread(controller.clone(), &mut db, handler).await?;
+    let thread = spawn_bot_thread(controller.bot.clone(), controller.db.clone(), handler).await?;
 
     let info = BotInfo {
         name: bi.name.clone(),
@@ -212,11 +217,11 @@ pub async fn start_bot(
 }
 
 pub async fn spawn_bot_thread(
-    bc: BotController,
-    db: &mut DB,
+    bot: Bot,
+    mut db: DB,
     handler: BotHandler,
 ) -> BotResult<JoinHandle<BotResult<()>>> {
-    let state_mgr = MongodbStorage::from_db(db, Json)
+    let state_mgr = MongodbStorage::from_db(&mut db, Json)
         .await
         .map_err(DbError::from)?;
     let thread = std::thread::spawn(move || -> BotResult<()> {
@@ -227,8 +232,8 @@ pub async fn spawn_bot_thread(
             .build()?;
 
         rt.block_on(
-            Dispatcher::builder(bc.bot, handler)
-                .dependencies(dptree::deps![bc.db, state_mgr])
+            Dispatcher::builder(bot, handler)
+                .dependencies(dptree::deps![db, state_mgr])
                 .build()
                 .dispatch(),
         );
diff --git a/src/main.rs b/src/main.rs
index c3a2f91..b95c921 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -108,6 +108,12 @@ pub struct BotController {
     pub runner: Runner,
 }
 
+impl Drop for BotController {
+    fn drop(&mut self) {
+        info!("called drop for BotController");
+    }
+}
+
 unsafe impl Send for BotController {}
 
 const MAIN_BOT_SCRIPT: &str = include_str!(concat!(env!("CARGO_MANIFEST_DIR"), "/mainbot.js"));