fix: error handling

Author: RozmarinUS

examples/lavalink/src/app.dto.ts renamed to examples/lavalink/src/dtos/query.dto.ts

@@ -1,5 +1,5 @@
 import { SearchPlatform } from 'lavalink-client';
-import { StringOption } from '../../../packages';
+import { StringOption } from '../../../../packages';
 
 export class QueryDto {
   @StringOption({

examples/lavalink/src/source.autocomplete.ts renamed to examples/lavalink/src/interceptor/source-autocomplete.interceptor.ts

@@ -1,7 +1,7 @@
 import { Injectable } from '@nestjs/common';
 import { AutocompleteInteraction } from 'discord.js';
 import { DefaultSources } from 'lavalink-client';
-import { AutocompleteInterceptor } from '../../../packages';
+import { AutocompleteInterceptor } from '../../../../packages';
 
 @Injectable()
 export class SourceAutocompleteInterceptor extends AutocompleteInterceptor {
@@ -12,7 +12,6 @@ export class SourceAutocompleteInterceptor extends AutocompleteInterceptor {
     if (focused.name === 'source') {
       choices = [DefaultSources.soundcloud]; // Note that some Sources needs extra plugins/configuration to property work
     }
-
     return interaction.respond(
       choices
         .filter((choice) => choice.startsWith(focused.value.toString()))

examples/lavalink/src/app.gateway.ts renamed to examples/lavalink/src/levalink-example.gateway.ts

@@ -1,4 +1,4 @@
-import { Injectable, Logger } from '@nestjs/common';
+import { Injectable, Logger, UseInterceptors } from '@nestjs/common';
 import {
   Context,
   OnNodeManager,
@@ -11,42 +11,57 @@ import {
   SlashCommandContext,
   Options,
 } from '../../../packages';
-import { QueryDto } from './app.dto';
+import { QueryDto } from './dtos/query.dto';
+import { SourceAutocompleteInterceptor } from './interceptor/source-autocomplete.interceptor';
 
 @Injectable()
-export class AppGateway {
+export class LevalinkExampleGateway {
   constructor(
     private readonly playerManager: PlayerManager,
     private readonly lavalinkService: NestCordLavalinkService,
   ) {}
 
-  private readonly logger = new Logger(AppGateway.name);
+  private readonly logger = new Logger(LevalinkExampleGateway.name);
 
   @OnNodeManager('connect')
   public onConnect(@Context() [node]: NodeManagerContextOf<'connect'>) {
     this.logger.log(`Node: ${node.options.host} Connected`);
   }
 
+  @OnNodeManager('disconnect')
+  public onNodeDisconnect(@Context() [node]: NodeManagerContextOf<'disconnect'>) {
+    this.logger.log(`Node: ${node.options.host} disconnected`);
+  }
+
   @OnLavalinkManager('playerCreate')
   public onPlayerCreate(@Context() [player]: LavalinkManagerContextOf<'playerCreate'>) {
     this.logger.log(`Player created at ${player.guildId}`);
   }
 
+  @UseInterceptors(SourceAutocompleteInterceptor)
   @SlashCommand({
     name: 'play',
     description: 'play a track',
   })
   async handlePlay(@Context() [interaction]: SlashCommandContext, @Options() { query, source }: QueryDto) {
+    if (!interaction.inCachedGuild()) {
+      return;
+    }
+    if (!interaction.member.voice.channel) {
+      return interaction.reply({
+        content: `You must located in the voice channel.`,
+        ephemeral: true,
+      });
+    }
+
     const player =
       this.playerManager.get(interaction.guild.id) ??
       this.playerManager.create({
         ...this.lavalinkService.extractInfoForPlayer(interaction),
-        // optional configurations:
         selfDeaf: true,
         selfMute: false,
         volume: 50,
       });
-
     await player.connect();
 
     const res = await player.search(
@@ -78,6 +93,11 @@ export class AppGateway {
       });
     }
 
-    player.stopPlaying();
+    await player.stopPlaying();
+    await player.disconnect();
+
+    return interaction.reply({
+      content: 'Player has been stoped',
+    });
   }
 }

examples/lavalink/src/app.module.ts renamed to examples/lavalink/src/levalink-example.module.ts

@@ -1,7 +1,7 @@
 import { GatewayIntentBits, Partials } from 'discord.js';
 import { NestCordModule } from '../../../packages/core';
 import { Module } from '@nestjs/common';
-import { AppGateway } from './app.gateway';
+import { LevalinkExampleGateway } from './levalink-example.gateway';
 import { NestCordLavalinkModule } from '../../../packages';
 
 @Module({
@@ -28,6 +28,6 @@ import { NestCordLavalinkModule } from '../../../packages';
       ],
     }),
   ],
-  providers: [AppGateway],
+  providers: [LevalinkExampleGateway],
 })
-export class AppModule {}
+export class LevalinkExampleModule {}

examples/lavalink/src/main.ts

@@ -1,8 +1,8 @@
 import { NestFactory } from '@nestjs/core';
-import { AppModule } from './app.module';
+import { LevalinkExampleModule } from './levalink-example.module';
 
 async function bootstrap() {
-  const app = await NestFactory.create(AppModule);
+  const app = await NestFactory.create(LevalinkExampleModule);
   await app.listen(2400);
 }
 bootstrap();

packages/lavalink/src/nestcord-lavalink.module.ts

@@ -1,7 +1,7 @@
 import { NestCordLavaLinkConfigurableModule, LAVALINK_MODULE_OPTIONS } from './nestcord-lavalink.module-definition';
 import { Global, Inject, Logger, Module, OnApplicationShutdown, OnModuleInit } from '@nestjs/common';
 import * as ProvidersMap from './providers';
-import { DestroyReasons, LavalinkManager } from 'lavalink-client';
+import { DestroyReasons, LavalinkManager, NodeManager } from 'lavalink-client';
 import { Client } from 'discord.js';
 import { LavalinkListenersModule } from './listeners';
 import { NestCordLavalinkModuleOptions } from './nestcord-lavalink-options.interface';
@@ -24,6 +24,7 @@ export class NestCordLavalinkModule
   public constructor(
     private readonly client: Client,
     private readonly lavalinkManager: LavalinkManager,
+    private readonly nodeManager: NodeManager,
     @Inject(LAVALINK_MODULE_OPTIONS)
     private readonly options: NestCordLavalinkModuleOptions,
   ) {
@@ -42,6 +43,10 @@ export class NestCordLavalinkModule
       this.logger.log('Lavalink Manager Initialized');
     });
 
+    this.nodeManager.on('error', (node, error) => {
+      this.logger.error(`An error occured while connecting to the node: ${node.id}`, error.stack);
+    });
+
     this.client.on('raw', (data) => {
       this.lavalinkManager.sendRawData(data);
     });