@jnode/server/src/util.js

Simple web server package for Node.js.

/*
@jnode/server/util.js
v2

Simple web server package for Node.js.

by JustApple
*/

// receive body
function receiveBody(req, max = 1024 * 1024) {
    return new Promise((resolve, reject) => {
        const parts = [];
        let size = 0;
        req.on('data', (chunk) => {
            size += chunk.length;
            if (size > max) return;
            parts.push(chunk);
        });
        req.on('end', () => {
            if (size > max) reject(413);
            else resolve(Buffer.concat(parts));
        });
        req.on('error', err => reject(err));
    });
}

function setCookie(res, key, value, options) {
    const cookieHeaders = res.getHeader('set-cookie');
    let cookie = `${key}=${encodeURIComponent(value)}`;

    if (options) {
        for (let i in options) {
            if (!options[i]) continue;
            cookie += '; ';
            cookie += i;
            cookie += options[i] === true ? '' : '=' + options[i];
        }
    }

    if (Array.isArray(cookieHeaders)) {
        cookieHeaders.push(cookie);
        res.setHeader('Set-Cookie', cookieHeaders);
    } else if (cookieHeaders === undefined) {
        res.setHeader('Set-Cookie', cookie);
    } else {
        res.setHeader('Set-Cookie', [cookieHeaders, cookie]);
    }
}

// export
module.exports = {
    receiveBody,
    setCookie
};