Newt registration?

server/routers/newt/handleNewtMessage.ts

@@ -1,22 +0,0 @@
-// messageHandlers/chat.ts
-import { MessageHandler } from "../ws";
-
-export const handleNewtMessage: MessageHandler = async (context) => {
-    const { message, senderNewtId, sendToClient } = context;
-    
-    // Process chat message
-    // ... your chat logic here ...
-
-    // Example response
-    return {
-        message: {
-            type: 'newt_response',
-            data: {
-                originalMessage: message.data,
-                timestamp: new Date().toISOString()
-            }
-        },
-        broadcast: false,  // Send to all clients
-        excludeSender: false  // Include sender in broadcast
-    };
-};

server/routers/newt/handleRegisterMessage.ts

@@ -0,0 +1,42 @@
+import db from "@server/db";
+import { MessageHandler } from "../ws";
+import { sites } from "@server/db/schema";
+import { eq } from "drizzle-orm";
+
+export const handleRegisterMessage: MessageHandler = async (context) => {
+    const { message, newt, sendToClient } = context;
+    
+    if (!newt) {
+        console.log("Newt not found");
+        return;
+    }
+
+    if (!newt.siteId) {
+        console.log("Newt has no site!"); // TODO: Maybe we create the site here?
+        return;
+    }
+    
+    const siteId = newt.siteId;
+        
+    // get the site
+    const site = await db
+    .select()
+    .from(sites)
+    .where(eq(sites.siteId, siteId))
+    .limit(1);
+
+
+        const { publicKey } = message.data;
+
+    return {
+        message: {
+            type: 'newt/wg/connect',
+            data: {
+                publicKey: 'publicKey',
+
+            }
+        },
+        broadcast: false,  // Send to all clients
+        excludeSender: false  // Include sender in broadcast
+    };
+};

server/routers/newt/index.ts

@@ -1,3 +1,3 @@
 export * from "./createNewt";
 export * from "./getToken";
-export * from "./handleNewtMessage";
+export * from "./handleRegisterMessage";