Sharkey/src/models/notification.ts

120 lines
3.2 KiB
TypeScript
Raw Normal View History

import * as mongo from 'mongodb';
2018-06-18 03:54:53 +03:00
const deepcopy = require('deepcopy');
2018-03-29 14:32:18 +03:00
import db from '../db/mongodb';
2018-10-16 05:38:09 +03:00
import isObjectId from '../misc/is-objectid';
2018-02-02 01:06:01 +02:00
import { IUser, pack as packUser } from './user';
2018-04-07 20:30:37 +03:00
import { pack as packNote } from './note';
2017-01-17 02:17:52 +02:00
2018-02-02 01:06:01 +02:00
const Notification = db.get<INotification>('notifications');
2018-08-02 03:29:46 +03:00
Notification.createIndex('notifieeId');
2018-02-02 01:06:01 +02:00
export default Notification;
export interface INotification {
_id: mongo.ObjectID;
2018-03-29 08:48:47 +03:00
createdAt: Date;
2017-11-10 17:27:02 +02:00
/**
*
*/
notifiee?: IUser;
/**
*
*/
2018-03-29 08:48:47 +03:00
notifieeId: mongo.ObjectID;
2017-11-10 17:27:02 +02:00
/**
2017-11-10 19:25:24 +02:00
* (initiator)Origin
2017-11-10 17:27:02 +02:00
*/
notifier?: IUser;
/**
2017-11-10 19:25:24 +02:00
* (initiator)Origin
2017-11-10 17:27:02 +02:00
*/
2018-03-29 08:48:47 +03:00
notifierId: mongo.ObjectID;
2017-11-10 17:27:02 +02:00
/**
*
* follow -
* mention - 稿
* reply - (Watchしている)稿
2018-04-07 20:30:37 +03:00
* renote - (Watchしている)稿Renoteされた
* quote - (Watchしている)稿Renoteされた
2017-11-10 17:27:02 +02:00
* reaction - (Watchしている)稿
* poll_vote - (Watchしている)稿
*/
2018-04-07 20:30:37 +03:00
type: 'follow' | 'mention' | 'reply' | 'renote' | 'quote' | 'reaction' | 'poll_vote';
2017-11-10 17:27:02 +02:00
/**
*
*/
2018-03-29 08:48:47 +03:00
isRead: Boolean;
}
2018-02-02 01:06:01 +02:00
export const packMany = async (
notifications: any[]
) => {
2018-10-31 04:20:54 +02:00
return (await Promise.all(notifications.map(n => pack(n))));
};
2018-02-02 01:06:01 +02:00
/**
* Pack a notification for API response
*/
export const pack = (notification: any) => new Promise<any>(async (resolve, reject) => {
let _notification: any;
// Populate the notification if 'notification' is ID
2018-10-16 05:38:09 +03:00
if (isObjectId(notification)) {
2018-02-02 01:06:01 +02:00
_notification = await Notification.findOne({
_id: notification
});
} else if (typeof notification === 'string') {
_notification = await Notification.findOne({
_id: new mongo.ObjectID(notification)
});
} else {
_notification = deepcopy(notification);
}
// Rename _id to id
_notification.id = _notification._id;
delete _notification._id;
2018-03-29 08:48:47 +03:00
// Rename notifierId to userId
_notification.userId = _notification.notifierId;
delete _notification.notifierId;
2018-02-02 01:06:01 +02:00
2018-03-29 08:48:47 +03:00
const me = _notification.notifieeId;
delete _notification.notifieeId;
2018-02-02 01:06:01 +02:00
// Populate notifier
2018-03-29 08:48:47 +03:00
_notification.user = await packUser(_notification.userId, me);
2018-02-02 01:06:01 +02:00
switch (_notification.type) {
case 'follow':
2018-06-02 10:13:32 +03:00
case 'receiveFollowRequest':
2018-02-02 01:06:01 +02:00
// nope
break;
case 'mention':
case 'reply':
2018-04-07 20:30:37 +03:00
case 'renote':
2018-02-02 01:06:01 +02:00
case 'quote':
case 'reaction':
case 'poll_vote':
2018-04-07 20:30:37 +03:00
// Populate note
_notification.note = await packNote(_notification.noteId, me);
// (データベースの不具合などで)投稿が見つからなかったら
if (_notification.note == null) {
2018-10-10 20:19:21 +03:00
console.warn(`[DAMAGED DB] (missing) pkg: notification -> note :: ${_notification.id} (note ${_notification.noteId})`);
return resolve(null);
}
2018-02-02 01:06:01 +02:00
break;
default:
console.error(`Unknown type: ${_notification.type}`);
break;
}
resolve(_notification);
});