rudeshark.net/src/server/api/endpoints/users/recommendation.ts

91 lines
2.2 KiB
TypeScript
Raw Normal View History

const ms = require('ms');
2017-03-08 19:50:09 +01:00
import $ from 'cafy';
2018-06-18 02:54:53 +02:00
import User, { pack, ILocalUser } from '../../../../models/user';
2018-04-19 05:43:25 +02:00
import { getFriendIds } from '../../common/get-friends';
2018-04-18 11:46:38 +02:00
import Mute from '../../../../models/mute';
2018-10-06 09:03:18 +02:00
import * as request from 'request'
import config from '../../../../config'
2016-12-28 23:49:51 +01:00
2018-07-16 21:36:44 +02:00
export const meta = {
desc: {
2018-08-28 23:59:43 +02:00
'ja-JP': 'おすすめのユーザー一覧を取得します。'
2018-07-16 21:36:44 +02:00
},
requireCredential: true,
kind: 'account-read'
};
2018-07-05 19:58:29 +02:00
export default (params: any, me: ILocalUser) => new Promise(async (res, rej) => {
2018-10-06 10:19:41 +02:00
if (config.user_recommendation && config.user_recommendation.external) {
2018-10-06 09:03:18 +02:00
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
2018-10-06 10:21:27 +02:00
const url = engine
.replace('{{host}}', hostName)
.replace('{{user}}', userName)
.replace('{{limit}}', limit)
.replace('{{offset}}', offset)
2018-10-06 09:03:18 +02:00
request(
{
url: url,
timeout: timeout,
json: true,
followRedirect: true,
followAllRedirects: true
},
2018-10-06 10:23:32 +02:00
(error: any, response: any, body: any) => {
2018-10-06 09:03:18 +02:00
if (!error && response.statusCode == 200) {
res(body)
} else {
res([])
2018-03-27 09:51:12 +02:00
}
2016-12-28 23:49:51 +01:00
}
2018-10-06 09:03:18 +02:00
)
} 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
}
});
2016-12-28 23:49:51 +01:00
2018-10-06 09:03:18 +02:00
// Serialize
res(await Promise.all(users.map(async user =>
await pack(user, me, { detail: true }))));
}
2016-12-28 23:49:51 +01:00
});