/
file-fields-interceptor.ts
56 lines (46 loc) · 1.35 KB
/
file-fields-interceptor.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
import { Observable, tap } from "rxjs";
import {
CallHandler,
ExecutionContext,
mixin,
NestInterceptor,
Type,
} from "@nestjs/common";
import { getMultipartRequest } from "../multipart/request";
import { transformUploadOptions, UploadOptions } from "../multipart/options";
import {
handleMultipartFileFields,
UploadField,
UploadFieldMapEntry,
uploadFieldsToMap,
} from "../multipart/handlers/file-fields";
export function FileFieldsInterceptor(
uploadFields: UploadField[],
options?: UploadOptions,
): Type<NestInterceptor> {
class MixinInterceptor implements NestInterceptor {
private readonly options: UploadOptions;
private readonly fieldsMap: Map<string, UploadFieldMapEntry>;
constructor() {
this.options = transformUploadOptions(options);
this.fieldsMap = uploadFieldsToMap(uploadFields);
}
async intercept(
context: ExecutionContext,
next: CallHandler,
): Promise<Observable<any>> {
const ctx = context.switchToHttp();
const req = getMultipartRequest(ctx);
const { body, files, remove } = await handleMultipartFileFields(
req,
this.fieldsMap,
this.options,
);
req.body = body;
req.storageFiles = files;
return next.handle().pipe(tap(remove));
}
}
const Interceptor = mixin(MixinInterceptor);
return Interceptor;
}