iceshrimp/src/server/api/endpoints/users/recommendation.ts

const ms = require('ms');
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 config from '../../../../config'

export const meta = {
	desc: {
		'ja-JP': 'おすすめのユーザー一覧を取得します。'
	},

	requireCredential: true,

	kind: 'account-read'
};

export default (params: any, me: ILocalUser) => new Promise(async (res, rej) => {
	if (config.user_recommendation && config.user_recommendation.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
		const url = engine
			.replace('{{host}}', hostName)
			.replace('{{user}}', userName)
			.replace('{{limit}}', limit)
			.replace('{{offset}}', offset)
		request(
			{
				url: url,
				timeout: timeout,
				json: true,
				followRedirect: true,
				followAllRedirects: true
			},
			function (error: any, response: any, body: any) {
				if (!error && response.statusCode == 200) {
					res(body)
				} else {
					res([])
				}
			}
		)
	} else {
		// Get 'limit' parameter
		const [limit = 10, limitErr] = $.num.optional.range(1, 100).get(params.limit);
		if (limitErr) return rej('invalid limit param');

		// 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 }))));
	}
});