wip

2 files changed, 231 insertions, 0 deletions

diff --git a/src/models/chart.ts b/src/models/chart.ts
new file mode 100644
index 0000000000..2063408d81
--- /dev/null
+++ b/src/models/chart.ts
@@ -0,0 +1,88 @@
+import * as mongo from 'mongodb';
+import db from '../db/mongodb';
+
+const Chart = db.get<IChart>('chart');
+Chart.createIndex('date', { unique: true });
+export default Chart;
+
+export interface IChart {
+	_id: mongo.ObjectID;
+
+	date: Date;
+
+	users: {
+		local: {
+			/**
+			 * この日時点での、ローカルのユーザーの総計
+			 */
+			total: number;
+
+			/**
+			 * この日時点での、ローカルのユーザー数の前日比
+			 */
+			diff: number;
+		};
+
+		remote: {
+			/**
+			 * この日時点での、リモートのユーザーの総計
+			 */
+			total: number;
+
+			/**
+			 * この日時点での、リモートのユーザー数の前日比
+			 */
+			diff: number;
+		};
+	};
+
+	notes: {
+		local: {
+			/**
+			 * この日時点での、ローカルの投稿の総計
+			 */
+			total: number;
+
+			diffs: {
+				/**
+				 * この日に行われた、ローカルの通常の投稿数の前日比
+				 */
+				normal: number;
+
+				/**
+				 * この日に行われた、ローカルのリプライの投稿数の前日比
+				 */
+				reply: number;
+
+				/**
+				 * この日に行われた、ローカルのRenoteの投稿数の前日比
+				 */
+				renote: number;
+			};
+		};
+
+		remote: {
+			/**
+			 * この日時点での、リモートの投稿の総計
+			 */
+			total: number;
+
+			diffs: {
+				/**
+				 * この日に行われた、リモートの通常の投稿数の前日比
+				 */
+				normal: number;
+
+				/**
+				 * この日に行われた、リモートのリプライの投稿数の前日比
+				 */
+				reply: number;
+
+				/**
+				 * この日に行われた、リモートのRenoteの投稿数の前日比
+				 */
+				renote: number;
+			};
+		};
+	};
+}
diff --git a/src/services/update-chart.ts b/src/services/update-chart.ts
new file mode 100644
index 0000000000..c1f212c9d8
--- /dev/null
+++ b/src/services/update-chart.ts
@@ -0,0 +1,143 @@
+import { INote } from '../models/note';
+import Chart, { IChart } from '../models/chart';
+import { isLocalUser } from '../models/user';
+
+async function getTodayStats(): Promise<IChart> {
+	const now = new Date();
+	const y = now.getFullYear();
+	const m = now.getMonth();
+	const d = now.getDate();
+	const today = new Date(y, m, d);
+
+	// 今日の統計
+	const todayStats = await Chart.findOne({
+		date: today
+	});
+
+	// 日付が変わってから、初めてのチャート更新なら
+	if (todayStats == null) {
+		// 最も最近の統計を持ってくる
+		// * 昨日何もチャートを更新するような出来事がなかった場合は、
+		//   統計がそもそも作られずドキュメントが存在しないということがあり得るため、
+		//   「昨日の」と決め打ちせずに「もっとも最近の」とします
+		const mostRecentStats = await Chart.findOne({}, {
+			sort: {
+				createdAt: -1
+			}
+		});
+
+		// 統計が存在しなかったら
+		// * Misskeyインスタンスを建てて初めてのチャート更新時など
+		if (mostRecentStats == null) {
+			// 空の統計を作成
+			const stats = await Chart.insert({
+				date: today,
+				users: {
+					local: {
+						total: 0,
+						diff: 0
+					},
+					remote: {
+						total: 0,
+						diff: 0
+					}
+				},
+				notes: {
+					local: {
+						total: 0,
+						diffs: {
+							normal: 0,
+							reply: 0,
+							renote: 0
+						}
+					},
+					remote: {
+						total: 0,
+						diffs: {
+							normal: 0,
+							reply: 0,
+							renote: 0
+						}
+					}
+				}
+			});
+
+			return stats;
+		} else {
+			// 今日の統計を初期挿入
+			const stats = await Chart.insert({
+				date: today,
+				users: {
+					local: {
+						total: mostRecentStats.users.local.total,
+						diff: 0
+					},
+					remote: {
+						total: mostRecentStats.users.remote.total,
+						diff: 0
+					}
+				},
+				notes: {
+					local: {
+						total: mostRecentStats.notes.local.total,
+						diffs: {
+							normal: 0,
+							reply: 0,
+							renote: 0
+						}
+					},
+					remote: {
+						total: mostRecentStats.notes.remote.total,
+						diffs: {
+							normal: 0,
+							reply: 0,
+							renote: 0
+						}
+					}
+				}
+			});
+
+			return stats;
+		}
+	} else {
+		return todayStats;
+	}
+}
+
+async function update(inc: any) {
+	const stats = await getTodayStats();
+
+	await Chart.findOneAndUpdate({
+		_id: stats._id
+	}, {
+		$inc: inc
+	});
+}
+
+export async function incNote(note: INote) {
+	const inc = {} as any;
+
+	if (isLocalUser(note._user)) {
+		inc['notes.local.total'] = 1;
+
+		if (note.replyId != null) {
+			inc['notes.local.diffs.reply'] = 1;
+		} else if (note.renoteId != null) {
+			inc['notes.local.diffs.renote'] = 1;
+		} else {
+			inc['notes.local.diffs.normal'] = 1;
+		}
+	} else {
+		inc['notes.remote.total'] = 1;
+
+		if (note.replyId != null) {
+			inc['notes.remote.diffs.reply'] = 1;
+		} else if (note.renoteId != null) {
+			inc['notes.remote.diffs.renote'] = 1;
+		} else {
+			inc['notes.remote.diffs.normal'] = 1;
+		}
+	}
+
+	await update(inc);
+}