Handle version in db

This commit is contained in:
Owen 2025-06-19 16:40:20 -04:00
parent bd7e96b8af
commit 377eb2b851
No known key found for this signature in database
GPG key ID: 8271FDFFD9E0CCBD

View file

@ -1,4 +1,4 @@
import { db } from "@server/db";
import { db, newts } from "@server/db";
import { MessageHandler } from "../ws";
import { exitNodes, Newt, resources, sites, Target, targets } from "@server/db";
import { eq, and, sql, inArray } from "drizzle-orm";
@ -38,7 +38,7 @@ export const handleNewtRegisterMessage: MessageHandler = async (context) => {
const siteId = newt.siteId;
const { publicKey, pingResults, backwardsCompatible } = message.data;
const { publicKey, pingResults, newtVersion, backwardsCompatible } = message.data;
if (!publicKey) {
logger.warn("Public key not provided");
return;
@ -59,6 +59,16 @@ export const handleNewtRegisterMessage: MessageHandler = async (context) => {
exitNodeId = bestPingResult.exitNodeId;
}
if (newtVersion) {
// update the newt version in the database
await db
.update(newts)
.set({
version: newtVersion as string,
})
.where(eq(newts.newtId, newt.newtId))
}
const [oldSite] = await db
.select()
.from(sites)