httputils.js

最新推荐文章于 2024-08-20 10:03:20 发布

原创最新推荐文章于 2024-08-20 10:03:20 发布 · 838 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#nodejs #cookie

nodejs 专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一个用于处理HTTP请求的Node.js模块，该模块能够发送GET和POST请求，并且能够解析和打包Cookie。通过使用此模块，开发者可以轻松地与Web服务进行交互，管理会话状态。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >


'use strict';

const querystring = require("querystring");
const http = require("http");

function parseCookies(res) {
    let list = {};
    let ra = res.headers['set-cookie'];
    //~ not every response get a `set-cookie`
    if (ra) {
        ra.forEach((rc) => {
            rc && rc.split(';').forEach((cookie) => {
                let parts = cookie.split('=');
                list[parts.shift().trim()] = parts.join('=');
            });
        });
    }
    return list;
}

function packCookie(opts) {
    let ar = [];
    for (let k in opts) {
        ar.push(`${k}=${opts[k]}`);
    }
    return ar.join("; ");
}

function getResDo(cb) {
    return (res) => {
        res.setEncoding('utf8');
        let rs = '';
        res.on('data', (chunk) => {
            rs += `${chunk}`;
        });
        res.on('end', () => {
            //console.log("response is %s", resJson);
            //console.log(res.headers);
            //console.log("====");
            //console.log(res.headers.cookie);// undefined
            //console.log(res.headers['set-cookie']); // an array 
            //console.log("Is an array: %s", Array.isArray(res.headers['set-cookie']));
            cb(res, rs);
        });
    };
}

function makeDoGetTo(targetHost, targetPort) {
    return (uri, cOpts) => {
        return new Promise((resolve, reject) => {
            let opts = {
                hostname: targetHost,
                port: targetPort,
                path: uri,
                method: 'GET',
                headers: {
                    Cookie: packCookie(cOpts)
                }
            };
            let req = http.request(opts, getResDo((res, rs) => resolve({
                response: rs,
                cookie: parseCookies(res),
                oldCookie: cOpts
            })));
            req.end();
        });
    }
}


function makeDoPostTo(targetHost, targetPort) {
    return (uri, postdata, cOpts) => {
        return new Promise((resolve, reject) => {
            let data = querystring.stringify(postdata);
            let postOpts = {
                hostname: targetHost,
                port: targetPort,
                path: uri,
                method: 'POST',
                headers: {
                    'Content-Type': 'application/x-www-form-urlencoded',
                    'Content-Length': data.length,
                    Cookie: packCookie(cOpts)
                }
            };
            let req = http.request(postOpts, getResDo(
                (res, rs) => resolve({
                    response: rs,
                    cookie: parseCookies(res),
                    oldCookie: cOpts
                })
            ));
            req.end(data);
        });
    }
}

module.exports = {
    makeDoGetTo,
    makeDoPostTo
};