133 lines
3.8 KiB
TypeScript
133 lines
3.8 KiB
TypeScript
import { publishMainStream } from '@/services/stream';
|
|
import { Note } from '@/models/entities/note';
|
|
import { User } from '@/models/entities/user';
|
|
import { NoteUnreads, AntennaNotes, Users, Followings, ChannelFollowings } from '@/models/index';
|
|
import { Not, IsNull, In } from 'typeorm';
|
|
import { Channel } from '@/models/entities/channel';
|
|
import { checkHitAntenna } from '@/misc/check-hit-antenna';
|
|
import { getAntennas } from '@/misc/antenna-cache';
|
|
import { readNotificationByQuery } from '@/server/api/common/read-notification';
|
|
import { Packed } from '@/misc/schema';
|
|
|
|
/**
|
|
* Mark notes as read
|
|
*/
|
|
export default async function(
|
|
userId: User['id'],
|
|
notes: (Note | Packed<'Note'>)[],
|
|
info?: {
|
|
following: Set<User['id']>;
|
|
followingChannels: Set<Channel['id']>;
|
|
}
|
|
) {
|
|
const following = info?.following ? info.following : new Set<string>((await Followings.find({
|
|
where: {
|
|
followerId: userId
|
|
},
|
|
select: ['followeeId']
|
|
})).map(x => x.followeeId));
|
|
const followingChannels = info?.followingChannels ? info.followingChannels : new Set<string>((await ChannelFollowings.find({
|
|
where: {
|
|
followerId: userId
|
|
},
|
|
select: ['followeeId']
|
|
})).map(x => x.followeeId));
|
|
|
|
const myAntennas = (await getAntennas()).filter(a => a.userId === userId);
|
|
const readMentions: (Note | Packed<'Note'>)[] = [];
|
|
const readSpecifiedNotes: (Note | Packed<'Note'>)[] = [];
|
|
const readChannelNotes: (Note | Packed<'Note'>)[] = [];
|
|
const readAntennaNotes: (Note | Packed<'Note'>)[] = [];
|
|
|
|
for (const note of notes) {
|
|
if (note.mentions && note.mentions.includes(userId)) {
|
|
readMentions.push(note);
|
|
} else if (note.visibleUserIds && note.visibleUserIds.includes(userId)) {
|
|
readSpecifiedNotes.push(note);
|
|
}
|
|
|
|
if (note.channelId && followingChannels.has(note.channelId)) {
|
|
readChannelNotes.push(note);
|
|
}
|
|
|
|
if (note.user != null) { // たぶんnullになることは無いはずだけど一応
|
|
for (const antenna of myAntennas) {
|
|
if (await checkHitAntenna(antenna, note, note.user as any, undefined, Array.from(following))) {
|
|
readAntennaNotes.push(note);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if ((readMentions.length > 0) || (readSpecifiedNotes.length > 0) || (readChannelNotes.length > 0)) {
|
|
// Remove the record
|
|
await NoteUnreads.delete({
|
|
userId: userId,
|
|
noteId: In([...readMentions.map(n => n.id), ...readSpecifiedNotes.map(n => n.id), ...readChannelNotes.map(n => n.id)]),
|
|
});
|
|
|
|
// TODO: ↓まとめてクエリしたい
|
|
|
|
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');
|
|
}
|
|
});
|
|
|
|
readNotificationByQuery(userId, {
|
|
noteId: In([...readMentions.map(n => n.id), ...readSpecifiedNotes.map(n => n.id)]),
|
|
});
|
|
}
|
|
|
|
if (readAntennaNotes.length > 0) {
|
|
await AntennaNotes.update({
|
|
antennaId: In(myAntennas.map(a => a.id)),
|
|
noteId: In(readAntennaNotes.map(n => n.id))
|
|
}, {
|
|
read: true
|
|
});
|
|
|
|
// TODO: まとめてクエリしたい
|
|
for (const antenna of myAntennas) {
|
|
const count = await AntennaNotes.count({
|
|
antennaId: antenna.id,
|
|
read: false
|
|
});
|
|
|
|
if (count === 0) {
|
|
publishMainStream(userId, 'readAntenna', antenna);
|
|
}
|
|
}
|
|
|
|
Users.getHasUnreadAntenna(userId).then(unread => {
|
|
if (!unread) {
|
|
publishMainStream(userId, 'readAllAntennas');
|
|
}
|
|
});
|
|
}
|
|
}
|