neos implementation

Author: purerosefallen

Diff for: Replay.js

@@ -196,16 +196,22 @@ export class YGOProMessagesHelper {
 		return sendBuffer;
 	}
 
-	sendMessage(socket: net.Socket, protostr: string, info?: string | Buffer | any) {
-		const sendBuffer = this.prepareMessage(protostr, info);
-		socket.write(sendBuffer);
+	send(socket: net.Socket | WebSocket, buffer: Buffer) {
+		return new Promise<Error | undefined>(done => {
+			if (socket['isWs']) {
+				const ws = socket as WebSocket;
+				// @ts-ignore
+				ws.send(buffer, {}, done);
+			} else {
+				const sock = socket as net.Socket;
+				sock.write(buffer, done);
+			}
+		})
 	}
 
-	sendMessageAsync(socket: net.Socket, protostr: string, info?: string | Buffer | any): Promise<Error> {
+	sendMessage(socket: net.Socket | WebSocket, protostr: string, info?: string | Buffer | any): Promise<Error> {
 		const sendBuffer = this.prepareMessage(protostr, info);
-		return new Promise(done => {
-			socket.write(sendBuffer, done);
-		});
+		return this.send(socket, sendBuffer);
 	}
 
 	addHandler(protostr: string, handler: (buffer: Buffer, info: any, datas: Buffer[], params: any) => Promise<boolean | string>, synchronous: boolean, priority: number) {

Diff for: YGOProMessages.js

@@ -196,6 +196,15 @@
       "banCount": 0,
       "ttl": 600
     },
+    "neos": {
+      "enabled": false,
+      "port": 7977,
+      "trusted_proxies": [
+        "127.0.0.1/8",
+        "::1/128"
+      ],
+      "ip_header": "x-forwarded-for"
+    },
     "test_mode": {
       "watch_public_hand": false,
       "no_connect_count_limit": false,

Diff for: YGOProMessages.ts

@@ -17,6 +17,7 @@
     "deepmerge": "^4.2.2",
     "formidable": "^1.2.6",
     "geoip-country-lite": "^1.0.0",
+    "ip6addr": "^0.2.5",
     "jszip": "^3.5.0",
     "load-json-file": "^6.2.0",
     "lzma": "^2.3.2",
@@ -46,6 +47,7 @@
   },
   "devDependencies": {
     "@types/bunyan": "^1.8.8",
+    "@types/ip6addr": "^0.2.3",
     "@types/node": "^17.0.19",
     "@types/underscore": "^1.11.4",
     "@types/ws": "^8.5.3",