ws

server/middleware/socket.ts

@@ -0,0 +1,52 @@
+import WebSocket, { WebSocketServer } from 'ws'
+
+type Client = {
+    id: string
+    send: (message: string) => void
+    readyState: number
+}
+
+declare global {
+    var wss: WebSocketServer
+    var clients: Client[]
+}
+
+let wss: WebSocketServer
+let clients: Client[] = []
+
+export default defineEventHandler((event) => {
+    if (!global.wss) {
+        console.log("Creating websocket server")
+        wss = new WebSocketServer({ server: event.node.res.socket?.server })
+        wss.on("connection", function (socket) {
+            console.log("Client connected")
+            socket.send("connecte")
+            socket.on("message", function (message) {
+                wss.clients.forEach(function (client) {
+                    console.log("Client", client)
+                    client.send('message')
+                    if (client == socket && client.readyState === WebSocket.OPEN) {
+                        clients.push({
+                            id: message.toString(),
+                            send: (data: string) => client.send(data),
+                            readyState: client.readyState,
+                        })
+                        global.clients = clients
+                    }
+                })
+            })
+            socket.on("close", function () {
+                clients = clients.filter((client) => client.readyState !== WebSocket.CLOSED)
+                global.clients = clients
+                console.log("Client disconnected")
+            })
+            socket.on("error", function (error) {
+                console.log("Error", error)
+            })
+            socket.on("pong", function (data) {
+                console.log("Pong", data)
+            })
+            global.wss = wss
+        })
+    }
+})