1- import type { Multipart } from "@fastify/multipart" ;
21import {
32 BadRequestException ,
43 ExecutionContext ,
5- InternalServerErrorException ,
64 createParamDecorator ,
75} from "@nestjs/common" ;
86import type { HttpArgumentsHost } from "@nestjs/common/interfaces" ;
97import type express from "express" ;
10- import type { FastifyReply , FastifyRequest } from "fastify" ;
11- import multer from "multer" ;
8+ import type ExpressMulter from "multer" ;
129
1310import type { IRequestFormDataProps } from "../options/IRequestFormDataProps" ;
1411import { Singleton } from "../utils/Singleton" ;
@@ -35,13 +32,13 @@ import { validate_request_form_data } from "./internal/validate_request_form_dat
3532 * 3. Only `boolean`, `bigint`, `number`, `string`, `Blob`, `File` or their array types are allowed
3633 * 4. By the way, union type never be not allowed
3734 *
38- * By the way, if you're using `fastify`, you have to setup `@ fastify/multipart `
35+ * By the way, if you're using `fastify`, you have to setup `fastify-multer `
3936 * and configure like below when composing the NestJS application. If you don't do
4037 * that, `@TypedFormData.Body()` will not work properly, and throw 500 internal
4138 * server error when `Blob` or `File` type being utilized.
4239 *
4340 * ```typescript
44- * import multipart from "fastify-multipart ";
41+ * import fastifyMulter from "fastify-multer ";
4542 * import { NestFactory } from "@nestjs/core";
4643 * import {
4744 * FastifyAdapter,
@@ -53,7 +50,7 @@ import { validate_request_form_data } from "./internal/validate_request_form_dat
5350 * AppModule,
5451 * new FastifyAdapter(),
5552 * );
56- * app.register(multipart );
53+ * app.register(fastifyMulter.contentParser );
5754 * await app.listen(3000);
5855 * }
5956 * ```
@@ -69,20 +66,29 @@ export namespace TypedFormData {
6966 *
7067 * Much easier and type safer than `@nest.UploadFile()` decorator.
7168 *
72- * @param props Automatically filled by transformer
69+ * @param factory Factory function ncreating the `multer` or `fastify-multer`
70+ * instance. In the factory function, you also can specify the
71+ * multer composition options like `storage` engine.
72+ */
73+ export function Body < Multer extends IMulterBase > (
74+ factory : ( ) => Multer | Promise < Multer > ,
75+ ) : ParameterDecorator ;
76+
77+ /**
78+ * @internal
7379 */
7480 export function Body < T extends object > (
75- props ?: IRequestFormDataProps < T > ,
81+ factory : ( ) => Promise < IMulterBase > ,
82+ props ?: IRequestFormDataProps < T > | undefined ,
7683 ) : ParameterDecorator {
7784 if ( typeof File === "undefined" )
7885 throw new Error (
7986 "Error on TypedFormData.Body(): 'File' class is not supported in the older version of NodeJS. Upgrade the NodeJS to the modern." ,
8087 ) ;
8188 const checker = validate_request_form_data ( props ) ;
82- const predicator = ( type : "express" | "fastify" ) =>
83- new Singleton ( ( ) =>
84- type === "express" ? decodeExpress ( props ! ) : decodeFastify ( props ! ) ,
85- ) ;
89+ const uploader = new Singleton ( async ( ) =>
90+ decode ( ( await factory ( ) ) as ExpressMulter . Multer , props ! ) ,
91+ ) ;
8692 return createParamDecorator ( async function TypedFormDataBody (
8793 _unknown : any ,
8894 context : ExecutionContext ,
@@ -93,11 +99,9 @@ export namespace TypedFormData {
9399 throw new BadRequestException (
94100 `Request body type is not "multipart/form-data".` ,
95101 ) ;
96-
97- const decoder = isExpressRequest ( request )
98- ? predicator ( "express" ) . get ( )
99- : predicator ( "fastify" ) . get ( ) ;
100- const data : FormData = await decoder ( {
102+ const data : FormData = await (
103+ await uploader . get ( )
104+ ) ( {
101105 request : request as any ,
102106 response : http . getResponse ( ) ,
103107 } ) ;
@@ -106,13 +110,27 @@ export namespace TypedFormData {
106110 return output ;
107111 } ) ( ) ;
108112 }
113+
114+ /**
115+ * Base type of the `multer` or `fastify-multer`.
116+ */
117+ export interface IMulterBase {
118+ single ( fieldName : string ) : any ;
119+ array ( fieldName : string , maxCount ?: number ) : any ;
120+ fields ( fields : readonly object [ ] ) : any ;
121+ any ( ) : any ;
122+ none ( ) : any ;
123+ }
109124}
110125
111126/**
112127 * @internal
113128 */
114- const decodeExpress = < T > ( props : IRequestFormDataProps < T > ) => {
115- const upload = multerApplication . get ( ) . fields (
129+ const decode = < T > (
130+ multer : ExpressMulter . Multer ,
131+ props : IRequestFormDataProps < T > ,
132+ ) => {
133+ const upload = multer . fields (
116134 props ! . files . map ( ( file ) => ( {
117135 name : file . name ,
118136 ...( file . limit === 1 ? { maxCount : 1 } : { } ) ,
@@ -141,40 +159,6 @@ const decodeExpress = <T>(props: IRequestFormDataProps<T>) => {
141159 } ;
142160} ;
143161
144- /**
145- * @internal
146- */
147- const decodeFastify =
148- < T > ( _props : IRequestFormDataProps < T > ) =>
149- async ( socket : {
150- request : FastifyRequest & {
151- parts ?( ) : AsyncIterableIterator < Multipart > ;
152- } ;
153- response : FastifyReply ;
154- } ) : Promise < FormData > => {
155- if (
156- socket . request . files === undefined ||
157- typeof socket . request . files !== "function"
158- )
159- throw new InternalServerErrorException (
160- "Have not configured the `fastify-multipart` plugin yet. Inquiry to the backend developer." ,
161- ) ;
162- const data : FormData = new FormData ( ) ;
163- for await ( const part of socket . request . parts ( ) )
164- if ( part . type === "file" )
165- data . append (
166- part . fieldname ,
167- new File ( [ await part . toBuffer ( ) ] , part . filename , {
168- type : part . mimetype ,
169- } ) ,
170- ) ;
171- else if ( Array . isArray ( part . value ) )
172- for ( const elem of part . value )
173- data . append ( part . fieldname , String ( elem ) ) ;
174- else data . append ( part . fieldname , String ( part . value ) ) ;
175- return data ;
176- } ;
177-
178162/**
179163 * @internal
180164 */
@@ -209,15 +193,3 @@ const isMultipartFormData = (text?: string): boolean =>
209193 . split ( ";" )
210194 . map ( ( str ) => str . trim ( ) )
211195 . some ( ( str ) => str === "multipart/form-data" ) ;
212-
213- /**
214- * @internal
215- */
216- const isExpressRequest = (
217- request : express . Request | FastifyRequest ,
218- ) : request is express . Request => ( request as express . Request ) . app !== undefined ;
219-
220- /**
221- * @internal
222- */
223- const multerApplication = new Singleton ( ( ) => multer ( ) ) ;
0 commit comments