Browse Source

Add message handler seperate file

Owen Schwartz 8 tháng trước cách đây
mục cha
commit
e141263b7e
2 tập tin đã thay đổi với 8 bổ sung6 xóa
  1. 6 0
      server/routers/messageHandlers.ts
  2. 2 6
      server/routers/ws.ts

+ 6 - 0
server/routers/messageHandlers.ts

@@ -0,0 +1,6 @@
+import { handleNewtMessage } from "./newt";
+import { MessageHandler } from "./ws";
+
+export const messageHandlers: Record<string, MessageHandler> = {
+    "newt": handleNewtMessage,
+};

+ 2 - 6
server/routers/ws.ts

@@ -7,7 +7,7 @@ import { Newt, newts, NewtSession } from "@server/db/schema";
 import { eq } from "drizzle-orm";
 import db from "@server/db";
 import { validateNewtSessionToken } from "@server/auth/newt";
-import { handleNewtMessage } from "./newt";
+import { messageHandlers } from "./messageHandlers";
 
 // Custom interfaces
 interface WebSocketRequest extends IncomingMessage {
@@ -47,10 +47,6 @@ interface HandlerContext {
 
 export type MessageHandler = (context: HandlerContext) => Promise<HandlerResponse | void>;
 
-const messageHandlers: Record<string, MessageHandler> = {
-    "newt": handleNewtMessage,
-};
-
 const router: Router = Router();
 const wss: WebSocketServer = new WebSocketServer({ noServer: true });
 
@@ -202,7 +198,7 @@ wss.on("connection", (ws: AuthenticatedWebSocket, request: WebSocketRequest) =>
     ws.on("message", async (data) => {
         try {
             const message: WSMessage = JSON.parse(data.toString());
-            console.log(`Message received from Newt ID ${newtId}:`, message);
+            // console.log(`Message received from Newt ID ${newtId}:`, message);
     
             // Validate message format
             if (!message.type || typeof message.type !== "string") {