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

90 lines
2.1 KiB
TypeScript
Raw Normal View History

const ms = require('ms');
2017-03-08 18:50:09 +00:00
import $ from 'cafy';
2018-11-02 04:47:44 +00:00
import User, { pack } from '../../../../models/user';
2018-04-19 03:43:25 +00:00
import { getFriendIds } from '../../common/get-friends';
2018-04-18 09:46:38 +00:00
import Mute from '../../../../models/mute';
2018-10-09 17:08:26 +00:00
import * as request from 'request';
import config from '../../../../config';
2018-11-02 04:47:44 +00:00
import define from '../../define';
2016-12-28 22:49:51 +00:00
2018-07-16 19:36:44 +00:00
export const meta = {
desc: {
2018-08-28 21:59:43 +00:00
'ja-JP': 'おすすめのユーザー一覧を取得します。'
2018-07-16 19:36:44 +00:00
},
requireCredential: true,
2018-11-02 03:49:08 +00:00
kind: 'account-read',
params: {
limit: {
validator: $.num.optional.range(1, 100),
default: 10
},
offset: {
validator: $.num.optional.min(0),
default: 0
}
}
2018-07-16 19:36:44 +00:00
};
2018-11-02 04:47:44 +00:00
export default define(meta, (ps, me) => new Promise(async (res, rej) => {
2018-10-06 08:19:41 +00:00
if (config.user_recommendation && config.user_recommendation.external) {
2018-10-09 17:08:26 +00:00
const userName = me.username;
const hostName = config.hostname;
2018-11-02 04:47:44 +00:00
const limit = ps.limit;
const offset = ps.offset;
2018-10-09 17:08:26 +00:00
const timeout = config.user_recommendation.timeout;
const engine = config.user_recommendation.engine;
2018-10-06 08:21:27 +00:00
const url = engine
.replace('{{host}}', hostName)
.replace('{{user}}', userName)
2018-11-02 04:47:44 +00:00
.replace('{{limit}}', limit.toString())
.replace('{{offset}}', offset.toString());
2018-10-09 17:08:26 +00:00
2018-10-13 04:13:15 +00:00
request({
url: url,
proxy: config.proxy,
timeout: timeout,
json: true,
followRedirect: true,
followAllRedirects: true
}, (error: any, response: any, body: any) => {
if (!error && response.statusCode == 200) {
res(body);
} else {
res([]);
2016-12-28 22:49:51 +00:00
}
2018-10-13 04:13:15 +00:00
});
2018-10-06 07:03:18 +00:00
} else {
// 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)
},
2018-10-09 17:04:16 +00:00
isLocked: { $ne: true },
lastUsedAt: {
$gte: new Date(Date.now() - ms('7days'))
},
host: null
2018-10-06 07:03:18 +00:00
}, {
2018-11-02 03:49:08 +00:00
limit: ps.limit,
skip: ps.offset,
2018-10-06 07:03:18 +00:00
sort: {
followersCount: -1
}
});
2016-12-28 22:49:51 +00:00
2018-11-02 03:49:08 +00:00
res(await Promise.all(users.map(user => pack(user, me, { detail: true }))));
2018-10-06 07:03:18 +00:00
}
2018-11-02 04:47:44 +00:00
}));