summaryrefslogtreecommitdiff
path: root/src/chart/per-user-notes.ts
diff options
context:
space:
mode:
Diffstat (limited to 'src/chart/per-user-notes.ts')
-rw-r--r--src/chart/per-user-notes.ts94
1 files changed, 94 insertions, 0 deletions
diff --git a/src/chart/per-user-notes.ts b/src/chart/per-user-notes.ts
new file mode 100644
index 0000000000..9558f5c839
--- /dev/null
+++ b/src/chart/per-user-notes.ts
@@ -0,0 +1,94 @@
+import autobind from 'autobind-decorator';
+import Chart, { Obj } from './';
+import Note, { INote } from '../models/note';
+import { IUser } from '../models/user';
+
+/**
+ * ユーザーごとの投稿に関するチャート
+ */
+type PerUserNotesLog = {
+ /**
+ * 集計期間時点での、全投稿数
+ */
+ total: number;
+
+ /**
+ * 増加した投稿数
+ */
+ inc: number;
+
+ /**
+ * 減少した投稿数
+ */
+ dec: number;
+
+ diffs: {
+ /**
+ * 通常の投稿数の差分
+ */
+ normal: number;
+
+ /**
+ * リプライの投稿数の差分
+ */
+ reply: number;
+
+ /**
+ * Renoteの投稿数の差分
+ */
+ renote: number;
+ };
+};
+
+class PerUserNotesChart extends Chart<PerUserNotesLog> {
+ constructor() {
+ super('perUserNotes', true);
+ }
+
+ @autobind
+ protected async getTemplate(init: boolean, latest?: PerUserNotesLog, group?: any): Promise<PerUserNotesLog> {
+ const [count] = init ? await Promise.all([
+ Note.count({ userId: group, deletedAt: null }),
+ ]) : [
+ latest ? latest.total : 0
+ ];
+
+ return {
+ total: count,
+ inc: 0,
+ dec: 0,
+ diffs: {
+ normal: 0,
+ reply: 0,
+ renote: 0
+ }
+ };
+ }
+
+ @autobind
+ public async update(user: IUser, note: INote, isAdditional: boolean) {
+ const update: Obj = {
+ diffs: {}
+ };
+
+ update.total = isAdditional ? 1 : -1;
+
+ if (isAdditional) {
+ update.inc = 1;
+ } else {
+ update.dec = 1;
+ }
+
+ if (note.replyId != null) {
+ update.diffs.reply = isAdditional ? 1 : -1;
+ } else if (note.renoteId != null) {
+ update.diffs.renote = isAdditional ? 1 : -1;
+ } else {
+ update.diffs.normal = isAdditional ? 1 : -1;
+ }
+
+ await this.inc(update, user._id);
+ }
+}
+
+export default new PerUserNotesChart();