-
-
Notifications
You must be signed in to change notification settings - Fork 87
/
classes.ts
69 lines (62 loc) · 2.15 KB
/
classes.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
57
58
59
60
61
62
63
64
65
66
67
68
69
import {
ApplyMetadataFn,
Constructor,
defaultStrategyInitializerOptions,
Dictionary,
Mapper,
MappingStrategyInitializer,
MappingStrategyInitializerOptions,
MetadataIdentifier,
} from '@automapper/core';
import 'reflect-metadata';
import { getMetadataList } from './get-metadata-list';
export function classes(
options: MappingStrategyInitializerOptions = {}
): MappingStrategyInitializer<Constructor> {
const {
destinationConstructor = (
_: Dictionary<object>,
destinationIdentifier: MetadataIdentifier
) => new (destinationIdentifier as Constructor)(),
applyMetadata,
postMap,
preMap,
} = { ...defaultStrategyInitializerOptions, ...options };
const metadataTracker = new Set();
return (mapper: Mapper) => ({
destinationConstructor,
mapper,
get applyMetadata(): ApplyMetadataFn {
return applyMetadata(this);
},
retrieveMetadata(...identifiers) {
const metadataMap = new Map();
for (let i = 0, length = identifiers.length; i < length; i++) {
const identifier = identifiers[i];
if (metadataTracker.has(identifier)) {
continue;
}
const [metadataList, nestedConstructors] =
getMetadataList(identifier);
metadataMap.set(identifier, metadataList);
metadataTracker.add(identifier);
if (nestedConstructors.length) {
const nestedConstructorsMetadataMap = this.retrieveMetadata(
...nestedConstructors
);
nestedConstructorsMetadataMap.forEach(
(nestedConstructorMetadataList, nestedConstructor) => {
metadataMap.set(
nestedConstructor,
nestedConstructorMetadataList
);
}
);
}
}
return metadataMap;
},
preMap,
postMap,
});
}