segnapunti/server.js

import { createServer } from 'http'
import express from 'express'
import { WebSocketServer } from 'ws'
import { fileURLToPath } from 'url'
import { dirname, join } from 'path'
import { setupWebSocketHandler } from './src/websocket-handler.js'
import { printServerInfo } from './src/server-utils.js'
import { loadState, saveState } from './src/persist.js'

const __filename = fileURLToPath(import.meta.url)
const __dirname = dirname(__filename)

const PORT = process.env.PORT || 3000
const distDir = join(__dirname, 'dist')

const app = express()

app.use(express.static(distDir, { index: false }))

app.get(['/', '/display', '/display/*splat'], (_req, res) => {
    res.sendFile(join(distDir, 'index.html'))
})

app.get(['/controller', '/controller/*splat'], (_req, res) => {
    res.sendFile(join(distDir, 'controller.html'))
})

const server = createServer(app)
const wss = new WebSocketServer({ noServer: true })
setupWebSocketHandler(wss, { initialState: loadState(), onStateChange: saveState })

server.on('upgrade', (request, socket, head) => {
    const pathname = new URL(request.url, `http://${request.headers.host}`).pathname
    if (pathname === '/ws') {
        wss.handleUpgrade(request, socket, head, (ws) => {
            wss.emit('connection', ws, request)
        })
    } else {
        socket.destroy()
    }
})

server.listen(PORT, '0.0.0.0', () => {
    printServerInfo(PORT)
})