mirror of
				https://github.com/crazy-max/ghaction-upx.git
				synced 2025-11-04 04:58:11 -07:00 
			
		
		
		
	
		
			
				
	
	
		
			52 lines
		
	
	
		
			1.2 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			52 lines
		
	
	
		
			1.2 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
'use strict';
 | 
						|
const bufferStream = require('./buffer-stream');
 | 
						|
 | 
						|
function getStream(inputStream, opts) {
 | 
						|
	if (!inputStream) {
 | 
						|
		return Promise.reject(new Error('Expected a stream'));
 | 
						|
	}
 | 
						|
 | 
						|
	opts = Object.assign({maxBuffer: Infinity}, opts);
 | 
						|
 | 
						|
	const maxBuffer = opts.maxBuffer;
 | 
						|
	let stream;
 | 
						|
	let clean;
 | 
						|
 | 
						|
	const p = new Promise((resolve, reject) => {
 | 
						|
		const error = err => {
 | 
						|
			if (err) { // null check
 | 
						|
				err.bufferedData = stream.getBufferedValue();
 | 
						|
			}
 | 
						|
 | 
						|
			reject(err);
 | 
						|
		};
 | 
						|
 | 
						|
		stream = bufferStream(opts);
 | 
						|
		inputStream.once('error', error);
 | 
						|
		inputStream.pipe(stream);
 | 
						|
 | 
						|
		stream.on('data', () => {
 | 
						|
			if (stream.getBufferedLength() > maxBuffer) {
 | 
						|
				reject(new Error('maxBuffer exceeded'));
 | 
						|
			}
 | 
						|
		});
 | 
						|
		stream.once('error', error);
 | 
						|
		stream.on('end', resolve);
 | 
						|
 | 
						|
		clean = () => {
 | 
						|
			// some streams doesn't implement the `stream.Readable` interface correctly
 | 
						|
			if (inputStream.unpipe) {
 | 
						|
				inputStream.unpipe(stream);
 | 
						|
			}
 | 
						|
		};
 | 
						|
	});
 | 
						|
 | 
						|
	p.then(clean, clean);
 | 
						|
 | 
						|
	return p.then(() => stream.getBufferedValue());
 | 
						|
}
 | 
						|
 | 
						|
module.exports = getStream;
 | 
						|
module.exports.buffer = (stream, opts) => getStream(stream, Object.assign({}, opts, {encoding: 'buffer'}));
 | 
						|
module.exports.array = (stream, opts) => getStream(stream, Object.assign({}, opts, {array: true}));
 |