Sharkey/packages/backend/src/core/NotificationService.ts

71 lines
2.2 KiB
TypeScript
Raw Normal View History

2022-09-17 21:27:08 +03:00
import { Inject, Injectable } from '@nestjs/common';
import { In } from 'typeorm';
import { DI } from '@/di-symbols.js';
2022-09-20 23:33:11 +03:00
import type { NotificationsRepository } from '@/models/index.js';
2022-09-17 21:27:08 +03:00
import type { User } from '@/models/entities/User.js';
import type { Notification } from '@/models/entities/Notification.js';
2022-12-04 03:16:03 +02:00
import { UserEntityService } from '@/core/entities/UserEntityService.js';
2023-02-17 08:15:36 +02:00
import { bindThis } from '@/decorators.js';
2022-09-17 21:27:08 +03:00
import { GlobalEventService } from './GlobalEventService.js';
import { PushNotificationService } from './PushNotificationService.js';
@Injectable()
export class NotificationService {
constructor(
@Inject(DI.notificationsRepository)
private notificationsRepository: NotificationsRepository,
private userEntityService: UserEntityService,
private globalEventService: GlobalEventService,
private pushNotificationService: PushNotificationService,
) {
}
@bindThis
2022-09-17 21:27:08 +03:00
public async readNotification(
userId: User['id'],
notificationIds: Notification['id'][],
) {
if (notificationIds.length === 0) return;
// Update documents
const result = await this.notificationsRepository.update({
notifieeId: userId,
id: In(notificationIds),
isRead: false,
}, {
isRead: true,
});
if (result.affected === 0) return;
2022-09-18 21:11:50 +03:00
if (!await this.userEntityService.getHasUnreadNotification(userId)) return this.postReadAllNotifications(userId);
else return this.postReadNotifications(userId, notificationIds);
2022-09-17 21:27:08 +03:00
}
@bindThis
2022-09-17 21:27:08 +03:00
public async readNotificationByQuery(
userId: User['id'],
query: Record<string, any>,
) {
const notificationIds = await this.notificationsRepository.findBy({
...query,
notifieeId: userId,
isRead: false,
}).then(notifications => notifications.map(notification => notification.id));
return this.readNotification(userId, notificationIds);
}
@bindThis
2022-09-18 21:11:50 +03:00
private postReadAllNotifications(userId: User['id']) {
2022-09-17 21:27:08 +03:00
this.globalEventService.publishMainStream(userId, 'readAllNotifications');
return this.pushNotificationService.pushNotification(userId, 'readAllNotifications', undefined);
}
@bindThis
2022-09-18 21:11:50 +03:00
private postReadNotifications(userId: User['id'], notificationIds: Notification['id'][]) {
2022-09-17 21:27:08 +03:00
return this.pushNotificationService.pushNotification(userId, 'readNotifications', { notificationIds });
}
}