server.js

const { createServer } = require('http')
const { parse } = require('url')
const next = require('next')
const { initWebSocketServer } = require('./lib/websocket-server')
require('dotenv').config()

const dev = process.env.NODE_ENV !== 'production'
const hostname = 'localhost'
const port = process.env.PORT || 3000
const wsPort = process.env.WS_PORT || 3001
const app = next({ dev, hostname, port })
const handle = app.getRequestHandler()

app.prepare().then(() => {
  // Next.js server
  const server = createServer(async (req, res) => {
    try {
      const parsedUrl = parse(req.url, true)
      await handle(req, res, parsedUrl)
    } catch (err) {
      console.error('Error occurred handling', req.url, err)
      res.statusCode = 500
      res.end('internal server error')
    }
  })

  // Separate WebSocket server on different port
  const wsServer = createServer((req, res) => {
    // Basic HTTP response for non-WebSocket requests
    res.writeHead(426, { 'Content-Type': 'text/plain' })
    res.end('This port is for WebSocket connections only')
  })
  
  // Initialize WebSocket server on separate port
  initWebSocketServer(wsServer)

  // Start both servers
  server.listen(port, (err) => {
    if (err) throw err
    console.log(`> Next.js ready on http://${hostname}:${port}`)
  })

  wsServer.listen(wsPort, (err) => {
    if (err) throw err
    console.log(`> WebSocket server ready on ws://${hostname}:${wsPort}/ws`)
  })
})