104 lines
2.5 KiB
TypeScript
104 lines
2.5 KiB
TypeScript
import $ from 'cafy'; import ID from '../../../../misc/cafy-id';
|
|
import Notification from '../../../../models/notification';
|
|
import Mute from '../../../../models/mute';
|
|
import { pack } from '../../../../models/notification';
|
|
import { getFriendIds } from '../../common/get-friends';
|
|
import read from '../../common/read-notification';
|
|
import { ILocalUser } from '../../../../models/user';
|
|
|
|
/**
|
|
* Get notifications
|
|
*/
|
|
export default (params: any, user: ILocalUser) => new Promise(async (res, rej) => {
|
|
// Get 'following' parameter
|
|
const [following = false, followingError] =
|
|
$.bool.optional.get(params.following);
|
|
if (followingError) return rej('invalid following param');
|
|
|
|
// Get 'markAsRead' parameter
|
|
const [markAsRead = true, markAsReadErr] = $.bool.optional.get(params.markAsRead);
|
|
if (markAsReadErr) return rej('invalid markAsRead param');
|
|
|
|
// Get 'type' parameter
|
|
const [type, typeErr] = $.arr($.str).optional.unique().get(params.type);
|
|
if (typeErr) return rej('invalid type param');
|
|
|
|
// Get 'limit' parameter
|
|
const [limit = 10, limitErr] = $.num.optional.range(1, 100).get(params.limit);
|
|
if (limitErr) return rej('invalid limit param');
|
|
|
|
// Get 'sinceId' parameter
|
|
const [sinceId, sinceIdErr] = $.type(ID).optional.get(params.sinceId);
|
|
if (sinceIdErr) return rej('invalid sinceId param');
|
|
|
|
// Get 'untilId' parameter
|
|
const [untilId, untilIdErr] = $.type(ID).optional.get(params.untilId);
|
|
if (untilIdErr) return rej('invalid untilId param');
|
|
|
|
// Check if both of sinceId and untilId is specified
|
|
if (sinceId && untilId) {
|
|
return rej('cannot set sinceId and untilId');
|
|
}
|
|
|
|
const mute = await Mute.find({
|
|
muterId: user._id,
|
|
deletedAt: { $exists: false }
|
|
});
|
|
|
|
const query = {
|
|
notifieeId: user._id,
|
|
$and: [{
|
|
notifierId: {
|
|
$nin: mute.map(m => m.muteeId)
|
|
}
|
|
}]
|
|
} as any;
|
|
|
|
const sort = {
|
|
_id: -1
|
|
};
|
|
|
|
if (following) {
|
|
// ID list of the user itself and other users who the user follows
|
|
const followingIds = await getFriendIds(user._id);
|
|
|
|
query.$and.push({
|
|
notifierId: {
|
|
$in: followingIds
|
|
}
|
|
});
|
|
}
|
|
|
|
if (type) {
|
|
query.type = {
|
|
$in: type
|
|
};
|
|
}
|
|
|
|
if (sinceId) {
|
|
sort._id = 1;
|
|
query._id = {
|
|
$gt: sinceId
|
|
};
|
|
} else if (untilId) {
|
|
query._id = {
|
|
$lt: untilId
|
|
};
|
|
}
|
|
|
|
// Issue query
|
|
const notifications = await Notification
|
|
.find(query, {
|
|
limit: limit,
|
|
sort: sort
|
|
});
|
|
|
|
// Serialize
|
|
res(await Promise.all(notifications.map(notification => pack(notification))));
|
|
|
|
// Mark all as read
|
|
if (notifications.length > 0 && markAsRead) {
|
|
read(user._id, notifications);
|
|
}
|
|
});
|