UNPKG

@nestjs/common

Version:

Nest - modern, fast, powerful node.js web framework (@common)

46 lines (45 loc) 1.47 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.StreamableFile = void 0; const stream_1 = require("stream"); const util_1 = require("util"); const shared_utils_1 = require("../utils/shared.utils"); class StreamableFile { constructor(bufferOrReadStream, options = {}) { var _a; var _b; this.options = options; this.handleError = (err, res) => { res.statusCode = 400; res.send(err.message); }; if (util_1.types.isUint8Array(bufferOrReadStream)) { this.stream = new stream_1.Readable(); this.stream.push(bufferOrReadStream); this.stream.push(null); (_a = (_b = this.options).length) !== null && _a !== void 0 ? _a : (_b.length = bufferOrReadStream.length); } else if (bufferOrReadStream.pipe && (0, shared_utils_1.isFunction)(bufferOrReadStream.pipe)) { this.stream = bufferOrReadStream; } } getStream() { return this.stream; } getHeaders() { const { type = 'application/octet-stream', disposition = undefined, length = undefined, } = this.options; return { type, disposition, length, }; } get errorHandler() { return this.handleError; } setErrorHandler(handler) { this.handleError = handler; return this; } } exports.StreamableFile = StreamableFile;