Added support for StreamableFile class

Author: davidschuette

src/app/app.controller.ts

@@ -1,15 +1,21 @@
-import { Controller, Get, Param, Post, Req, Res } from '@nestjs/common'
-import { AppService } from './app.service'
-
-import { FastifyRequest, FastifyReply } from 'fastify'
-import { Http2ServerResponse } from 'http2'
 import {
-  ApiTags,
-  ApiOperation,
+  Controller,
+  Get,
+  Param,
+  Post,
+  Req,
+  Res,
+  StreamableFile,
+} from '@nestjs/common'
+import {
   ApiConsumes,
-  ApiOkResponse,
   ApiCreatedResponse,
+  ApiOkResponse,
+  ApiOperation,
+  ApiTags,
 } from '@nestjs/swagger'
+import { FastifyReply, FastifyRequest } from 'fastify'
+import { AppService } from './app.service'
 import { File } from './models/file.entity'
 
 type Request = FastifyRequest
@@ -79,8 +85,8 @@ export class AppController {
   downloadFile(
     @Param('id') id: string,
     @Req() request: Request,
-    @Res() response: Response,
-  ) {
-    this.appService.download(id, request, response)
+    @Res({ passthrough: true }) response: Response,
+  ): Promise<StreamableFile> {
+    return this.appService.download(id, request, response)
   }
 }

src/app/app.service.ts

@@ -1,16 +1,16 @@
 import {
+  BadRequestException,
   Injectable,
   NotFoundException,
   ServiceUnavailableException,
-  BadRequestException,
+  StreamableFile,
 } from '@nestjs/common'
 import { InjectConnection, InjectModel } from '@nestjs/mongoose'
 import { FastifyReply, FastifyRequest } from 'fastify'
-import { Http2ServerResponse } from 'http2'
 import { GridFSBucket, ObjectId } from 'mongodb'
 import { Connection, Model, mongo } from 'mongoose'
-import { File } from './models/file.entity'
 import { Stream } from 'stream'
+import { File } from './models/file.entity'
 
 type Request = FastifyRequest
 type Response = FastifyReply
@@ -60,7 +60,11 @@ export class AppService {
     })
   }
 
-  async download(id: string, request: Request, response: Response) {
+  async download(
+    id: string,
+    request: Request,
+    response: Response,
+  ): Promise<StreamableFile> {
     try {
       if (!ObjectId.isValid(id)) {
         throw new BadRequestException(null, 'InvalidVideoId')
@@ -93,18 +97,10 @@ export class AppService {
           'Content-Disposition': `attachment; filename="${fileInfo.filename}"`,
         })
 
-        response.raw.on('close', () => {
-          readstream.destroy()
-        })
-
-        response.send(readstream)
+        return new StreamableFile(readstream)
       } else {
         const readstream = this.bucket.openDownloadStream(oId)
 
-        response.raw.on('close', () => {
-          readstream.destroy()
-        })
-
         response.status(200)
         response.headers({
           'Accept-Range': 'bytes',