Fix CLI exit process.

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## 0.2.4
+
+* Fix CLI exit.
+
 ## 0.2.3
 
 * Fix the CLI connection process.

package.json

@@ -1,6 +1,6 @@
 {
   "name": "xmppjs-chat-bot",
-  "version": "0.2.3",
+  "version": "0.2.4",
   "description": "Server-side XMPP chat bot",
   "engines": {
     "node": ">= 14.4.0"

src/cli/cli.ts

@@ -33,13 +33,32 @@ runCommand.action(async (options) => {
     process.on(sig, () => {
       console.info('Receiving signal: ' + sig)
       console.info('Shutdown...')
+      const promises: Array<Promise<any>> = []
       bots.forEach(bot => {
         console.info('Stopping the bot ' + bot.botName + '...')
-        bot.disconnect().then(() => {}, (err) => {
-          console.error(`Error when stopping the bot ${bot.botName}: ${err as string}`)
+        const p = new Promise<void>(resolve => {
+          bot.disconnect().then(() => {}, (err) => {
+            console.error(`Error when stopping the bot ${bot.botName}: ${err as string}`)
+          }).finally(() => resolve())
         })
+        promises.push(p)
       })
       bots.clear()
+
+      console.info('Waiting all bots to disconnect...')
+      Promise.all(promises).then(
+        () => {},
+        () => {}
+      ).finally(() => {
+        console.info('We can now exit.')
+        process.exit()
+      })
+
+      // Just in case, we also set a max timeout
+      setTimeout(() => {
+        console.error('It seems the bots have not disconnected within 1000ms, exiting anyway.')
+        process.exit(1)
+      }, 1000)
     })
   })