restpki-client/lib/rest-client.js

Classes to consume Lacuna Software REST PKI

'use strict';
const axios = require('axios');

const {RestUnreachableError} = require('./rest-unreachable-error');
const {ValidationResults} = require('./validation');
const {ValidationError} = require('./validation-error');
const {RestError} = require('./rest-error');
const {RestPkiError} = require('./rest-pki-error');
const {
	ReadableStreamClone,
	StreamUtils
} = require('./stream-utils');
const {VERSION} = require('./lib-version');

class RestClient {

	constructor(endpointUrl, accessToken = null, proxy = null, timeout = null) {
		this._endpointUrl = endpointUrl;
		this._accessToken = accessToken;
		this._proxy = proxy;
		this._timeout = timeout;

		let param = {
			baseURL: this._endpointUrl,
			json: true,
			responseType: 'application/json',
			maxContentLength: Infinity,
			maxBodyLength: Infinity
		};

		if (this._timeout) {
			param.time = this._timeout;
		}

		if (this._proxy) {
			param.proxy = this._proxy;
		}

		let header = {
			'X-RestPki-Client': `NodeJS ${VERSION}`
		};

		if (this._accessToken) {
			header.Authorization = `Bearer ${this._accessToken}`;
		}

		param.headers = header;

		this._instance = axios.create(param);
	}

	get(url) {
		let verb = 'GET';
		return this._instance.get(url)
			.then((response) => {
				if(!response.data) { // response is in response.data
					return typeof(response) == "string" ? JSON.parse(response) : response;
				}
				return typeof(response.data) == "string" ? JSON.parse(response.data) : response.data;
			})
			.catch((error) => {
				if (error.response) {
					let errObj = {value: error.status};
					this._checkResponse(errObj, error.response, verb, url);
					throw errObj.value;
				} else if (error.request) {
					throw new RestUnreachableError(verb, url, error);
				} else {
					throw error;
				}
			});
	}

	getAsBuffer(url) {
		return this._instance.get(url, {responseType: 'stream'})
			.then(function (response) {
				let stream = new ReadableStreamClone(response.data);
				return StreamUtils.streamToBuffer(stream);
			});
	}

	getAsStream(url) {
		return this._instance.get(url, {responseType: 'stream'})
			.then(function (response) {
				return new ReadableStreamClone(response.data);
			});
	}

	post(url, data = null) {
		let verb = 'POST';
		return this._instance.post(url, data)
			.then((response) => {
				if(!response.data) { // response is in response.data
					return typeof(response) == "string" ? JSON.parse(response) : response;
				}
				return typeof(response.data) == "string" ? JSON.parse(response.data) : response.data;
			})
			.catch((error) => {
				if (error.response) {
					let errObj = {value: error.status};
					this._checkResponse(errObj, error.response, verb, url);
					throw errObj.value;
				} else if (error.request) {
					throw new RestUnreachableError(verb, url, error);
				} else {
					throw error;
				}
			});
	}

	_checkResponse(errObj, restRes, verb, url) {
		try {
			let statusCode = restRes.status;
			let response = JSON.parse(restRes.data);
			if (statusCode === 422 && response.code !== null) {
				if (response.code === 'ValidationError') {
					let vr = new ValidationResults(response.validationResults);
					errObj.value = new ValidationError(verb, url, vr);
				} else {
					errObj.value = new RestPkiError(verb, url, response.code,
						response.detail);
				}
			} else {
				errObj.value = new RestError(verb, url, statusCode,
					response.message);
			}
		} catch (error) {
			errObj.value = new RestError(verb, url);
		}
	}

}

exports.RestClient = RestClient;