import { EntityRepository, Repository } from 'typeorm'; import { Users, Notes } from '..'; import { Notification } from '../entities/notification'; import { ensure } from '../../prelude/ensure'; import { awaitAll } from '../../prelude/await-all'; import { SchemaType } from '../../misc/schema'; export type PackedNotification = SchemaType; @EntityRepository(Notification) export class NotificationRepository extends Repository { public async pack( src: Notification['id'] | Notification, ): Promise { const notification = typeof src === 'object' ? src : await this.findOne(src).then(ensure); return await awaitAll({ id: notification.id, createdAt: notification.createdAt.toISOString(), type: notification.type, userId: notification.notifierId, user: Users.pack(notification.notifier || notification.notifierId), ...(notification.type === 'mention' ? { note: Notes.pack(notification.note || notification.noteId!), } : {}), ...(notification.type === 'reply' ? { note: Notes.pack(notification.note || notification.noteId!), } : {}), ...(notification.type === 'renote' ? { note: Notes.pack(notification.note || notification.noteId!), } : {}), ...(notification.type === 'quote' ? { note: Notes.pack(notification.note || notification.noteId!), } : {}), ...(notification.type === 'reaction' ? { note: Notes.pack(notification.note || notification.noteId!), reaction: notification.reaction } : {}), ...(notification.type === 'pollVote' ? { note: Notes.pack(notification.note || notification.noteId!), choice: notification.choice } : {}) }); } public packMany( notifications: any[], ) { return Promise.all(notifications.map(x => this.pack(x))); } } export const packedNotificationSchema = { type: 'object' as const, optional: false as const, nullable: false as const, properties: { id: { type: 'string' as const, optional: false as const, nullable: false as const, format: 'id', description: 'The unique identifier for this notification.', example: 'xxxxxxxxxx', }, createdAt: { type: 'string' as const, optional: false as const, nullable: false as const, format: 'date-time', description: 'The date that the notification was created.' }, type: { type: 'string' as const, optional: false as const, nullable: false as const, enum: ['follow', 'receiveFollowRequest', 'mention', 'reply', 'renote', 'quote', 'reaction', 'pollVote'], description: 'The type of the notification.' }, userId: { type: 'string' as const, optional: true as const, nullable: true as const, format: 'id', }, user: { type: 'object' as const, ref: 'User', optional: true as const, nullable: true as const, }, } };