From a0abe41c8aa379000ea4a03917faa5c526d142b3 Mon Sep 17 00:00:00 2001
From: Owen <owen@txv.io>
Date: Fri, 21 Feb 2025 17:13:20 -0500
Subject: [PATCH] Fix up ws messages

---
 server/routers/newt/handleGetConfigMessage.ts    | 4 ++--
 server/routers/newt/handleNewtRegisterMessage.ts | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/server/routers/newt/handleGetConfigMessage.ts b/server/routers/newt/handleGetConfigMessage.ts
index 391a2af7..7226ce24 100644
--- a/server/routers/newt/handleGetConfigMessage.ts
+++ b/server/routers/newt/handleGetConfigMessage.ts
@@ -95,7 +95,7 @@ export const handleGetConfigMessage: MessageHandler = async (context) => {
         clientsRes.map(async (client) => {
             return {
                 publicKey: client.pubKey,
-                allowedIps: "0.0.0.0/0" // TODO: We should lock this down more
+                allowedIps: ["0.0.0.0/0"] // TODO: We should lock this down more
             };
         })
     );
@@ -112,7 +112,7 @@ export const handleGetConfigMessage: MessageHandler = async (context) => {
         message: {
             type: "newt/wg/receive-config", // what to make the response type?
             data: {
-                config: configResponse
+                ...configResponse
             }
         },
         broadcast: false, // Send to all clients
diff --git a/server/routers/newt/handleNewtRegisterMessage.ts b/server/routers/newt/handleNewtRegisterMessage.ts
index ec60641f..21ad4ba6 100644
--- a/server/routers/newt/handleNewtRegisterMessage.ts
+++ b/server/routers/newt/handleNewtRegisterMessage.ts
@@ -9,7 +9,7 @@ import {
     targets
 } from "@server/db/schema";
 import { eq, and, sql } from "drizzle-orm";
-import { addPeer, deletePeer } from "../newt/peers";
+import { addPeer, deletePeer } from "../gerbil/peers";
 import logger from "@server/logger";
 
 export const handleNewtRegisterMessage: MessageHandler = async (context) => {