🚦 Merge #257: Updated onStop method

xooniverse · Jun 21, 2024 · ea777be · ea777be
2 parents a4e8995 + 3924e05
commit ea777be
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/lib/src/televerse/bot.dart b/lib/src/televerse/bot.dart
@@ -417,8 +417,8 @@ class Bot {
     if (isServerless) return;
     fetcher.onUpdate().listen(
       _onUpdate,
-      onDone: () {
-        _onStop.call();
+      onDone: () async {
+        await _onStop.call();
       },
     );
     try {
@@ -1480,12 +1480,12 @@ class Bot {
   }
 
   /// On Stop Handler
-  void Function() _onStop = () {};
+  FutureOr<void> Function() _onStop = () {};
 
   /// Registers a callback when the the bot is stopped.
   ///
   /// This can be used to clean up resources.
-  void onStop(void Function() callback) {
+  void onStop(FutureOr<void> Function() callback) {
     _onStop = callback;
   }