1 files changed, 256 insertions, 0 deletions
diff --git a/src/server/api/endpoints/chart.ts b/src/server/api/endpoints/chart.ts
new file mode 100644
index 0000000000..7da970131e
--- /dev/null
+++ b/src/server/api/endpoints/chart.ts
@@ -0,0 +1,256 @@
+import $ from 'cafy';
+import Stats, { IStats } from '../../../models/stats';
+import getParams from '../get-params';
+
+type Omit<T, K extends keyof T> = Pick<T, Exclude<keyof T, K>>;
+
+function migrateStats(stats: IStats[]) {
+	stats.forEach(stat => {
+		const isOldData =
+			stat.users.local.inc == null ||
+			stat.users.local.dec == null ||
+			stat.users.remote.inc == null ||
+			stat.users.remote.dec == null ||
+			stat.notes.local.inc == null ||
+			stat.notes.local.dec == null ||
+			stat.notes.remote.inc == null ||
+			stat.notes.remote.dec == null ||
+			stat.drive.local.incCount == null ||
+			stat.drive.local.decCount == null ||
+			stat.drive.local.incSize == null ||
+			stat.drive.local.decSize == null ||
+			stat.drive.remote.incCount == null ||
+			stat.drive.remote.decCount == null ||
+			stat.drive.remote.incSize == null ||
+			stat.drive.remote.decSize == null;
+
+		if (!isOldData) return;
+
+		stat.users.local.inc = (stat as any).users.local.diff;
+		stat.users.local.dec = 0;
+		stat.users.remote.inc = (stat as any).users.remote.diff;
+		stat.users.remote.dec = 0;
+		stat.notes.local.inc = (stat as any).notes.local.diff;
+		stat.notes.local.dec = 0;
+		stat.notes.remote.inc = (stat as any).notes.remote.diff;
+		stat.notes.remote.dec = 0;
+		stat.drive.local.incCount = (stat as any).drive.local.diffCount;
+		stat.drive.local.decCount = 0;
+		stat.drive.local.incSize = (stat as any).drive.local.diffSize;
+		stat.drive.local.decSize = 0;
+		stat.drive.remote.incCount = (stat as any).drive.remote.diffCount;
+		stat.drive.remote.decCount = 0;
+		stat.drive.remote.incSize = (stat as any).drive.remote.diffSize;
+		stat.drive.remote.decSize = 0;
+	});
+}
+
+export const meta = {
+	desc: {
+		'ja-JP': 'インスタンスの統計を取得します。'
+	},
+
+	params: {
+		limit: $.num.optional.range(1, 100).note({
+			default: 30,
+			desc: {
+				'ja-JP': '最大数'
+			}
+		}),
+	}
+};
+
+export default (params: any) => new Promise(async (res, rej) => {
+	const [ps, psErr] = getParams(meta, params);
+	if (psErr) throw psErr;
+
+	const daysRange = ps.limit;
+	const hoursRange = ps.limit;
+
+	const now = new Date();
+	const y = now.getFullYear();
+	const m = now.getMonth();
+	const d = now.getDate();
+	const h = now.getHours();
+
+	const [statsPerDay, statsPerHour] = await Promise.all([
+		Stats.find({
+			span: 'day',
+			date: {
+				$gt: new Date(y, m, d - daysRange)
+			}
+		}, {
+			sort: {
+				date: -1
+			},
+			fields: {
+				_id: 0
+			}
+		}),
+		Stats.find({
+			span: 'hour',
+			date: {
+				$gt: new Date(y, m, d, h - hoursRange)
+			}
+		}, {
+			sort: {
+				date: -1
+			},
+			fields: {
+				_id: 0
+			}
+		}),
+	]);
+
+	// 後方互換性のため
+	migrateStats(statsPerDay);
+	migrateStats(statsPerHour);
+
+	const format = (src: IStats[], span: 'day' | 'hour') => {
+		const chart: Array<Omit<Omit<IStats, '_id'>, 'span'>> = [];
+
+		const range =
+			span == 'day' ? daysRange :
+			span == 'hour' ? hoursRange :
+			null;
+
+		for (let i = (range - 1); i >= 0; i--) {
+			const current =
+				span == 'day' ? new Date(y, m, d - i) :
+				span == 'hour' ? new Date(y, m, d, h - i) :
+				null;
+
+			const stat = src.find(s => s.date.getTime() == current.getTime());
+
+			if (stat) {
+				chart.unshift(stat);
+			} else { // 隙間埋め
+				const mostRecent = src.find(s => s.date.getTime() < current.getTime());
+				if (mostRecent) {
+					chart.unshift({
+						date: current,
+						users: {
+							local: {
+								total: mostRecent.users.local.total,
+								inc: 0,
+								dec: 0
+							},
+							remote: {
+								total: mostRecent.users.remote.total,
+								inc: 0,
+								dec: 0
+							}
+						},
+						notes: {
+							local: {
+								total: mostRecent.notes.local.total,
+								inc: 0,
+								dec: 0,
+								diffs: {
+									normal: 0,
+									reply: 0,
+									renote: 0
+								}
+							},
+							remote: {
+								total: mostRecent.notes.remote.total,
+								inc: 0,
+								dec: 0,
+								diffs: {
+									normal: 0,
+									reply: 0,
+									renote: 0
+								}
+							}
+						},
+						drive: {
+							local: {
+								totalCount: mostRecent.drive.local.totalCount,
+								totalSize: mostRecent.drive.local.totalSize,
+								incCount: 0,
+								incSize: 0,
+								decCount: 0,
+								decSize: 0
+							},
+							remote: {
+								totalCount: mostRecent.drive.remote.totalCount,
+								totalSize: mostRecent.drive.remote.totalSize,
+								incCount: 0,
+								incSize: 0,
+								decCount: 0,
+								decSize: 0
+							}
+						}
+					});
+				} else {
+					chart.unshift({
+						date: current,
+						users: {
+							local: {
+								total: 0,
+								inc: 0,
+								dec: 0
+							},
+							remote: {
+								total: 0,
+								inc: 0,
+								dec: 0
+							}
+						},
+						notes: {
+							local: {
+								total: 0,
+								inc: 0,
+								dec: 0,
+								diffs: {
+									normal: 0,
+									reply: 0,
+									renote: 0
+								}
+							},
+							remote: {
+								total: 0,
+								inc: 0,
+								dec: 0,
+								diffs: {
+									normal: 0,
+									reply: 0,
+									renote: 0
+								}
+							}
+						},
+						drive: {
+							local: {
+								totalCount: 0,
+								totalSize: 0,
+								incCount: 0,
+								incSize: 0,
+								decCount: 0,
+								decSize: 0
+							},
+							remote: {
+								totalCount: 0,
+								totalSize: 0,
+								incCount: 0,
+								incSize: 0,
+								decCount: 0,
+								decSize: 0
+							}
+						}
+					});
+				}
+			}
+		}
+
+		chart.forEach(x => {
+			delete (x as any).span;
+		});
+
+		return chart;
+	};
+
+	res({
+		perDay: format(statsPerDay, 'day'),
+		perHour: format(statsPerHour, 'hour')
+	});
+});