Merge pull request #203 from AthennaIO/develop

feat(repl): shutdown providers on .exit

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@athenna/core",
-  "version": "4.17.0",
+  "version": "4.18.0",
   "description": "The plug and play Node.js framework.",
   "license": "MIT",
   "author": "João Lenon <[email protected]>",

src/applications/Repl.ts

@@ -36,12 +36,13 @@ export class Repl {
     Logger.gray("const { User } = await import('#app/models/User')\n")
 
     Logger.write(
-      `${Color.yellow.bold('To see all commands available type:')} .help\n\n`
+      `${Color.yellow.bold('To see all commands available type:')} .help\n`
     )
 
     repl
       .setPrompt(Color.purple.bold('Athenna ') + Color.green.bold('❯ '))
       .displayPrompt(false)
+      .shutdownProviders()
       .commandImpl(Ls)
       .commandImpl(Clean)
 

src/repl/ReplImpl.ts

@@ -7,9 +7,10 @@
  * file that was distributed with this source code.
  */
 
-import { Is, Module, Options, String } from '@athenna/common'
+import { LoadHelper } from '#src/helpers/LoadHelper'
 import type { Command } from '#src/repl/helpers/Command'
 import type { REPLServer, ReplOptions } from 'node:repl'
+import { Is, Module, Options, String } from '@athenna/common'
 import { CommandBuilder } from '#src/repl/helpers/CommandBuilder'
 
 type REPLWriteKey = {
@@ -41,6 +42,18 @@ export class ReplImpl {
     return this.session
   }
 
+  /**
+   * Shutdown all Athenna providers on exit.
+   */
+  public shutdownProviders() {
+    this.session?.on('exit', async () => {
+      await LoadHelper.shutdownProviders()
+      process.exit(0)
+    })
+
+    return this
+  }
+
   /**
    * Define a command in the repl session.
    */

tests/unit/applications/ReplTest.ts

@@ -57,13 +57,14 @@ export default class ReplTest {
     Repl.when('setPrompt').return(Repl)
     Repl.when('displayPrompt').return(Repl)
     Repl.when('commandImpl').return(Repl)
+    Repl.when('shutdownProviders').return(Repl)
 
     await ReplApp.boot()
 
     assert.calledWith(process.stdout.write, chalkRainbow(figlet.textSync('REPL\n')) + '\n')
     assert.calledWith(process.stdout.write, Color.gray('To import your modules use dynamic imports:\n') + '\n')
     assert.calledWith(process.stdout.write, Color.gray("const { User } = await import('#app/models/User')\n") + '\n')
-    assert.calledWith(process.stdout.write, Color.yellow.bold('To see all commands available type:') + ' .help\n\n\n')
+    assert.calledWith(process.stdout.write, Color.yellow.bold('To see all commands available type:') + ' .help\n\n')
   }
 
   @Test()
@@ -74,6 +75,7 @@ export default class ReplTest {
     Repl.when('setPrompt').return(Repl)
     Repl.when('displayPrompt').return(Repl)
     Repl.when('commandImpl').return(Repl)
+    Repl.when('shutdownProviders').return(Repl)
 
     await ReplApp.boot()
 
@@ -89,6 +91,7 @@ export default class ReplTest {
     Repl.when('setPrompt').return(Repl)
     Repl.when('displayPrompt').return(Repl)
     Repl.when('commandImpl').return(Repl)
+    Repl.when('shutdownProviders').return(Repl)
 
     await ReplApp.boot()