Log IP addresses of gateway connections

1 files changed, 5 insertions, 1 deletions

diff --git a/gateway/src/events/Connection.ts b/gateway/src/events/Connection.ts

index 4954cd08..2d41b7a3 100644
--- a/gateway/src/events/Connection.ts
+++ b/gateway/src/events/Connection.ts
@@ -8,6 +8,7 @@ import { Close } from "./Close";
 import { Message } from "./Message";
 import { createDeflate } from "zlib";
 import { URL } from "url";
+import { Config } from "@fosscord/util";
 var erlpack: any;
 try {
 	erlpack = require("@yukikaze-bot/erlpack");
@@ -22,12 +23,15 @@ export async function Connection(
 	socket: WebSocket,
 	request: IncomingMessage
 ) {
+	const forwardedFor = Config.get().security.forwadedFor;
+	const ipAddress = forwardedFor ? request.headers[forwardedFor] as string : request.socket.remoteAddress;
+
 	try {
 		// @ts-ignore
 		socket.on("close", Close);
 		// @ts-ignore
 		socket.on("message", Message);
-		console.log(`[Gateway] Connections: ${this.clients.size}`);
+		console.log(`[Gateway] New connection from ${ipAddress}, total ${this.clients.size}`);
 
 		const { searchParams } = new URL(`http://localhost${request.url}`);
 		// @ts-ignore