From 9e4a769d6937bf7a4002f118309537f9c820c3f6 Mon Sep 17 00:00:00 2001
From: Rory& <root@rory.gay>
Date: Sun, 6 Jul 2025 11:02:18 +0200
Subject: [PATCH] Fix forgotten function signature changes

---
 .../channels/#channel_id/messages/#message_id/index.ts     | 7 +++++--
 src/api/routes/channels/#channel_id/messages/index.ts      | 7 +++++--
 src/gateway/util/WebSocket.ts                              | 1 -
 3 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/api/routes/channels/#channel_id/messages/#message_id/index.ts b/src/api/routes/channels/#channel_id/messages/#message_id/index.ts
index d91c705d..1d06506f 100644
--- a/src/api/routes/channels/#channel_id/messages/#message_id/index.ts
+++ b/src/api/routes/channels/#channel_id/messages/#message_id/index.ts
@@ -30,7 +30,7 @@ import {
 	emitEvent,
 	getPermission,
 	getRights,
-	uploadFile
+	uploadFile, NewUrlUserSignatureData,
 } from "@spacebar/util";
 import { Request, Response, Router } from "express";
 import { HTTPError } from "lambert-server";
@@ -245,7 +245,10 @@ router.put(
 			console.error("[Message] post-message handler failed", e),
 		);
 
-		return res.json(message.withSignedAttachments(req));
+		return res.json(message.withSignedAttachments(new NewUrlUserSignatureData({
+			ip: req.ip,
+			userAgent: req.headers["user-agent"] as string,
+		})));
 	},
 );
 
diff --git a/src/api/routes/channels/#channel_id/messages/index.ts b/src/api/routes/channels/#channel_id/messages/index.ts
index 743bc07e..a33b7494 100644
--- a/src/api/routes/channels/#channel_id/messages/index.ts
+++ b/src/api/routes/channels/#channel_id/messages/index.ts
@@ -36,7 +36,7 @@ import {
 	getPermission,
 	isTextChannel,
 	getUrlSignature,
-	uploadFile, NewUrlSignatureData,
+	uploadFile, NewUrlSignatureData, NewUrlUserSignatureData,
 } from "@spacebar/util";
 import { Request, Response, Router } from "express";
 import { HTTPError } from "lambert-server";
@@ -439,7 +439,10 @@ router.post(
 			console.error("[Message] post-message handler failed", e),
 		);
 
-		return res.json(message.withSignedAttachments(req));
+		return res.json(message.withSignedAttachments(new NewUrlUserSignatureData({
+			ip: req.ip,
+			userAgent: req.headers["user-agent"] as string,
+		})));
 	},
 );
 
diff --git a/src/gateway/util/WebSocket.ts b/src/gateway/util/WebSocket.ts
index 4dcec55e..2f8fa7cf 100644
--- a/src/gateway/util/WebSocket.ts
+++ b/src/gateway/util/WebSocket.ts
@@ -20,7 +20,6 @@ import { Intents, ListenEventOpts, Permissions } from "@spacebar/util";
 import WS from "ws";
 import { Deflate, Inflate } from "fast-zlib";
 import { Capabilities } from "./Capabilities";
-import { Request } from "express";
 
 export interface WebSocket extends WS {
 	version: number;