egirlskey/src/server/api/endpoints/aggregation/posts.ts

import $ from 'cafy';
import Note from '../../../../models/note';

/**
 * Aggregate notes
 */
export default (params: any) => new Promise(async (res, rej) => {
	// Get 'limit' parameter
	const [limit = 365, limitErr] = $.num.optional.range(1, 365).get(params.limit);
	if (limitErr) return rej('invalid limit param');

	const datas = await Note
		.aggregate([
			{ $project: {
				renoteId: '$renoteId',
				replyId: '$replyId',
				createdAt: { $add: ['$createdAt', 9 * 60 * 60 * 1000] } // Convert into JST
			}},
			{ $project: {
				date: {
					year: { $year: '$createdAt' },
					month: { $month: '$createdAt' },
					day: { $dayOfMonth: '$createdAt' }
				},
				type: {
					$cond: {
						if: { $ne: ['$renoteId', null] },
						then: 'renote',
						else: {
							$cond: {
								if: { $ne: ['$replyId', null] },
								then: 'reply',
								else: 'note'
							}
						}
					}
				}}
			},
			{ $group: { _id: {
				date: '$date',
				type: '$type'
			}, count: { $sum: 1 } } },
			{ $group: {
				_id: '$_id.date',
				data: { $addToSet: {
					type: '$_id.type',
					count: '$count'
				}}
			} }
		]);

	datas.forEach((data: any) => {
		data.date = data._id;
		delete data._id;

		data.notes = (data.data.filter((x: any) => x.type == 'note')[0] || { count: 0 }).count;
		data.renotes = (data.data.filter((x: any) => x.type == 'renote')[0] || { count: 0 }).count;
		data.replies = (data.data.filter((x: any) => x.type == 'reply')[0] || { count: 0 }).count;

		delete data.data;
	});

	const graph = [];

	for (let i = 0; i < limit; i++) {
		const day = new Date(new Date().setDate(new Date().getDate() - i));

		const data = datas.filter((d: any) =>
			d.date.year == day.getFullYear() && d.date.month == day.getMonth() + 1 && d.date.day == day.getDate()
		)[0];

		if (data) {
			graph.push(data);
		} else {
			graph.push({
				notes: 0,
				renotes: 0,
				replies: 0
			});
		}
	}

	res(graph);
});