misskey/src/services/note/read.ts

68 lines
1.4 KiB
TypeScript
Raw Normal View History

2018-09-19 07:18:34 +02:00
import * as mongo from 'mongodb';
2018-10-16 04:38:09 +02:00
import isObjectId from '../../misc/is-objectid';
import { publishMainStream } from '../../stream';
2018-09-19 07:18:34 +02:00
import User from '../../models/user';
import NoteUnread from '../../models/note-unread';
/**
* Mark a note as read
*/
export default (
user: string | mongo.ObjectID,
note: string | mongo.ObjectID
) => new Promise<any>(async (resolve, reject) => {
2018-10-16 04:38:09 +02:00
const userId: mongo.ObjectID = isObjectId(user)
2018-09-19 07:18:34 +02:00
? user as mongo.ObjectID
: new mongo.ObjectID(user);
2018-10-16 04:38:09 +02:00
const noteId: mongo.ObjectID = isObjectId(note)
2018-09-19 07:18:34 +02:00
? note as mongo.ObjectID
: new mongo.ObjectID(note);
// Remove document
2018-09-23 09:05:26 +02:00
const res = await NoteUnread.remove({
2018-09-19 07:18:34 +02:00
userId: userId,
noteId: noteId
});
2018-09-23 09:05:26 +02:00
if (res.deletedCount == 0) {
return;
}
2018-09-19 07:18:34 +02:00
const count1 = await NoteUnread
.count({
userId: userId,
isSpecified: false
}, {
limit: 1
});
const count2 = await NoteUnread
.count({
userId: userId,
isSpecified: true
}, {
limit: 1
});
if (count1 == 0 || count2 == 0) {
User.update({ _id: userId }, {
$set: {
hasUnreadMentions: count1 != 0 || count2 != 0,
hasUnreadSpecifiedNotes: count2 != 0
}
});
}
if (count1 == 0) {
// 全て既読になったイベントを発行
publishMainStream(userId, 'readAllUnreadMentions');
2018-09-19 07:18:34 +02:00
}
if (count2 == 0) {
// 全て既読になったイベントを発行
publishMainStream(userId, 'readAllUnreadSpecifiedNotes');
2018-09-19 07:18:34 +02:00
}
});