-
Notifications
You must be signed in to change notification settings - Fork 469
/
response-object-mapper.ts
50 lines (46 loc) · 1.18 KB
/
response-object-mapper.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
import { omit } from 'lodash';
import { ApiResponseSchemaHost } from '../decorators';
import { getSchemaPath } from '../utils';
import { MimetypeContentWrapper } from './mimetype-content-wrapper';
export class ResponseObjectMapper {
private readonly mimetypeContentWrapper = new MimetypeContentWrapper();
toArrayRefObject(
response: Record<string, any>,
name: string,
produces: string[]
) {
return {
...response,
...this.mimetypeContentWrapper.wrap(produces, {
schema: {
type: 'array',
items: {
$ref: getSchemaPath(name)
}
}
})
};
}
toRefObject(response: Record<string, any>, name: string, produces: string[]) {
return {
...response,
...this.mimetypeContentWrapper.wrap(produces, {
schema: {
$ref: getSchemaPath(name)
}
})
};
}
wrapSchemaWithContent(response: ApiResponseSchemaHost, produces: string[]) {
if (!response.schema) {
return response;
}
const content = this.mimetypeContentWrapper.wrap(produces, {
schema: response.schema
});
return {
...omit(response, 'schema'),
...content
};
}
}