move websocket server intialization to utils

global.d.ts

@@ -0,0 +1,7 @@
+import { ViteDevServer } from "vite";
+
+declare module 'nitropack/types' {
+  interface NitroApp {
+    viteDevServer?: ViteDevServer
+  }
+}

routes/ws.ts

@@ -1,12 +1,10 @@
 export default lazyEventHandler(async () => {
-  const ws = new (await import("ws")).WebSocketServer({ noServer: true });
   return eventHandler((event) => {
-    event.context.ws = ws;
     if (event.headers.get("upgrade") !== "websocket") {
       sendWebResponse(event, new Response(null, { status: 426 }));
       return;
     }
-    ws.handleUpgrade(event.node.req, event.node.req.socket, Buffer.alloc(0), (client) => {
+    useWebsocketServer().handleUpgrade(event.node.req, event.node.req.socket, Buffer.alloc(0), (client) => {
       console.log("connected");
     });
     event._handled = true;

runtime/dev-handler.ts

@@ -1,3 +1,3 @@
 export default lazyEventHandler(async () => {
-  return fromNodeMiddleware(useNitroApp().vite.middlewares);
+  return fromNodeMiddleware(useNitroApp().viteDevServer.middlewares);
 });

runtime/dev-server-plugin.ts

@@ -1,6 +1,6 @@
-export default defineNitroPlugin(async (nitroApp) => {
+export default defineNitroPlugin(async nitroApp => {
   const { createServer } = await import("vite");
-  nitroApp.vite = await createServer({
+  nitroApp.viteDevServer = await createServer({
     configFile: false,
     root: "./client",
     server: {

utils/ws.ts

@@ -0,0 +1,7 @@
+import { WebSocketServer } from "ws";
+
+const server = new WebSocketServer({ noServer: true });
+
+export function useWebsocketServer() {
+  return server;
+}