server.ts

// server.ts
import blitz from "blitz/custom-server"
import { BlitzApiHandler } from "blitz"
import * as http from "http"
import * as socketio from "socket.io"
import express, { Express, Request, Response } from "express"
import { parse } from "url"
import { log } from "next/dist/server/lib/logging"
import { Message } from "@prisma/client"

const { PORT = "3000" } = process.env
const dev = process.env.NODE_ENV !== "production"
const blitzApp = blitz({ dev })
const blitzHandler: BlitzApiHandler = blitzApp.getRequestHandler()

blitzApp.prepare().then(async () => {
  const app: Express = express()
  const server: http.Server = http.createServer(app)
  const io: socketio.Server = new socketio.Server()
  io.attach(server)

  app.get("/hello", async (_: Request, res: Response) => {
    res.send("Hello World")
  })
  app.set("proxy", 1)

  const sockets: socketio.Socket[] = []

  io.on("connection", (socket: socketio.Socket) => {
    console.log("connection", socket.handshake.query)
    sockets.push(socket)

    const { roomId } = socket.handshake.query
    if (roomId) socket.join(`blitz-chat-socket-test-${roomId}`)

    const { conversationId } = socket.handshake.query
    if (conversationId) socket.join(`blitz-chat-${conversationId}`)

    socket.emit("status", "Hello from Socket.io")

    // socket-test handling
    socket.on("hi-from-client", (data) => {
      console.log("hi-from-client received, sending hi-from-server", data, socket.rooms)
      // sockets.forEach((s) => s.emit("hi-from-server", data))
      // https://socket.io/docs/v3/rooms/
      socket.rooms.forEach((room) => {
        if (room.startsWith("blitz-chat-socket-test-")) {
          io.to(room).emit("hi-from-server", data)
          // socket.to(room).emit("hi-from-server", data)
        }
      })
    })

    // conversation handling
    socket.on("new-message", (data: Message) => {
      console.log("new-message", data)
      const roomName = `blitz-chat-${data.conversationId}`
      io.to(roomName).emit("new-remote-message", data)
    })

    socket.on("disconnect", () => {
      console.log("client disconnected")
      socket.disconnect()
    })
  })

  app.all("*", (req: any, res: any) => blitzHandler(req, res))

  server.listen(PORT, () => {
    console.log(`> Ready on http://localhost:${PORT}`)
  })
})