rudeshark.net/src/server/api/endpoints/i/notifications.ts

93 lines
2.3 KiB
TypeScript
Raw Normal View History

2018-07-07 12:19:00 +02:00
import $ from 'cafy'; import ID from '../../../../misc/cafy-id';
2018-03-29 13:32:18 +02:00
import Notification from '../../../../models/notification';
import Mute from '../../../../models/mute';
import { pack } from '../../../../models/notification';
2018-04-19 05:43:25 +02:00
import { getFriendIds } from '../../common/get-friends';
import read from '../../common/read-notification';
2018-06-18 02:54:53 +02:00
import { ILocalUser } from '../../../../models/user';
2016-12-28 23:49:51 +01:00
/**
* Get notifications
*/
2018-07-05 19:58:29 +02:00
export default (params: any, user: ILocalUser) => new Promise(async (res, rej) => {
2016-12-28 23:49:51 +01:00
// Get 'following' parameter
2017-03-05 04:00:39 +01:00
const [following = false, followingError] =
2018-07-05 16:36:07 +02:00
$.bool.optional.get(params.following);
2017-03-03 00:56:07 +01:00
if (followingError) return rej('invalid following param');
2016-12-28 23:49:51 +01:00
2018-03-29 07:48:47 +02:00
// Get 'markAsRead' parameter
2018-07-05 16:36:07 +02:00
const [markAsRead = true, markAsReadErr] = $.bool.optional.get(params.markAsRead);
2018-03-29 07:48:47 +02:00
if (markAsReadErr) return rej('invalid markAsRead param');
2016-12-28 23:49:51 +01:00
// Get 'limit' parameter
2018-07-05 16:36:07 +02:00
const [limit = 10, limitErr] = $.num.optional.range(1, 100).get(params.limit);
2017-03-03 00:56:07 +01:00
if (limitErr) return rej('invalid limit param');
2016-12-28 23:49:51 +01:00
2018-03-29 07:48:47 +02:00
// Get 'sinceId' parameter
2018-07-05 16:36:07 +02:00
const [sinceId, sinceIdErr] = $.type(ID).optional.get(params.sinceId);
2018-03-29 07:48:47 +02:00
if (sinceIdErr) return rej('invalid sinceId param');
2017-03-03 00:56:07 +01:00
2018-03-29 07:48:47 +02:00
// Get 'untilId' parameter
2018-07-05 16:36:07 +02:00
const [untilId, untilIdErr] = $.type(ID).optional.get(params.untilId);
2018-03-29 07:48:47 +02:00
if (untilIdErr) return rej('invalid untilId param');
2016-12-28 23:49:51 +01:00
2018-03-29 07:48:47 +02:00
// Check if both of sinceId and untilId is specified
2017-12-20 18:20:02 +01:00
if (sinceId && untilId) {
2018-03-29 07:48:47 +02:00
return rej('cannot set sinceId and untilId');
2016-12-28 23:49:51 +01:00
}
2017-12-21 23:26:23 +01:00
const mute = await Mute.find({
2018-08-02 02:27:30 +02:00
muterId: user._id
2017-12-21 23:26:23 +01:00
});
2016-12-28 23:49:51 +01:00
const query = {
2018-03-29 07:48:47 +02:00
notifieeId: user._id,
2017-12-21 23:26:23 +01:00
$and: [{
2018-03-29 07:48:47 +02:00
notifierId: {
$nin: mute.map(m => m.muteeId)
2017-12-21 23:26:23 +01:00
}
}]
2017-03-03 00:56:07 +01:00
} as any;
2016-12-28 23:49:51 +01:00
const sort = {
_id: -1
};
if (following) {
2017-12-21 23:26:23 +01:00
// ID list of the user itself and other users who the user follows
2018-04-19 05:43:25 +02:00
const followingIds = await getFriendIds(user._id);
2016-12-28 23:49:51 +01:00
2017-12-21 23:26:23 +01:00
query.$and.push({
2018-03-29 07:48:47 +02:00
notifierId: {
2017-12-21 23:26:23 +01:00
$in: followingIds
}
});
2016-12-28 23:49:51 +01:00
}
2017-03-03 00:56:07 +01:00
if (sinceId) {
2016-12-28 23:49:51 +01:00
sort._id = 1;
query._id = {
2017-03-03 00:56:07 +01:00
$gt: sinceId
2016-12-28 23:49:51 +01:00
};
2017-12-20 18:20:02 +01:00
} else if (untilId) {
2016-12-28 23:49:51 +01:00
query._id = {
2017-12-20 18:20:02 +01:00
$lt: untilId
2016-12-28 23:49:51 +01:00
};
}
// Issue query
const notifications = await Notification
2017-01-17 03:11:22 +01:00
.find(query, {
2016-12-28 23:49:51 +01:00
limit: limit,
sort: sort
2017-01-17 03:11:22 +01:00
});
2016-12-28 23:49:51 +01:00
// Serialize
2018-05-22 04:39:48 +02:00
res(await Promise.all(notifications.map(notification => pack(notification))));
2016-12-28 23:49:51 +01:00
// Mark all as read
2016-12-28 23:49:51 +01:00
if (notifications.length > 0 && markAsRead) {
read(user._id, notifications);
2016-12-28 23:49:51 +01:00
}
});