MisskeyIO/src/services/note/read.ts

import { publishMainStream } from '../stream';
import { Note } from '../../models/entities/note';
import { User } from '../../models/entities/user';
import { NoteUnreads, Antennas, AntennaNotes, Users } from '../../models';
import { Not, IsNull, In } from 'typeorm';

/**
 * Mark notes as read
 */
export default async function(
	userId: User['id'],
	noteIds: Note['id'][]
) {
	async function careNoteUnreads() {
		// Remove the record
		await NoteUnreads.delete({
			userId: userId,
			noteId: In(noteIds),
		});

		NoteUnreads.count({
			userId: userId,
			isMentioned: true
		}).then(mentionsCount => {
			if (mentionsCount === 0) {
				// 全て既読になったイベントを発行
				publishMainStream(userId, 'readAllUnreadMentions');
			}
		});

		NoteUnreads.count({
			userId: userId,
			isSpecified: true
		}).then(specifiedCount => {
			if (specifiedCount === 0) {
				// 全て既読になったイベントを発行
				publishMainStream(userId, 'readAllUnreadSpecifiedNotes');
			}
		});

		NoteUnreads.count({
			userId: userId,
			noteChannelId: Not(IsNull())
		}).then(channelNoteCount => {
			if (channelNoteCount === 0) {
				// 全て既読になったイベントを発行
				publishMainStream(userId, 'readAllChannels');
			}
		});
	}

	async function careAntenna() {
		const antennas = await Antennas.find({ userId });

		await Promise.all(antennas.map(async antenna => {
			const countBefore = await AntennaNotes.count({
				antennaId: antenna.id,
				read: false
			});

			if (countBefore === 0) return;

			await AntennaNotes.update({
				antennaId: antenna.id,
				noteId: In(noteIds)
			}, {
				read: true
			});

			const countAfter = await AntennaNotes.count({
				antennaId: antenna.id,
				read: false
			});

			if (countAfter === 0) {
				publishMainStream(userId, 'readAntenna', antenna);
			}
		}));

		Users.getHasUnreadAntenna(userId).then(unread => {
			if (!unread) {
				publishMainStream(userId, 'readAllAntennas');
			}
		});
	}

	careNoteUnreads();
	careAntenna();
}