Updated how api is structured in codora-framework-bot

Author: westshgit

crates/codora-framework-bot/src/adapter/mod.rs

@@ -2,4 +2,7 @@
 pub mod discord;
 
 #[cfg(feature = "codora-framework-bot-telegram")]
+/// Telegram
+///
+///
 pub mod telegram;

crates/codora-framework-bot/src/bot/context.rs

@@ -0,0 +1,52 @@
+use crate::{bot::Bot, listener::Listener};
+use std::{
+    pin::Pin,
+    task::{Context, Poll},
+};
+
+#[derive(Debug, thiserror::Error)]
+#[error("{0}")]
+pub enum Error {
+    #[error("[Error]: Server error happened")]
+    ServeError,
+}
+
+pub type Result<T, E = Error> = core::result::Result<T, E>;
+
+#[derive(Debug)]
+pub struct BotContext<T> {
+    bot_handler: T,
+    // extension
+}
+
+impl<T> BotContext<T> {
+    pub fn builder(bot_handler: T) -> BotContextBuilder<T> {
+        BotContextBuilder { bot_handler }
+    }
+
+    pub async fn run_with_listener<L>(self, listener: L) -> Result<()>
+    where
+        L: Listener,
+    {
+        todo!()
+    }
+}
+
+#[derive(Debug)]
+pub struct BotContextBuilder<T> {
+    bot_handler: T,
+}
+
+impl<T> BotContextBuilder<T> {
+    pub fn with_graceful_shutdown<F>(self, shutdown_handler: F) -> BotContextBuilder<T>
+    where
+        F: Future<Output = ()> + Send + 'static,
+    {
+        todo!()
+    }
+
+    pub fn build(self) -> BotContext<T> {
+        let BotContextBuilder { bot_handler } = self;
+        BotContext { bot_handler }
+    }
+}

crates/codora-framework-bot/src/bot/mod.rs

@@ -1,3 +1,5 @@
+pub mod context;
+
 /// Bot contract
 ///
 ///
@@ -11,11 +13,5 @@ pub trait Bot<Request> {
     fn handle(&self, request: Request) -> impl Future<Output = Result<(), Self::Error>>;
 }
 
-#[derive(Debug, new)]
-pub struct Context<T> {
-    bot_handler: T,
-    // extension
-}
-
 // Request would be Telegram or Discord Updates
 // impl<T> Bot<T> where T: BotHandler {}

crates/codora-framework-bot/src/lib.rs

@@ -18,16 +18,21 @@ pub mod api;
 pub mod bot;
 pub mod handler;
 mod listener;
-mod serve;
-pub use serve::{Error as ServeError, serve};
 
 #[cfg(feature = "codora-framework-bot-telegram")]
 use codora_framework_macro::Command;
 
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::adapter::telegram::{TGBot, TGBotOption, TGWebhook};
+    use crate::{
+        adapter::telegram::{TGBot, TGBotOption, TGWebhook},
+        bot::context::BotContext,
+    };
+
+    async fn handler() {
+        // implement the graceful shutdown here
+    }
 
     #[tokio::test]
     async fn codora_framework_bot_test() -> anyhow::Result<()> {
@@ -40,7 +45,12 @@ mod tests {
         let bot = TGBot::with(TGBotOption {}).command(Command::Start);
 
         // Identical api with axum yes
-        let res = serve(listener, bot).await?;
+        let res = BotContext::builder(bot)
+            .with_graceful_shutdown(handler())
+            .build()
+            // .run()
+            .run_with_listener(listener)
+            .await?;
         Ok(())
     }
 }