import { EntityRepository, Repository } from 'typeorm'; import { UserGroupInvitation } from '../entities/user-group-invitation'; import { UserGroups } from '..'; @EntityRepository(UserGroupInvitation) export class UserGroupInvitationRepository extends Repository { public async pack( src: UserGroupInvitation['id'] | UserGroupInvitation, ) { const invitation = typeof src === 'object' ? src : await this.findOneOrFail(src); return { id: invitation.id, group: await UserGroups.pack(invitation.userGroup || invitation.userGroupId), }; } public packMany( invitations: any[], ) { return Promise.all(invitations.map(x => this.pack(x))); } }