43 lines
1.5 KiB
JavaScript
43 lines
1.5 KiB
JavaScript
"use strict";
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.headStream = void 0;
|
|
const stream_1 = require("stream");
|
|
const headStream_browser_1 = require("./headStream.browser");
|
|
const stream_type_check_1 = require("./stream-type-check");
|
|
const headStream = (stream, bytes) => {
|
|
if ((0, stream_type_check_1.isReadableStream)(stream)) {
|
|
return (0, headStream_browser_1.headStream)(stream, bytes);
|
|
}
|
|
return new Promise((resolve, reject) => {
|
|
const collector = new Collector();
|
|
collector.limit = bytes;
|
|
stream.pipe(collector);
|
|
stream.on("error", (err) => {
|
|
collector.end();
|
|
reject(err);
|
|
});
|
|
collector.on("error", reject);
|
|
collector.on("finish", function () {
|
|
const bytes = new Uint8Array(Buffer.concat(this.buffers));
|
|
resolve(bytes);
|
|
});
|
|
});
|
|
};
|
|
exports.headStream = headStream;
|
|
class Collector extends stream_1.Writable {
|
|
buffers = [];
|
|
limit = Infinity;
|
|
bytesBuffered = 0;
|
|
_write(chunk, encoding, callback) {
|
|
this.buffers.push(chunk);
|
|
this.bytesBuffered += chunk.byteLength ?? 0;
|
|
if (this.bytesBuffered >= this.limit) {
|
|
const excess = this.bytesBuffered - this.limit;
|
|
const tailBuffer = this.buffers[this.buffers.length - 1];
|
|
this.buffers[this.buffers.length - 1] = tailBuffer.subarray(0, tailBuffer.byteLength - excess);
|
|
this.emit("finish");
|
|
}
|
|
callback();
|
|
}
|
|
}
|