get on-off back for triple listeners

jehy · Nov 29, 2021 · 6b5f7e1 · 6b5f7e1
1 parent 1728c46
commit 6b5f7e1
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/telegramServer.ts b/src/telegramServer.ts
@@ -231,9 +231,15 @@ export class TelegramServer extends EventEmitter {
 
   async waitUserMessage() {
     return new Promise<void>((resolve) => {
-      this.once('AddedUserMessage', () => resolve());
-      this.once('AddedUserCommand', () => resolve());
-      this.once('AddedUserCallbackQuery', () => resolve());
+      const messageHandler = () => {
+        this.off('AddedUserMessage', messageHandler);
+        this.off('AddedUserCommand', messageHandler);
+        this.off('AddedUserCallbackQuery', messageHandler);
+        resolve();
+      };
+      this.on('AddedUserMessage', messageHandler);
+      this.on('AddedUserCommand', messageHandler);
+      this.on('AddedUserCallbackQuery', messageHandler);
     });
   }