iceshrimp/src/api/common/read-notification.ts

import * as mongo from 'mongodb';
import { default as Notification, INotification } from '../models/notification';
import publishUserStream from '../event';

/**
 * Mark as read notification(s)
 */
export default (
	user: string | mongo.ObjectID,
	message: string | string[] | INotification | INotification[] | mongo.ObjectID | mongo.ObjectID[]
) => new Promise<any>(async (resolve, reject) => {

	const userId = mongo.ObjectID.prototype.isPrototypeOf(user)
		? user
		: new mongo.ObjectID(user);

	const ids: mongo.ObjectID[] = Array.isArray(message)
		? mongo.ObjectID.prototype.isPrototypeOf(message[0])
			? (message as mongo.ObjectID[])
			: typeof message[0] === 'string'
				? (message as string[]).map(m => new mongo.ObjectID(m))
				: (message as INotification[]).map(m => m._id)
		: mongo.ObjectID.prototype.isPrototypeOf(message)
			? [(message as mongo.ObjectID)]
			: typeof message === 'string'
				? [new mongo.ObjectID(message)]
				: [(message as INotification)._id];

	// Update documents
	await Notification.update({
		_id: { $in: ids },
		is_read: false
	}, {
		$set: {
			is_read: true
		}
	}, {
		multi: true
	});

	// Calc count of my unread notifications
	const count = await Notification
		.count({
			notifiee_id: userId,
			is_read: false
		});

	if (count == 0) {
		// 全ての(いままで未読だった)通知を(これで)読みましたよというイベントを発行
		publishUserStream(userId, 'read_all_notifications');
	}
});