restpki-client/lib/file-result.js

Classes to consume Lacuna Software REST PKI

'use strict';
const fs = require('fs');
const {Readable} = require('stream');

const {FileModel} = require('./file-model');
const {StreamUtils} = require('./stream-utils');

class FileResult {

	constructor(client, file) {
		this._client = client;
		this._file = new FileModel(file);
	}

	async openRead() {
		if (this._file.content) {
			let readable = new Readable();
			readable._read = () => {
			}; // _read is required but you can noop it
			readable.push(this._file.content);
			readable.push(null);
			return readable;
		} else {
			if (this._file.url) {
				try {
					return await this._client.getRestClient().getAsStream(this._file.url);
				} catch (err) {
					throw err;
				}
			} else {
				throw new Error('Cannot open stream of empty file');
			}
		}
	}

	async getContent() {
		if (this._file.content) {
			return new Buffer(this._file.content, 'base64');
		}
		try {
			let content = await this.openRead();
			return await StreamUtils.streamToBuffer(content);
		} catch (err) {
			throw new Error('Error reading stream');
		}
	}

	async writeToFile(path) {
		if (this._file) {
			try {
				let content = await this.getContent();
				fs.writeFileSync(path, new Buffer(content, 'base64'));
			} catch (err) {
				throw err;
			}
		}
	}

	getFile() {
		return this._file;
	}

	setFile(file) {
		this._file = file;
	}

	get file() {
		return this.getFile();
	}

	set file(file) {
		return this.setFile(file);
	}
}

exports.FileResult = FileResult;