import { EntityRepository, Repository } from 'typeorm'; import { Users } from '../index'; import { Following } from '@/models/entities/following'; import { awaitAll } from '@/prelude/await-all'; import { Packed } from '@/misc/schema'; import { User } from '@/models/entities/user'; type LocalFollowerFollowing = Following & { followerHost: null; followerInbox: null; followerSharedInbox: null; }; type RemoteFollowerFollowing = Following & { followerHost: string; followerInbox: string; followerSharedInbox: string; }; type LocalFolloweeFollowing = Following & { followeeHost: null; followeeInbox: null; followeeSharedInbox: null; }; type RemoteFolloweeFollowing = Following & { followeeHost: string; followeeInbox: string; followeeSharedInbox: string; }; @EntityRepository(Following) export class FollowingRepository extends Repository { public isLocalFollower(following: Following): following is LocalFollowerFollowing { return following.followerHost == null; } public isRemoteFollower(following: Following): following is RemoteFollowerFollowing { return following.followerHost != null; } public isLocalFollowee(following: Following): following is LocalFolloweeFollowing { return following.followeeHost == null; } public isRemoteFollowee(following: Following): following is RemoteFolloweeFollowing { return following.followeeHost != null; } public async pack( src: Following['id'] | Following, me?: { id: User['id'] } | null | undefined, opts?: { populateFollowee?: boolean; populateFollower?: boolean; } ): Promise> { const following = typeof src === 'object' ? src : await this.findOneOrFail(src); if (opts == null) opts = {}; return await awaitAll({ id: following.id, createdAt: following.createdAt.toISOString(), followeeId: following.followeeId, followerId: following.followerId, followee: opts.populateFollowee ? Users.pack(following.followee || following.followeeId, me, { detail: true, }) : undefined, follower: opts.populateFollower ? Users.pack(following.follower || following.followerId, me, { detail: true, }) : undefined, }); } public packMany( followings: any[], me?: { id: User['id'] } | null | undefined, opts?: { populateFollowee?: boolean; populateFollower?: boolean; } ) { return Promise.all(followings.map(x => this.pack(x, me, opts))); } }