feat: improve error handler

src/lib/node-handler.ts

@@ -0,0 +1,12 @@
+import { Logger } from "@nestjs/common";
+
+export function NodeHandler(): void {
+	const logger = new Logger("NodeHandler");
+	process.on("uncaughtException", (error, origin) => {
+		logger.error("Uncaught Exception:", error, "Origin: ", origin);
+	});
+
+	process.on("unhandledRejection", (reason, promise) => {
+		logger.error("Unhandled Rejection at:", promise, "reason:", reason);
+	});
+}

src/lib/sharding.ts

@@ -1,7 +1,8 @@
+import path from "node:path";
+import { Config } from "@/modules/config/types";
 import { Logger } from "@nestjs/common";
 import type { ConfigService } from "@nestjs/config";
 import { ShardingManager as _ShardingManager } from "discord.js";
-import path from "node:path";
 
 export class ShardingManager extends _ShardingManager {
 	public constructor(private readonly config: ConfigService) {
@@ -19,10 +20,6 @@ export class ShardingManager extends _ShardingManager {
 		this.spawn();
 
 		this.on("shardCreate", (shard) => {
-			shard.on("reconnecting", () => {
-				this.logger.warn(`Reconnecting shard: [${shard.id}]`);
-			});
-
 			shard.on("spawn", () => {
 				this.logger.log(`Spawned shard: [${shard.id}]`);
 			});
@@ -31,6 +28,24 @@ export class ShardingManager extends _ShardingManager {
 				this.logger.log(`Shard [${shard.id}] is ready`);
 			});
 
+			shard.on("disconnect", () => {
+				this.logger.error(`Shard [${shard.id}] disconnected`);
+			});
+
+			shard.on("reconnecting", () => {
+				this.logger.warn(`Reconnecting shard: [${shard.id}]`);
+			});
+
+			shard.on("resume", () => {
+				this.logger.log(`Shard [${shard.id}] resumed`);
+			});
+
+			shard.on("message", (message) => {
+				if (this.config.getOrThrow<Config["Debug"]>("Debug").Shard) {
+					this.logger.verbose(message);
+				}
+			});
+
 			shard.on("death", () => {
 				this.logger.fatal(`Died shard: [${shard.id}]`);
 			});

src/main.ts

@@ -3,8 +3,10 @@ import { ConfigService } from "@nestjs/config";
 import { NestFactory } from "@nestjs/core";
 import { AppModule } from "./app.module";
 import { ShardingManager, TopGGAutoPoster } from "./lib";
+import { NodeHandler } from "./lib/node-handler";
 
 async function bootstrap() {
+	NodeHandler();
 	const app = await NestFactory.create(AppModule);
 	const configService = app.get<ConfigService>(ConfigService);
 	const logger = new Logger("Main");

src/modules/config/Config.ts

@@ -36,6 +36,7 @@ export const config = (): Config => ({
 	},
 	Debug: {
 		Client: true,
+		Shard: false,
 		Translations: false,
 		Lavalink: false,
 		PremiumMusicPlayer: false,

src/modules/config/types/index.d.ts

@@ -36,6 +36,7 @@ export interface Config {
 	};
 	Debug: {
 		Client: boolean;
+		Shard: boolean;
 		Translations: boolean;
 		Lavalink: boolean;
 		PremiumMusicPlayer: boolean;