New Logging (#53)

Author: ImJustNon

src/app.module.ts

@@ -5,6 +5,7 @@ import { ServeStaticModule } from "@nestjs/serve-static";
 import { AuthModule as BetterAuthModule } from "@thallesp/nestjs-better-auth";
 import { AppController } from "./app.controller";
 import { AppService } from "./app.service";
+import { logger } from "./common/middlewares/Logger.middleware";
 import { config } from "./config/app.config";
 import { AuthModule } from "./core/auth/auth.module";
 import { EmailModule } from "./core/email/email.module";
@@ -86,13 +87,7 @@ import { UtilModule } from "./modules/util/util.module";
 	providers: [AppService, PrismaService, LoggerService],
 })
 export class AppModule {
-	// configure(consumer: MiddlewareConsumer) {
-	// 	consumer
-	// 		.apply(StudentAuthMiddleware)
-	// 		.exclude(
-	// 			{ path: "/student/user/auth", method: RequestMethod.GET },
-	// 			{ path: "/student/user/auth/callback", method: RequestMethod.GET },
-	// 		)
-	// 		.forRoutes("*");
-	// }
+	configure(consumer: MiddlewareConsumer) {
+		consumer.apply(logger).forRoutes("*");
+	}
 }

src/common/middlewares/Logger.middleware.ts

@@ -0,0 +1,64 @@
+import { NextFunction, Request, Response } from "express";
+import { DiscordWebhook } from "src/utils/discord-webhook";
+
+const discordWebhook = new DiscordWebhook();
+
+export function logger(req: Request, res: Response, next: NextFunction) {
+	const startTime = Date.now();
+	let responseBody: any = null;
+
+	// Capture original methods
+	const originalJson = res.json.bind(res);
+	const originalSend = res.send.bind(res);
+	const originalEnd = res.end;
+
+	// Override json to capture response body
+	res.json = (body: any) => {
+		responseBody = body;
+		return originalJson(body);
+	};
+
+	// Override send to capture response body
+	res.send = (body: any) => {
+		if (typeof body === "string") {
+			try {
+				responseBody = JSON.parse(body);
+			} catch {
+				responseBody = body;
+			}
+		} else {
+			responseBody = body;
+		}
+		return originalSend(body);
+	};
+
+	// Override end to send log
+	res.end = function (chunk?: any, encoding?: any, callback?: any) {
+		const responseTime = Date.now() - startTime;
+
+		// Get user session (attached by better-auth middleware)
+		const userSession = (req as any).session || null;
+
+		// Send to Discord webhook
+		discordWebhook.send(
+			discordWebhook.requestEmbed({
+				method: req.method,
+				url: req.originalUrl || req.url,
+				ip: req.ip || req.headers["x-forwarded-for"]?.toString() || "Unknown",
+				userAgent: req.headers["user-agent"] || "Unknown",
+				body: req.body,
+				query: req.query,
+				headers: req.headers,
+				userSession,
+				statusCode: res.statusCode,
+				responseTime,
+				responseBody,
+			}),
+		);
+
+		// Call original end
+		return originalEnd.call(this, chunk, encoding, callback);
+	} as typeof res.end;
+
+	next();
+}

src/utils/discord-webhook.ts

@@ -59,4 +59,55 @@ export class DiscordWebhook {
 			},
 		};
 	}
+
+	requestEmbed(data: {
+		method: string;
+		url: string;
+		ip: string;
+		userAgent: string;
+		body?: any;
+		query?: any;
+		headers?: Record<string, string | string[] | undefined>;
+		userSession?: any;
+		statusCode: number;
+		responseTime: number;
+		responseBody?: any;
+	}): EmbedMessage {
+		const statusColor = data.statusCode >= 400 ? 16711680 : data.statusCode >= 300 ? 16776960 : 65280;
+		const bodyStr = data.body && Object.keys(data.body).length > 0 ? JSON.stringify(data.body, null, 2).slice(0, 500) : "No body";
+		const queryStr = data.query && Object.keys(data.query).length > 0 ? JSON.stringify(data.query, null, 2).slice(0, 300) : "No query";
+		const responseStr = data.responseBody ? JSON.stringify(data.responseBody, null, 2).slice(0, 800) : "No response body";
+
+		// Filter sensitive headers
+		const safeHeaders = data.headers
+			? Object.fromEntries(
+					Object.entries(data.headers)
+						.filter(([key]) => !["authorization", "cookie", "set-cookie"].includes(key.toLowerCase()))
+						.slice(0, 10),
+				)
+			: {};
+		const headersStr = Object.keys(safeHeaders).length > 0 ? JSON.stringify(safeHeaders, null, 2).slice(0, 400) : "No headers";
+
+		// User session as JSON
+		const sessionStr = data.userSession ? JSON.stringify(data.userSession, null, 2).slice(0, 800) : "Not authenticated";
+
+		return {
+			title: `${data.method} ${data.url}`,
+			description: [
+				`**Status:** ${data.statusCode}`,
+				`**Response Time:** ${data.responseTime}ms`,
+				`**IP:** ${data.ip}`,
+				`**User-Agent:** ${data.userAgent?.slice(0, 100) || "Unknown"}`,
+				`**User Session:** \`\`\`json\n${sessionStr}\`\`\``,
+				`**Headers:** \`\`\`json\n${headersStr}\`\`\``,
+				`**Query:** \`\`\`json\n${queryStr}\`\`\``,
+				`**Request Body:** \`\`\`json\n${bodyStr}\`\`\``,
+				`**Response Body:** \`\`\`json\n${responseStr}\`\`\``,
+			].join("\n"),
+			color: statusColor,
+			footer: {
+				text: "Backend Service | ComCamp 37",
+			},
+		};
+	}
 }