Working on new exit node picking

server/routers/newt/handleNewtPingRequestMessage.ts

@@ -0,0 +1,38 @@
+import { db } from "@server/db";
+import { MessageHandler } from "../ws";
+import { exitNodes, Newt } from "@server/db";
+import logger from "@server/logger";
+
+export const handleNewtPingRequestMessage: MessageHandler = async (context) => {
+    const { message, client, sendToClient } = context;
+    const newt = client as Newt;
+
+    logger.info("Handling ping request newt message!");
+
+    if (!newt) {
+        logger.warn("Newt not found");
+        return;
+    }
+
+    // TODO: pick which nodes to send and ping better than just all of them
+    const exitNodesList = await db
+        .select()
+        .from(exitNodes);
+
+    let exitNodesPayload = exitNodesList.map((node) => ({
+        exitNodeId: node.exitNodeId,
+        endpoint: node.endpoint,
+        weight: 0 // TODO: Implement weight calculation if needed depending on load
+    }));
+
+    return {
+        message: {
+            type: "newt/ping/exitNodes",
+            data: { 
+                exitNodes: exitNodesPayload
+            }
+        },
+        broadcast: false, // Send to all clients
+        excludeSender: false // Include sender in broadcast
+    };
+};