Function server

backend/src/server/server.js:29–201

(config = {})

const WebSocket = require('ws');
const { parse } = require('url');
function server(config = {}) {
    const ROUTES = [];
    const WARES = [];
    const WS_ROUTES = [];
    const wss = new WebSocket.Server({ noServer: true });
    const SERVER = http.createServer((req, res) => {
        let u = parse(req.url, true);
        req.query = u.query || {};
        req.path = u.pathname;
        req.hostname = (req.headers.host || '').split(':')[0].replace(/[^\w.-]/g, '');
        req.ip = (req.socket.remoteAddress || '').replace(/[^\w.:]/g, '');
        res.statusCode = 200;
        res.status = (x) => {
            res.statusCode = x;
            return res;
        };
        res.json = (x) => {
            res.writeHead(res.statusCode || 200, { 'Content-Type': 'application/json' });
            res.end(JSON.stringify(x));
        };
        res.send = (x) => {
            if (x === undefined || x === null)
                x = '';
            if (typeof x === 'object')
                return res.json(x);
            res.writeHead(res.statusCode || 200, { 'Content-Type': 'text/plain' });
            res.end(String(x));
        };
        res.set = (k, v) => { res.setHeader(k, v); return res; };
        let r = matchRoute(req.method.toUpperCase(), req.path);
        req.params = r ? r.params : {};
        let fns = [...WARES];
        fns.push(r ? (req, res, next) => r.handler(req, res, next) : (_req, res) => res.status(404).end('404: Not Found'));
        let i = 0;
        let next = () => {
            if (i < fns.length)
                fns[i++](req, res, next);
        };
        next();
    });
    SERVER.on('upgrade', (req, socket, head) => {
        let u = parse(req.url || '', true);
        let path = u.pathname;
        if (!path || path.includes('..') || /[\0-\x1F\x7F]/.test(path)) {
            socket.destroy();
            return;
        }
        for (let i = 0; i < WS_ROUTES.length; i++) {
            let r = WS_ROUTES[i];
            if (r.path === path) {
                wss.handleUpgrade(req, socket, head, (ws) => {
                    ws.req = req;
                    r.handler(ws, req);
                });
                return;
            }
        }
        socket.destroy();