Repository URL to install this package:
|
Version:
1.2.7 ▾
|
'use strict';
/**
* Module dependencies.
*/
const CookieJar = require('cookiejar').CookieJar;
const CookieAccess = require('cookiejar').CookieAccessInfo;
const parse = require('url').parse;
const request = require("../..");
const AgentBase = require("../agent-base");
let methods = require('methods');
/**
* Expose `Agent`.
*/
module.exports = Agent;
/**
* Initialize a new `Agent`.
*
* @api public
*/
function Agent(options) {
if (!(this instanceof Agent)) {
return new Agent(options);
}
AgentBase.call(this);
this.jar = new CookieJar();
if (options) {
if (options.ca) {
this.ca(options.ca);
}
if (options.key) {
this.key(options.key);
}
if (options.pfx) {
this.pfx(options.pfx);
}
if (options.cert) {
this.cert(options.cert);
}
}
}
Agent.prototype = Object.create(AgentBase.prototype);
/**
* Save the cookies in the given `res` to
* the agent's cookie jar for persistence.
*
* @param {Response} res
* @api private
*/
Agent.prototype._saveCookies = function (res) {
const cookies = res.headers['set-cookie'];
if (cookies) this.jar.setCookies(cookies);
};
/**
* Attach cookies when available to the given `req`.
*
* @param {Request} req
* @api private
*/
Agent.prototype._attachCookies = function (req) {
const url = parse(req.url);
const access = CookieAccess(url.hostname, url.pathname, 'https:' == url.protocol);
const cookies = this.jar.getCookies(access).toValueString();
req.cookies = cookies;
};
methods.forEach(name => {
const method = name.toUpperCase();
Agent.prototype[name] = function (url, fn) {
const req = new request.Request(method, url);
req.on('response', this._saveCookies.bind(this));
req.on('redirect', this._saveCookies.bind(this));
req.on('redirect', this._attachCookies.bind(this, req));
this._attachCookies(req);
this._setDefaults(req);
if (fn) {
req.end(fn);
}
return req;
};
});
Agent.prototype.del = Agent.prototype['delete'];