feat(runtime): implement runtime trait

crates/aionbot-adapter-onebot/src/lib.rs

@@ -2,6 +2,7 @@ use std::sync::Arc;
 
 use aionbot_core::runtime::{Runtime, StateManager};
 use anyhow::Result;
+use onebot_v11::connect::ws_reverse::ReverseWsConnect;
 
 pub extern crate aionbot_core;
 
@@ -17,11 +18,20 @@ impl Adapter for aionbot_core::event::Event {
     }
 }
 
-#[derive(Default)]
 pub struct OnebotRuntime {
+    connect: Option<Arc<ReverseWsConnect>>,
     state: Arc<StateManager>,
 }
 
+impl Default for OnebotRuntime {
+    fn default() -> Self {
+        Self {
+            connect: None,
+            state: Arc::new(StateManager::default()),
+        }
+    }
+}
+
 impl Runtime for OnebotRuntime {
     fn set_manager(mut self, manager: Arc<StateManager>) -> Self {
         self.state = manager;
@@ -32,6 +42,19 @@ impl Runtime for OnebotRuntime {
         &self.state
     }
 
+    async fn prepare(&mut self) -> Result<()> {
+        self.connect = Some(ReverseWsConnect::new(Default::default()).await?);
+        Ok(())
+    }
+
+    fn setup(&mut self, setup: Box<dyn FnOnce(&Self) + Send + Sync>) {
+        setup(self);
+    }
+
+    async fn finalize(&mut self) -> Result<()> {
+        Ok(())
+    }
+
     async fn run(&self) -> Result<()> {
         Ok(())
     }

crates/aionbot-core/src/runtime.rs

@@ -30,12 +30,17 @@ pub struct Builder<R: Runtime + Default> {
     handler: Arc<Option<Handler>>,
     runtime: R,
     state: Arc<StateManager>,
+    setup: Option<Box<dyn FnOnce(&R) + Send + Sync>>,
 }
 
 impl<R> Builder<R>
 where
     R: Runtime + Default + Send + 'static,
 {
+    pub fn setup(&mut self, setup: Box<dyn FnOnce(&R) + Send + Sync>) {
+        self.setup = Some(setup);
+    }
+
     pub fn invoke_handler(mut self, entries: Vec<Entry>) -> Self {
         self.handler = Arc::new(Some(Handler::new(entries)));
         self
@@ -47,6 +52,12 @@ where
     }
 
     pub async fn run(&mut self) -> Result<()> {
+        println!("Starting bot...");
+        self.runtime.prepare().await?;
+        if let Some(setup) = self.setup.take() {
+            self.runtime.setup(setup);
+        }
+        self.runtime.finalize().await?;
         self.runtime.run().await
     }
 }
@@ -62,6 +73,7 @@ where
             handler: Arc::new(None),
             runtime,
             state: Arc::clone(&manager),
+            setup: None,
         }
     }
 }
@@ -72,9 +84,17 @@ pub trait Runtime {
 
     fn manager(&self) -> &StateManager;
 
+    fn prepare(&mut self) -> impl std::future::Future<Output = Result<()>> + Send {
+        async move { Ok(()) }
+    }
+
     fn setup(&mut self, setup: Box<dyn FnOnce(&Self) + Send + Sync>) {
         setup(self)
     }
 
+    fn finalize(&mut self) -> impl std::future::Future<Output = Result<()>> + Send {
+        async move { Ok(()) }
+    }
+
     fn run(&self) -> impl std::future::Future<Output = Result<()>> + Send;
 }