feat: add hooks before/after connect/disconnect

Author: ricardogobbosouza

src/module.ts

@@ -16,7 +16,7 @@ const DEFAULTS: ModuleOptions = {
 
 const nuxtModule: Module<ModuleOptions> = function (moduleOptions) {
   this.nuxt.hook('builder:extendPlugins', (plugins: NuxtOptionsPlugin[]) => {
-    const options: ModuleOptions = defu(
+    const options = defu(
       this.options.echo || {},
       moduleOptions,
       DEFAULTS

src/runtime/echo.ts

@@ -5,7 +5,7 @@ import { ModuleOptions } from './types'
 
 export class Echo extends BaseEcho {
   ctx: Context;
-  config: Partial<ModuleOptions>;
+  config: ModuleOptions;
 
   constructor (ctx: Context, config: Partial<ModuleOptions> = {}) {
     // when enabled authModule, start broadcaster with null
@@ -58,11 +58,28 @@ export class Echo extends BaseEcho {
         if (this.config.connectOnLogin && loggedIn) {
           // set broadcaster when user logged in
           this.options.broadcaster = this.config.broadcaster
+
+          if (this.config.onBeforeConnect) {
+            await this.config.onBeforeConnect()
+          }
+
           this.connect()
+
+          if (this.config.onAfterConnect) {
+            await this.config.onAfterConnect()
+          }
         }
 
         if (this.config.disconnectOnLogout && !loggedIn && this.connector) {
+          if (this.config.onBeforeDisconnect) {
+            await this.config.onBeforeDisconnect()
+          }
+
           this.disconnect()
+
+          if (this.config.onAfterDisconnect) {
+            await this.config.onAfterDisconnect()
+          }
         }
       }).bind(this)
     }

src/runtime/index.ts

@@ -10,10 +10,12 @@ declare module '@nuxt/types' {
     $echo: Echo;
   }
 
+  // Nuxt 2.14+
   interface NuxtConfig {
     echo?: Partial<ModuleOptions>
   }
 
+  // Nuxt 2.9 - 2.13
   interface Configuration {
     echo?: Partial<ModuleOptions>
   }

src/runtime/types.ts

@@ -8,4 +8,8 @@ export interface ModuleOptions extends EchoOptions {
   connectOnLogin?: boolean,
   disconnectOnLogout?: boolean,
   optionsPath?: string,
+  onBeforeConnect?: Function,
+  onAfterConnect?: Function,
+  onBeforeDisconnect?: Function,
+  onAfterDisconnect?: Function,
 }