From 96959f4d914585936ac72bcb5b60431e83394fbf Mon Sep 17 00:00:00 2001
From: Hakaba Hitoyo <example@example.com>
Date: Sat, 6 Oct 2018 16:03:18 +0900
Subject: [PATCH] external user recommendation

---
 .config/example.yml                           |   7 ++
 src/config/types.ts                           |   6 +
 .../api/endpoints/users/recommendation.ts     | 106 ++++++++++++------
 3 files changed, 84 insertions(+), 35 deletions(-)

diff --git a/.config/example.yml b/.config/example.yml
index ebad17183..0f38e593b 100644
--- a/.config/example.yml
+++ b/.config/example.yml
@@ -159,3 +159,10 @@ drive:
 
 # Summaly proxy
 # summalyProxy: "http://example.com"
+
+# User recommendation
+user_recommendation:
+  external: true
+  engine: http://vinayaka.distsn.org/cgi-bin/vinayaka-user-match-misskey-api.cgi?{{host}}+{{user}}+{{limit}}+{{offset}}
+  timeout: 300000
+
diff --git a/src/config/types.ts b/src/config/types.ts
index 003185acc..5c7b8260d 100644
--- a/src/config/types.ts
+++ b/src/config/types.ts
@@ -96,6 +96,12 @@ export type Source = {
 	google_maps_api_key: string;
 
 	clusterLimit?: number;
+
+	user_recommendation: {
+		external: boolean;
+		engine: string;
+		timeout: number;
+	};
 };
 
 /**
diff --git a/src/server/api/endpoints/users/recommendation.ts b/src/server/api/endpoints/users/recommendation.ts
index e0a5cb9e3..b0ad75caa 100644
--- a/src/server/api/endpoints/users/recommendation.ts
+++ b/src/server/api/endpoints/users/recommendation.ts
@@ -3,6 +3,9 @@ import $ from 'cafy';
 import User, { pack, ILocalUser } from '../../../../models/user';
 import { getFriendIds } from '../../common/get-friends';
 import Mute from '../../../../models/mute';
+import * as request from 'request'
+import { ILocalUser } from '../../../../models/user'
+import config from '../../../../config'
 
 export const meta = {
 	desc: {
@@ -15,44 +18,77 @@ export const meta = {
 };
 
 export default (params: any, me: ILocalUser) => new Promise(async (res, rej) => {
-	// Get 'limit' parameter
-	const [limit = 10, limitErr] = $.num.optional.range(1, 100).get(params.limit);
-	if (limitErr) return rej('invalid limit param');
+	var external = config.user_recommendation.external
 
-	// Get 'offset' parameter
-	const [offset = 0, offsetErr] = $.num.optional.min(0).get(params.offset);
-	if (offsetErr) return rej('invalid offset param');
-
-	// ID list of the user itself and other users who the user follows
-	const followingIds = await getFriendIds(me._id);
-
-	// ミュートしているユーザーを取得
-	const mutedUserIds = (await Mute.find({
-		muterId: me._id
-	})).map(m => m.muteeId);
-
-	const users = await User
-		.find({
-			_id: {
-				$nin: followingIds.concat(mutedUserIds)
+	if (external) {
+		var userName = me.username
+		var hostName = config.hostname
+		var limit = params.limit
+		var offset = params.offset
+		var timeout = config.user_recommendation.timeout
+		var engine = config.user_recommendation.engine
+		var url
+		url = engine
+		url = url.replace('{{host}}', hostName)
+		url = url.replace('{{user}}', userName)
+		url = url.replace('{{limit}}', limit)
+		url = url.replace('{{offset}}', offset)
+		request(
+			{
+				url: url,
+				timeout: timeout,
+				json: true,
+				followRedirect: true,
+				followAllRedirects: true
 			},
-			isLocked: false,
-			$or: [{
-				lastUsedAt: {
-					$gte: new Date(Date.now() - ms('7days'))
+			function (error: any, response: any, body: any) {
+				if (!error && response.statusCode == 200) {
+					res(body)
+				} else {
+					res([])
 				}
-			}, {
-				host: null
-			}]
-		}, {
-			limit: limit,
-			skip: offset,
-			sort: {
-				followersCount: -1
 			}
-		});
+		)
+	} else {
+		// Get 'limit' parameter
+		const [limit = 10, limitErr] = $.num.optional.range(1, 100).get(params.limit);
+		if (limitErr) return rej('invalid limit param');
 
-	// Serialize
-	res(await Promise.all(users.map(async user =>
-		await pack(user, me, { detail: true }))));
+		// Get 'offset' parameter
+		const [offset = 0, offsetErr] = $.num.optional.min(0).get(params.offset);
+		if (offsetErr) return rej('invalid offset param');
+
+		// ID list of the user itself and other users who the user follows
+		const followingIds = await getFriendIds(me._id);
+
+		// ミュートしているユーザーを取得
+		const mutedUserIds = (await Mute.find({
+			muterId: me._id
+		})).map(m => m.muteeId);
+
+		const users = await User
+			.find({
+				_id: {
+					$nin: followingIds.concat(mutedUserIds)
+				},
+				isLocked: false,
+				$or: [{
+					lastUsedAt: {
+						$gte: new Date(Date.now() - ms('7days'))
+					}
+				}, {
+					host: null
+				}]
+			}, {
+				limit: limit,
+				skip: offset,
+				sort: {
+					followersCount: -1
+				}
+			});
+
+		// Serialize
+		res(await Promise.all(users.map(async user =>
+			await pack(user, me, { detail: true }))));
+	}
 });