summaryrefslogtreecommitdiff
path: root/src/api/endpoints/aggregation
diff options
context:
space:
mode:
Diffstat (limited to 'src/api/endpoints/aggregation')
-rw-r--r--src/api/endpoints/aggregation/posts.ts95
-rw-r--r--src/api/endpoints/aggregation/users.ts59
2 files changed, 154 insertions, 0 deletions
diff --git a/src/api/endpoints/aggregation/posts.ts b/src/api/endpoints/aggregation/posts.ts
new file mode 100644
index 0000000000..573816e4fd
--- /dev/null
+++ b/src/api/endpoints/aggregation/posts.ts
@@ -0,0 +1,95 @@
+/**
+ * Module dependencies
+ */
+import $ from 'cafy';
+import Post from '../../models/post';
+
+/**
+ * Aggregate posts
+ *
+ * @param {any} params
+ * @return {Promise<any>}
+ */
+module.exports = params => new Promise(async (res, rej) => {
+ // Get 'limit' parameter
+ const [limit = 365, limitErr] = $(params.limit).optional.number().range(1, 365).$;
+ if (limitErr) return rej('invalid limit param');
+
+ const datas = await Post
+ .aggregate([
+ { $project: {
+ repost_id: '$repost_id',
+ reply_to_id: '$reply_to_id',
+ created_at: { $add: ['$created_at', 9 * 60 * 60 * 1000] } // Convert into JST
+ }},
+ { $project: {
+ date: {
+ year: { $year: '$created_at' },
+ month: { $month: '$created_at' },
+ day: { $dayOfMonth: '$created_at' }
+ },
+ type: {
+ $cond: {
+ if: { $ne: ['$repost_id', null] },
+ then: 'repost',
+ else: {
+ $cond: {
+ if: { $ne: ['$reply_to_id', null] },
+ then: 'reply',
+ else: 'post'
+ }
+ }
+ }
+ }}
+ },
+ { $group: { _id: {
+ date: '$date',
+ type: '$type'
+ }, count: { $sum: 1 } } },
+ { $group: {
+ _id: '$_id.date',
+ data: { $addToSet: {
+ type: '$_id.type',
+ count: '$count'
+ }}
+ } }
+ ]);
+
+ datas.forEach(data => {
+ data.date = data._id;
+ delete data._id;
+
+ data.posts = (data.data.filter(x => x.type == 'post')[0] || { count: 0 }).count;
+ data.reposts = (data.data.filter(x => x.type == 'repost')[0] || { count: 0 }).count;
+ data.replies = (data.data.filter(x => 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 =>
+ 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({
+ date: {
+ year: day.getFullYear(),
+ month: day.getMonth() + 1, // In JavaScript, month is zero-based.
+ day: day.getDate()
+ },
+ posts: 0,
+ reposts: 0,
+ replies: 0
+ });
+ }
+ }
+
+ res(graph);
+});
diff --git a/src/api/endpoints/aggregation/users.ts b/src/api/endpoints/aggregation/users.ts
new file mode 100644
index 0000000000..9be4ca12a1
--- /dev/null
+++ b/src/api/endpoints/aggregation/users.ts
@@ -0,0 +1,59 @@
+/**
+ * Module dependencies
+ */
+import $ from 'cafy';
+import User from '../../models/user';
+
+/**
+ * Aggregate users
+ *
+ * @param {any} params
+ * @return {Promise<any>}
+ */
+module.exports = params => new Promise(async (res, rej) => {
+ // Get 'limit' parameter
+ const [limit = 365, limitErr] = $(params.limit).optional.number().range(1, 365).$;
+ if (limitErr) return rej('invalid limit param');
+
+ const startTime = new Date(new Date().setMonth(new Date().getMonth() - 1));
+
+ const users = await User
+ .find({
+ $or: [
+ { deleted_at: { $exists: false } },
+ { deleted_at: { $gt: startTime } }
+ ]
+ }, {
+ _id: false,
+ created_at: true,
+ deleted_at: true
+ }, {
+ sort: { created_at: -1 }
+ });
+
+ const graph = [];
+
+ for (let i = 0; i < limit; i++) {
+ let day = new Date(new Date().setDate(new Date().getDate() - i));
+ day = new Date(day.setMilliseconds(999));
+ day = new Date(day.setSeconds(59));
+ day = new Date(day.setMinutes(59));
+ day = new Date(day.setHours(23));
+ // day = day.getTime();
+
+ const count = users.filter(f =>
+ f.created_at < day && (f.deleted_at == null || f.deleted_at > day)
+ ).length;
+
+ graph.push({
+ date: {
+ year: day.getFullYear(),
+ month: day.getMonth() + 1, // In JavaScript, month is zero-based.
+ day: day.getDate()
+ },
+ count: count
+ });
+ }
+
+ res(graph);
+});