1 files changed, 128 insertions, 0 deletions
diff --git a/src/chart/per-user-following.ts b/src/chart/per-user-following.ts
new file mode 100644
index 0000000000..fac4a1619f
--- /dev/null
+++ b/src/chart/per-user-following.ts
@@ -0,0 +1,128 @@
+import autobind from 'autobind-decorator';
+import Chart, { Obj } from './';
+import Following from '../models/following';
+import { IUser, isLocalUser } from '../models/user';
+
+/**
+ * ユーザーごとのフォローに関するチャート
+ */
+type PerUserFollowingLog = {
+	local: {
+		/**
+		 * フォローしている
+		 */
+		followings: {
+			/**
+			 * 合計
+			 */
+			total: number;
+
+			/**
+			 * フォローした数
+			 */
+			inc: number;
+
+			/**
+			 * フォロー解除した数
+			 */
+			dec: number;
+		};
+
+		/**
+		 * フォローされている
+		 */
+		followers: {
+			/**
+			 * 合計
+			 */
+			total: number;
+
+			/**
+			 * フォローされた数
+			 */
+			inc: number;
+
+			/**
+			 * フォロー解除された数
+			 */
+			dec: number;
+		};
+	};
+
+	remote: PerUserFollowingLog['local'];
+};
+
+class PerUserFollowingChart extends Chart<PerUserFollowingLog> {
+	constructor() {
+		super('perUserFollowing', true);
+	}
+
+	@autobind
+	protected async getTemplate(init: boolean, latest?: PerUserFollowingLog, group?: any): Promise<PerUserFollowingLog> {
+		const [
+			localFollowingsCount,
+			localFollowersCount,
+			remoteFollowingsCount,
+			remoteFollowersCount
+		] = init ? await Promise.all([
+			Following.count({ followerId: group, '_followee.host': null }),
+			Following.count({ followeeId: group, '_follower.host': null }),
+			Following.count({ followerId: group, '_followee.host': { $ne: null } }),
+			Following.count({ followeeId: group, '_follower.host': { $ne: null } })
+		]) : [
+			latest ? latest.local.followings.total : 0,
+			latest ? latest.local.followers.total : 0,
+			latest ? latest.remote.followings.total : 0,
+			latest ? latest.remote.followers.total : 0
+		];
+
+		return {
+			local: {
+				followings: {
+					total: localFollowingsCount,
+					inc: 0,
+					dec: 0
+				},
+				followers: {
+					total: localFollowersCount,
+					inc: 0,
+					dec: 0
+				}
+			},
+			remote: {
+				followings: {
+					total: remoteFollowingsCount,
+					inc: 0,
+					dec: 0
+				},
+				followers: {
+					total: remoteFollowersCount,
+					inc: 0,
+					dec: 0
+				}
+			}
+		};
+	}
+
+	@autobind
+	public async update(follower: IUser, followee: IUser, isFollow: boolean) {
+		const update: Obj = {};
+
+		update.total = isFollow ? 1 : -1;
+
+		if (isFollow) {
+			update.inc = 1;
+		} else {
+			update.dec = 1;
+		}
+
+		this.inc({
+			[isLocalUser(follower) ? 'local' : 'remote']: { followings: update }
+		}, follower._id);
+		this.inc({
+			[isLocalUser(followee) ? 'local' : 'remote']: { followers: update }
+		}, followee._id);
+	}
+}
+
+export default new PerUserFollowingChart();