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 = this._baseUrl = `${u.protocol}://${u.resource}${myPort}`; this.get = bent(baseUrl, 'GET', 'buffer', 200, 201); this.post = bent(baseUrl, 'POST', 'buffer', 200, 201); this.username = hook.username; this.password = hook.password; assert(isAbsoluteUrl(this.url)); assert(['GET', 'POST'].includes(this.method)); const {stats} = require('../../').srf.locals; this.stats = stats; } get baseUrl() { return this._baseUrl; } /** * 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, params}, `Requestor:request ${method} ${url}`); const startAt = process.hrtime(); let buf; try { buf = isRelativeUrl(url) ? await this.post(url, params, this.authHeader) : await bent(method, 'buffer', 200, 201)(url, params, basicAuth(username, password)); } catch (err) { this.logger.info({baseUrl: this.baseUrl, url: err.statusCode}, `web callback returned unexpected error code ${err.statusCode}`); throw err; } const diff = process.hrtime(startAt); const time = diff[0] * 1e3 + diff[1] * 1e-6; const rtt = time.toFixed(0); this.logger.debug(`Requestor:request ${method} ${url} succeeded in ${rtt}ms`); if (buf) this.stats.histogram('app.hook.response_time', rtt, ['hook_type:app']); 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;