Sharkey/src/server/api/endpoints/users/search.ts

156 lines
3.4 KiB
TypeScript
Raw Normal View History

2017-03-08 20:50:09 +02:00
import $ from 'cafy';
2016-12-29 00:49:51 +02:00
const escapeRegexp = require('escape-regexp');
2018-11-02 06:47:44 +02:00
import User, { pack, validateUsername, IUser } from '../../../../models/user';
import define from '../../define';
2018-08-06 12:28:27 +03:00
export const meta = {
desc: {
2018-08-29 00:59:43 +03:00
'ja-JP': 'ユーザーを検索します。'
2018-08-06 12:28:27 +03:00
},
requireCredential: false,
params: {
2018-11-01 20:32:24 +02:00
query: {
validator: $.str,
2018-08-06 12:28:27 +03:00
desc: {
2018-08-29 00:59:43 +03:00
'ja-JP': 'クエリ'
2018-08-06 12:28:27 +03:00
}
2018-11-01 20:32:24 +02:00
},
2018-08-06 12:28:27 +03:00
2018-11-01 20:32:24 +02:00
offset: {
validator: $.num.optional.min(0),
2018-08-06 12:28:27 +03:00
default: 0,
desc: {
2018-08-29 00:59:43 +03:00
'ja-JP': 'オフセット'
2018-08-06 12:28:27 +03:00
}
2018-11-01 20:32:24 +02:00
},
2018-08-06 12:28:27 +03:00
2018-11-01 20:32:24 +02:00
limit: {
validator: $.num.optional.range(1, 100),
2018-08-06 12:28:27 +03:00
default: 10,
desc: {
2018-08-29 00:59:43 +03:00
'ja-JP': '取得する数'
2018-08-06 12:28:27 +03:00
}
2018-11-01 20:32:24 +02:00
},
2018-08-06 12:28:27 +03:00
2018-11-01 20:32:24 +02:00
localOnly: {
validator: $.bool.optional,
2018-08-06 12:28:27 +03:00
default: false,
desc: {
2018-08-29 00:59:43 +03:00
'ja-JP': 'ローカルユーザーのみ検索対象にするか否か'
2018-08-06 12:28:27 +03:00
}
2018-11-01 20:32:24 +02:00
},
2018-08-06 12:28:27 +03:00
},
};
2016-12-29 00:49:51 +02:00
2018-11-02 06:47:44 +02:00
export default define(meta, (ps, me) => new Promise(async (res, rej) => {
2018-08-06 12:28:27 +03:00
const isUsername = validateUsername(ps.query.replace('@', ''));
let users: IUser[] = [];
if (isUsername) {
users = await User
.find({
host: null,
usernameLower: new RegExp('^' + escapeRegexp(ps.query.replace('@', '').toLowerCase()))
2016-12-29 00:49:51 +02:00
}, {
2018-08-06 12:28:27 +03:00
limit: ps.limit,
skip: ps.offset
});
if (users.length < ps.limit && !ps.localOnly) {
const otherUsers = await User
.find({
host: { $ne: null },
usernameLower: new RegExp('^' + escapeRegexp(ps.query.replace('@', '').toLowerCase()))
}, {
limit: ps.limit - users.length
});
users = users.concat(otherUsers);
}
if (users.length < ps.limit) {
const otherUsers = await User
.find({
_id: { $nin: users.map(u => u._id) },
host: null,
usernameLower: new RegExp(escapeRegexp(ps.query.replace('@', '').toLowerCase()))
}, {
limit: ps.limit - users.length
});
users = users.concat(otherUsers);
}
if (users.length < ps.limit && !ps.localOnly) {
const otherUsers = await User
.find({
_id: { $nin: users.map(u => u._id) },
host: { $ne: null },
usernameLower: new RegExp(escapeRegexp(ps.query.replace('@', '').toLowerCase()))
}, {
limit: ps.limit - users.length
});
users = users.concat(otherUsers);
}
}
if (users.length < ps.limit) {
const otherUsers = await User
.find({
_id: { $nin: users.map(u => u._id) },
host: null,
name: new RegExp('^' + escapeRegexp(ps.query.toLowerCase()))
}, {
limit: ps.limit - users.length
});
users = users.concat(otherUsers);
}
if (users.length < ps.limit && !ps.localOnly) {
const otherUsers = await User
.find({
_id: { $nin: users.map(u => u._id) },
host: { $ne: null },
name: new RegExp('^' + escapeRegexp(ps.query.toLowerCase()))
}, {
limit: ps.limit - users.length
});
users = users.concat(otherUsers);
}
if (users.length < ps.limit) {
const otherUsers = await User
.find({
_id: { $nin: users.map(u => u._id) },
host: null,
name: new RegExp(escapeRegexp(ps.query.toLowerCase()))
}, {
limit: ps.limit - users.length
});
users = users.concat(otherUsers);
}
if (users.length < ps.limit && !ps.localOnly) {
const otherUsers = await User
.find({
_id: { $nin: users.map(u => u._id) },
host: { $ne: null },
name: new RegExp(escapeRegexp(ps.query.toLowerCase()))
}, {
limit: ps.limit - users.length
});
users = users.concat(otherUsers);
}
2016-12-29 00:49:51 +02:00
// Serialize
2018-06-18 03:54:53 +03:00
res(await Promise.all(users.map(user => pack(user, me, { detail: true }))));
2018-11-02 06:47:44 +02:00
}));