jambonz-feature-server/lib/utils/requestor.js

const bent = require('bent');
const parseUrl = require('parse-url');
const assert = require('assert');

const toBase64 = (str) => Buffer.from(str || '', 'utf8').toString('base64');

function basicAuth(username, password) {
  if (!username || !password) return {};
  const creds = `${username}:${password || ''}`;
  const header = `Basic ${toBase64(creds)}`;
  return {Authorization: header};
}

function isRelativeUrl(u) {
  return typeof u === 'string' && u.startsWith('/');
}

function isAbsoluteUrl(u) {
  return typeof u === 'string' &&
    u.startsWith('https://') || u.startsWith('http://');
}

class Requestor {
  constructor(logger, hook) {
    assert(typeof hook === 'object');

    this.logger = logger;
    this.url = hook.url;
    this.method = hook.method || 'POST';
    this.authHeader = basicAuth(hook.username, hook.password);

    const u = parseUrl(this.url);
    const myPort = u.port ? `:${u.port}` : '';
    const baseUrl = `${u.protocol}://${u.resource}${myPort}`;

    this.get = bent(baseUrl, 'GET', 'buffer', 200, 201);
    this.post = bent(baseUrl, 'POST', 'buffer', 200, 201);

    assert(isAbsoluteUrl(this.url));
    assert(['GET', 'POST'].includes(this.method));
  }

  /**
   * Make an HTTP request.
   * All requests use json bodies.
   * All requests expect a 200 statusCode on success
   * @param {object|string} hook - may be a absolute or relative url, or an object
   * @param {string} [hook.url] - an absolute or relative url
   * @param {string} [hook.method] - 'GET' or 'POST'
   * @param {string} [hook.username] - if basic auth is protecting the endpoint
   * @param {string} [hook.password] - if basic auth is protecting the endpoint
   * @param {object} [params] - request parameters
   */
  async request(hook, params) {
    params = params || null;
    const url = hook.url || hook;
    const method = hook.method || 'POST';
    const {username, password} = typeof hook === 'object' ? hook : {};

    assert.ok(url, 'Requestor:request url was not provided');
    assert.ok, (['GET', 'POST'].includes(method), `Requestor:request method must be 'GET' or 'POST' not ${method}`);


    this.logger.debug({hook}, `Requestor:request ${method} ${url}`);
    const buf = isRelativeUrl(url) ?
      await this.post(url, params, this.authHeader) :
      await bent(method, 'buffer', 200, 201)(url, params, basicAuth(username, password));
    //this.logger.debug({body: }, `Requestor:request ${method} ${url} succeeded`);

    if (buf && buf.toString().length > 0) {
      try {
        const json = JSON.parse(buf.toString());
        return json;
      }
      catch (err) {
        this.logger.debug({err, url, method}, `Requestor:request returned non-JSON content: '${buf.toString()}'`);
      }
    }
  }
}

module.exports = Requestor;