[ADD] Logger middleware

Author: BliTz037

src/app.module.ts

@@ -1,13 +1,20 @@
-import { Module } from '@nestjs/common';
+import { MiddlewareConsumer, Module, NestModule } from '@nestjs/common';
 import { AppController } from './app.controller';
 import { AppService } from './app.service';
 import { MinecraftModule } from './minecraft/minecraft.module';
 import { SourceModule } from './source/source.module';
 import { FivemModule } from './fivem/fivem.module';
+import { LoggerMiddleware } from 'src/middleware/logger.middleware';
 
 @Module({
   imports: [MinecraftModule, SourceModule, FivemModule],
   controllers: [AppController],
   providers: [AppService],
 })
-export class AppModule {}
+export class AppModule implements NestModule {
+  configure(consumer: MiddlewareConsumer) {
+    consumer
+      .apply(LoggerMiddleware)
+      .forRoutes('*');
+  }
+}

src/main.ts

@@ -1,9 +1,10 @@
 import { NestFactory } from '@nestjs/core';
 import { AppModule } from './app.module';
 import { DocumentBuilder, SwaggerModule } from '@nestjs/swagger';
-import { ValidationPipe } from '@nestjs/common';
+import { Logger, ValidationPipe } from '@nestjs/common';
 
 async function bootstrap() {
+  const port: number | string = process.env.PORT || 3000;
   const app = await NestFactory.create(AppModule);
   const swaggerConfig = new DocumentBuilder()
   .setTitle("Game Server Tracker - API")
@@ -16,6 +17,7 @@ async function bootstrap() {
   SwaggerModule.setup('api-docs', app, swaggerDocument);
   app.useGlobalPipes(new ValidationPipe({ disableErrorMessages: false, whitelist: true }))
   app.enableCors();
-  await app.listen(process.env.PORT || 3000);
+  await app.listen(port);
+  Logger.log(`GST Lametric API is running. Listening on port ${port}`, "Bootstrap");
 }
 bootstrap();

src/middleware/logger.middleware.ts

@@ -0,0 +1,18 @@
+import { Injectable, Logger, NestMiddleware } from "@nestjs/common";
+import { Request, Response, NextFunction } from 'express';
+
+@Injectable()
+export class LoggerMiddleware implements NestMiddleware {
+    private logger: Logger = new Logger('HTTP');
+    use(req: Request, res: Response, next: NextFunction) {
+        res.on("finish", () => {
+            if (res.statusCode >= 500)
+                this.logger.error(`HTTP request ${req.method} ${req.url} ${res.statusCode} ${res.statusMessage}`);
+            else if (res.statusCode >= 400)
+                this.logger.warn(`HTTP request ${req.method} ${req.url} ${res.statusCode} ${res.statusMessage}`);
+            else
+                this.logger.log(`HTTP request ${req.method} ${req.url} ${res.statusCode}`);
+        });
+        next();
+    }
+}