OutFleet/src/services/telegram/bot.rs

use teloxide::{Bot, prelude::*};
use tokio::sync::oneshot;

use crate::database::DatabaseManager;
use super::handlers;

/// Run the bot polling loop
pub async fn run_polling(
    bot: Bot,
    db: DatabaseManager,
    mut shutdown_rx: oneshot::Receiver<()>,
) {
    tracing::info!("Starting Telegram bot polling...");

    let handler = Update::filter_message()
        .branch(
            dptree::entry()
                .filter_command::<handlers::Command>()
                .endpoint(handlers::handle_command)
        )
        .branch(
            dptree::endpoint(handlers::handle_message)
        );

    let mut dispatcher = Dispatcher::builder(bot.clone(), handler)
        .dependencies(dptree::deps![db])
        .enable_ctrlc_handler()
        .build();

    // Run dispatcher with shutdown signal
    tokio::select! {
        _ = dispatcher.dispatch() => {
            tracing::info!("Telegram bot polling stopped");
        }
        _ = shutdown_rx => {
            tracing::info!("Telegram bot received shutdown signal");
        }
    }
}