Bot::register now also supports passing an Arc directly (#75)

Possseidon · Oct 14, 2023 · bc02e19 · bc02e19
1 parent b592fc1
commit bc02e19
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/tranquil/src/bot.rs b/tranquil/src/bot.rs
@@ -74,6 +74,22 @@ impl Default for Bot {
     }
 }
 
+pub trait IntoArcModule {
+    fn into_arc_module(self) -> Arc<dyn Module>;
+}
+
+impl<T: Module + 'static> IntoArcModule for T {
+    fn into_arc_module(self) -> Arc<dyn Module> {
+        Arc::new(self)
+    }
+}
+
+impl<T: Module + 'static> IntoArcModule for Arc<T> {
+    fn into_arc_module(self) -> Arc<dyn Module> {
+        self
+    }
+}
+
 impl Bot {
     pub fn new() -> Self {
         Default::default()
@@ -87,8 +103,8 @@ impl Bot {
         self
     }
 
-    pub fn register(mut self, module: impl Module + 'static) -> Self {
-        self.modules.push(Arc::new(module));
+    pub fn register(mut self, module: impl IntoArcModule) -> Self {
+        self.modules.push(module.into_arc_module());
         self
     }