1 |
|
2 |
|
3 |
|
4 |
|
5 |
|
6 | import {Getter} from '@loopback/core';
|
7 | import {cloneDeep} from 'lodash';
|
8 | import {EntityNotFoundError, TypeResolver} from '../../';
|
9 | import {DataObject, Options} from '../../common-types';
|
10 | import {Entity} from '../../model';
|
11 | import {constrainFilter, EntityCrudRepository} from '../../repositories';
|
12 |
|
13 |
|
14 |
|
15 | export interface BelongsToRepository<Target extends Entity> {
|
16 | |
17 |
|
18 |
|
19 |
|
20 |
|
21 |
|
22 |
|
23 |
|
24 |
|
25 |
|
26 | get(
|
27 | options?: Options & {polymorphicType?: string | string[]},
|
28 | ): Promise<Target>;
|
29 | }
|
30 |
|
31 | export class DefaultBelongsToRepository<
|
32 | TargetEntity extends Entity,
|
33 | TargetId,
|
34 | TargetRepository extends EntityCrudRepository<TargetEntity, TargetId>,
|
35 | > implements BelongsToRepository<TargetEntity>
|
36 | {
|
37 | |
38 |
|
39 |
|
40 |
|
41 |
|
42 |
|
43 |
|
44 |
|
45 |
|
46 |
|
47 |
|
48 |
|
49 |
|
50 | constructor(
|
51 | public getTargetRepository:
|
52 | | Getter<TargetRepository>
|
53 | | {
|
54 | [repoType: string]: Getter<TargetRepository>;
|
55 | },
|
56 | public constraint: DataObject<TargetEntity>,
|
57 | public targetResolver: TypeResolver<Entity, typeof Entity>,
|
58 | ) {
|
59 | if (typeof getTargetRepository === 'function') {
|
60 | this.getTargetRepositoryDict = {
|
61 | [targetResolver().name]:
|
62 | getTargetRepository as Getter<TargetRepository>,
|
63 | };
|
64 | } else {
|
65 | this.getTargetRepositoryDict = getTargetRepository as {
|
66 | [repoType: string]: Getter<TargetRepository>;
|
67 | };
|
68 | }
|
69 | }
|
70 |
|
71 | public getTargetRepositoryDict: {
|
72 | [repoType: string]: Getter<TargetRepository>;
|
73 | };
|
74 |
|
75 | async get(
|
76 | options?: Options & {polymorphicType?: string | string[]},
|
77 | ): Promise<TargetEntity> {
|
78 | let polymorphicTypes = options?.polymorphicType;
|
79 | let allKeys: string[];
|
80 | if (Object.keys(this.getTargetRepositoryDict).length <= 1) {
|
81 | allKeys = Object.keys(this.getTargetRepositoryDict);
|
82 | } else if (!polymorphicTypes || polymorphicTypes.length === 0) {
|
83 | console.warn(
|
84 | 'It is highly recommended to specify the polymorphicTypes param when using polymorphic types.',
|
85 | );
|
86 | allKeys = Object.keys(this.getTargetRepositoryDict);
|
87 | } else {
|
88 | if (typeof polymorphicTypes === 'string') {
|
89 | polymorphicTypes = [polymorphicTypes];
|
90 | }
|
91 | allKeys = [];
|
92 | new Set(polymorphicTypes!).forEach(element => {
|
93 | if (Object.keys(this.getTargetRepositoryDict).includes(element)) {
|
94 | allKeys.push(element);
|
95 | }
|
96 | });
|
97 | }
|
98 | let result: TargetEntity[] = [];
|
99 | for (const key of allKeys) {
|
100 | const targetRepository = await this.getTargetRepositoryDict[key]();
|
101 | result = result.concat(
|
102 | await targetRepository.find(
|
103 | constrainFilter(undefined, this.constraint),
|
104 | Object.assign(cloneDeep(options ?? {}), {polymorphicType: key}),
|
105 | ),
|
106 | );
|
107 | if (result.length >= 1) {
|
108 | return result[0];
|
109 | }
|
110 | }
|
111 |
|
112 | const id = 'constraint ' + JSON.stringify(this.constraint);
|
113 | throw new EntityNotFoundError(this.targetResolver().name, id);
|
114 | }
|
115 | }
|