93 lines
1.9 KiB
JavaScript
93 lines
1.9 KiB
JavaScript
|
'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'];
|