From 649177985dd7fa7ef220e3b61f4007eec3cd3e7f Mon Sep 17 00:00:00 2001
From: syuilo <syuilotan@yahoo.co.jp>
Date: Thu, 1 Nov 2018 00:11:21 +0900
Subject: [PATCH] [API] Implement users/relation

---
 src/models/user.ts                         | 101 +++++++++++----------
 src/server/api/endpoints/users/relation.ts |  28 ++++++
 2 files changed, 81 insertions(+), 48 deletions(-)
 create mode 100644 src/server/api/endpoints/users/relation.ts

diff --git a/src/models/user.ts b/src/models/user.ts
index ce877fe277..1a596c6217 100644
--- a/src/models/user.ts
+++ b/src/models/user.ts
@@ -155,6 +155,50 @@ export function isValidBirthday(birthday: string): boolean {
 }
 //#endregion
 
+export async function getRelation(me: mongo.ObjectId, target: mongo.ObjectId) {
+	const [following1, following2, followReq1, followReq2, toBlocking, fromBlocked, mute] = await Promise.all([
+		Following.findOne({
+			followerId: me,
+			followeeId: target
+		}),
+		Following.findOne({
+			followerId: target,
+			followeeId: me
+		}),
+		FollowRequest.findOne({
+			followerId: me,
+			followeeId: target
+		}),
+		FollowRequest.findOne({
+			followerId: target,
+			followeeId: me
+		}),
+		Blocking.findOne({
+			blockerId: me,
+			blockeeId: target
+		}),
+		Blocking.findOne({
+			blockerId: target,
+			blockeeId: me
+		}),
+		Mute.findOne({
+			muterId: me,
+			muteeId: target
+		})
+	]);
+
+	return {
+		isFollowing: following1 !== null,
+		isStalking: following1 && following1.stalk,
+		hasPendingFollowRequestFromYou: followReq1 !== null,
+		hasPendingFollowRequestToYou: followReq2 !== null,
+		isFollowed: following2 !== null,
+		isBlocking: toBlocking !== null,
+		isBlocked: fromBlocked !== null,
+		isMuted: mute !== null
+	};
+}
+
 /**
  * Pack a user for API response
  *
@@ -271,55 +315,16 @@ export const pack = (
 	}
 
 	if (meId && !meId.equals(_user.id) && opts.detail) {
-		const [following1, following2, followReq1, followReq2, toBlocking, fromBlocked, mute] = await Promise.all([
-			Following.findOne({
-				followerId: meId,
-				followeeId: _user.id
-			}),
-			Following.findOne({
-				followerId: _user.id,
-				followeeId: meId
-			}),
-			FollowRequest.findOne({
-				followerId: meId,
-				followeeId: _user.id
-			}),
-			FollowRequest.findOne({
-				followerId: _user.id,
-				followeeId: meId
-			}),
-			Blocking.findOne({
-				blockerId: meId,
-				blockeeId: _user.id
-			}),
-			Blocking.findOne({
-				blockerId: _user.id,
-				blockeeId: meId
-			}),
-			Mute.findOne({
-				muterId: meId,
-				muteeId: _user.id
-			})
-		]);
+		const relation = await getRelation(meId, _user.id);
 
-		// Whether the user is following
-		_user.isFollowing = following1 !== null;
-		_user.isStalking = following1 && following1.stalk;
-
-		_user.hasPendingFollowRequestFromYou = followReq1 !== null;
-		_user.hasPendingFollowRequestToYou = followReq2 !== null;
-
-		// Whether the user is followed
-		_user.isFollowed = following2 !== null;
-
-		// Whether the user is blocking
-		_user.isBlocking = toBlocking !== null;
-
-		// Whether the user is blocked
-		_user.isBlocked = fromBlocked !== null;
-
-		// Whether the user is muted
-		_user.isMuted = mute !== null;
+		_user.isFollowing = relation.isFollowing;
+		_user.isFollowed = relation.isFollowed;
+		_user.isStalking = relation.isStalking;
+		_user.hasPendingFollowRequestFromYou = relation.hasPendingFollowRequestFromYou;
+		_user.hasPendingFollowRequestToYou = relation.hasPendingFollowRequestToYou;
+		_user.isBlocking = relation.isBlocking;
+		_user.isBlocked = relation.isBlocked;
+		_user.isMuted = relation.isMuted;
 	}
 
 	if (opts.detail) {
diff --git a/src/server/api/endpoints/users/relation.ts b/src/server/api/endpoints/users/relation.ts
new file mode 100644
index 0000000000..d29065d11f
--- /dev/null
+++ b/src/server/api/endpoints/users/relation.ts
@@ -0,0 +1,28 @@
+import $ from 'cafy'; import ID from '../../../../misc/cafy-id';
+import { ILocalUser, getRelation } from '../../../../models/user';
+import getParams from '../../get-params';
+
+export const meta = {
+	desc: {
+		'ja-JP': 'ユーザー間のリレーションを取得します。'
+	},
+
+	params: {
+		userId: $.or($.type(ID), $.arr($.type(ID)).unique()).note({
+			desc: {
+				'ja-JP': 'ユーザーID (配列でも可)'
+			}
+		})
+	}
+};
+
+export default (params: any, me: ILocalUser) => new Promise(async (res, rej) => {
+	const [ps, psErr] = getParams(meta, params);
+	if (psErr) return rej(psErr);
+
+	const ids = Array.isArray(ps.userId) ? ps.userId : [ps.userId];
+
+	const relations = await Promise.all(ids.map(id => getRelation(me._id, id)));
+
+	res(Array.isArray(ps.userId) ? relations : relations[0]);
+});