// Native const { parse } = require('url'); const http = require('http'); const https = require('https'); // Packages const fetch = require('node-fetch'); /** * Returns a `fetch` version with a similar * API to the browser's configured with a * HTTP2 agent. * * It encodes `body` automatically as JSON. * * @param {String} host * @return {Function} fetch */ module.exports = class Agent { constructor(url, { tls = true, debug } = {}) { this._url = url; const parsed = parse(url); this._protocol = parsed.protocol; this._debug = debug; if (tls) { this._initAgent(); } } _initAgent() { const module = this._protocol === 'https:' ? https : http; this._agent = new module.Agent({ keepAlive: true, keepAliveMsecs: 10000, maxSockets: 8 }).on('error', err => this._onError(err, this._agent)); } _onError(err, agent) { if (this._debug) { console.log(`> [debug] agent connection error ${err}\n${err.stack}`); } if (this._agent === agent) { this._agent = null; } } fetch(path, opts = {}) { if (!this._agent) { if (this._debug) { console.log('> [debug] re-initializing agent'); } this._initAgent(); } const { body } = opts; if (this._agent) { opts.agent = this._agent; } if (body && typeof body === 'object' && typeof body.pipe !== 'function') { opts.headers['Content-Type'] = 'application/json'; opts.body = JSON.stringify(body); } if (opts.body && typeof body.pipe !== 'function') { opts.headers['Content-Length'] = Buffer.byteLength(opts.body); } return fetch(this._url + path, opts); } close() { if (this._debug) { console.log('> [debug] closing agent'); } if (this._agent) { this._agent.destroy(); } } };