now-cli/lib/agent.js

// 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();
    }
  }
};