problemi/server.js

import path from 'path'
import fs from 'fs/promises'

import express from 'express'
import morgan from 'morgan'

import { createServer as createViteServer } from 'vite'

import { createApiRouter } from './server/routes.js'

const HTML_ROUTES = ['/', '/login', '/problem/:id']

const config = {
    isDevelopment: process.env.MODE === 'development',
    port: process.env.PORT || 3000,
}

if (config.isDevelopment) {
    console.log(`[Config] Running in development mode`)
    console.log(`[Config] PORT = ${config.port}`)
}

function mountIndexHtmlRoutes(r, serveIndexHtml) {
    for (const route of HTML_ROUTES) {
        console.log(`[Server] Mounted index html for "${route}"`)
        r.get(route, serveIndexHtml)
    }

    r.use('*', (_req, res) => {
        res.redirect('/')
    })
}

async function createDevRouter() {
    const r = express.Router()

    const vite = await createViteServer({
        server: { middlewareMode: true },
        appType: 'custom',
    })

    r.use(vite.middlewares)
    mountIndexHtmlRoutes(r, async (req, res, next) => {
        try {
            const indexHtml = await fs.readFile(path.resolve('./index.html'), 'utf-8')
            const transformedTemplate = await vite.transformIndexHtml(req.originalUrl, indexHtml)

            // Load (to be bundled) entry point for server side rendering
            const { render } = await vite.ssrLoadModule('./client/entry-server.jsx')

            const html = transformedTemplate.replace('<!-- SSR OUTLET -->', render(req.originalUrl))

            res.send(html)
        } catch (error) {
            vite.ssrFixStacktrace(error)
            next(error)
        }
    })

    return r
}

async function createProductionRouter() {
    // Load bundled entry point for server side rendering
    const { render } = await import('./dist/server/entry-server.js')

    const r = new express.Router()

    r.use('/', express.static('dist/client'))

    mountIndexHtmlRoutes(r, async (req, res) => {
        const transformedTemplate = await fs.readFile(
            path.resolve('./dist/client/index.html'),
            'utf-8'
        )

        const html = transformedTemplate.replace('<!-- SSR OUTLET -->', render(req.originalUrl))

        res.send(html)
    })

    return r
}

async function main() {
    const app = express()

    app.use(morgan('dev'))

    app.use('/', await createApiRouter())

    if (config.isDevelopment) {
        app.use('/', await createDevRouter())
    } else {
        app.use('/', await createProductionRouter())
    }

    app.listen(config.port, () => {
        console.log(`Listening on port ${config.port}...`)
    })
}

main()