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

117 lines
2.7 KiB
TypeScript
Raw Normal View History

import * as ms from 'ms';
2017-03-08 19:50:09 +01:00
import $ from 'cafy';
import User, { pack, ILocalUser } from '../../../../models/user';
2018-04-19 05:43:25 +02:00
import { getFriendIds } from '../../common/get-friends';
import * as request from 'request-promise-native';
2018-10-09 19:08:26 +02:00
import config from '../../../../config';
2018-11-02 05:47:44 +01:00
import define from '../../define';
import fetchMeta from '../../../../misc/fetch-meta';
import resolveUser from '../../../../remote/resolve-user';
2019-02-01 01:57:51 +01:00
import { getHideUserIds } from '../../common/get-hide-users';
import { apiLogger } from '../../logger';
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,
2018-11-02 04:49:08 +01:00
kind: 'account-read',
params: {
limit: {
2019-02-13 08:33:07 +01:00
validator: $.optional.num.range(1, 100),
2018-11-02 04:49:08 +01:00
default: 10
},
offset: {
2019-02-13 08:33:07 +01:00
validator: $.optional.num.min(0),
2018-11-02 04:49:08 +01:00
default: 0
}
}
2018-07-16 21:36:44 +02:00
};
2018-11-02 05:47:44 +01:00
export default define(meta, (ps, me) => new Promise(async (res, rej) => {
const instance = await fetchMeta();
if (instance.enableExternalUserRecommendation) {
2018-10-09 19:08:26 +02:00
const userName = me.username;
const hostName = config.hostname;
2018-11-02 05:47:44 +01:00
const limit = ps.limit;
const offset = ps.offset;
const timeout = instance.externalUserRecommendationTimeout;
const engine = instance.externalUserRecommendationEngine;
2018-10-06 10:21:27 +02:00
const url = engine
.replace('{{host}}', hostName)
.replace('{{user}}', userName)
2018-11-02 05:47:44 +01:00
.replace('{{limit}}', limit.toString())
.replace('{{offset}}', offset.toString());
2018-10-09 19:08:26 +02:00
2018-10-13 06:13:15 +02:00
request({
url: url,
proxy: config.proxy,
2018-10-13 06:13:15 +02:00
timeout: timeout,
json: true,
followRedirect: true,
followAllRedirects: true
})
.then(body => convertUsers(body, me))
.then(packed => res(packed))
.catch(e => rej(e));
2018-10-06 09:03:18 +02:00
} else {
// ID list of the user itself and other users who the user follows
const followingIds = await getFriendIds(me._id);
2019-02-01 01:57:51 +01:00
// 隠すユーザーを取得
const hideUserIds = await getHideUserIds(me);
2018-10-06 09:03:18 +02:00
const users = await User
.find({
_id: {
2019-02-01 01:57:51 +01:00
$nin: followingIds.concat(hideUserIds)
2018-10-06 09:03:18 +02:00
},
2018-10-09 19:04:16 +02:00
isLocked: { $ne: true },
2018-11-23 23:04:29 +01:00
updatedAt: {
$gte: new Date(Date.now() - ms('7days'))
},
host: null
2018-10-06 09:03:18 +02:00
}, {
2018-11-02 04:49:08 +01:00
limit: ps.limit,
skip: ps.offset,
2018-10-06 09:03:18 +02:00
sort: {
followersCount: -1
}
});
2016-12-28 23:49:51 +01:00
2018-11-02 04:49:08 +01:00
res(await Promise.all(users.map(user => pack(user, me, { detail: true }))));
2018-10-06 09:03:18 +02:00
}
2018-11-02 05:47:44 +01:00
}));
type IRecommendUser = {
name: string;
username: string;
host: string;
description: string;
avatarUrl: string;
};
/**
* Resolve/Pack dummy users
*/
async function convertUsers(src: IRecommendUser[], me: ILocalUser) {
const packed = await Promise.all(src.map(async x => {
const user = await resolveUser(x.username, x.host)
.catch(() => {
apiLogger.warn(`Can't resolve ${x.username}@${x.host}`);
return null;
});
if (user == null) return x;
return await pack(user, me, { detail: true });
}));
return packed;
}