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

99 lines
2.2 KiB
TypeScript
Raw Normal View History

2016-12-28 23:49:51 +01:00
/**
* Module dependencies
*/
import * as mongo from 'mongodb';
2017-03-08 19:50:09 +01:00
import $ from 'cafy';
2018-03-29 13:32:18 +02:00
import User, { pack } from '../../../../models/user';
2018-04-02 06:15:53 +02:00
import config from '../../../../config';
2016-12-28 23:49:51 +01:00
const escapeRegexp = require('escape-regexp');
/**
* Search a user
*
2017-03-01 09:37:01 +01:00
* @param {any} params
* @param {any} me
* @return {Promise<any>}
2016-12-28 23:49:51 +01:00
*/
2017-03-03 20:28:38 +01:00
module.exports = (params, me) => new Promise(async (res, rej) => {
2016-12-28 23:49:51 +01:00
// Get 'query' parameter
2017-03-08 19:50:09 +01:00
const [query, queryError] = $(params.query).string().pipe(x => x != '').$;
2017-03-02 23:47:14 +01:00
if (queryError) return rej('invalid query param');
2016-12-28 23:49:51 +01:00
// Get 'offset' parameter
2017-03-08 19:50:09 +01:00
const [offset = 0, offsetErr] = $(params.offset).optional.number().min(0).$;
2017-03-02 23:47:14 +01:00
if (offsetErr) return rej('invalid offset param');
2016-12-28 23:49:51 +01:00
// Get 'max' parameter
2017-03-08 19:50:09 +01:00
const [max = 10, maxErr] = $(params.max).optional.number().range(1, 30).$;
2017-03-02 23:47:14 +01:00
if (maxErr) return rej('invalid max param');
2016-12-28 23:49:51 +01:00
2017-03-08 19:50:09 +01:00
// If Elasticsearch is available, search by $
2016-12-28 23:49:51 +01:00
// If not, search by MongoDB
(config.elasticsearch.enable ? byElasticsearch : byNative)
(res, rej, me, query, offset, max);
});
// Search by MongoDB
async function byNative(res, rej, me, query, offset, max) {
const escapedQuery = escapeRegexp(query);
// Search users
const users = await User
.find({
$or: [{
2018-03-29 07:48:47 +02:00
usernameLower: new RegExp(escapedQuery.replace('@', '').toLowerCase())
2016-12-28 23:49:51 +01:00
}, {
name: new RegExp(escapedQuery)
}]
2017-02-16 03:20:41 +01:00
}, {
limit: max
2017-01-17 03:11:22 +01:00
});
2016-12-28 23:49:51 +01:00
// Serialize
res(await Promise.all(users.map(async user =>
2018-02-04 06:52:33 +01:00
await pack(user, me, { detail: true }))));
2016-12-28 23:49:51 +01:00
}
// Search by Elasticsearch
async function byElasticsearch(res, rej, me, query, offset, max) {
const es = require('../../db/elasticsearch');
es.search({
index: 'misskey',
type: 'user',
body: {
size: max,
from: offset,
query: {
simple_query_string: {
fields: ['username', 'name', 'bio'],
query: query,
default_operator: 'and'
}
}
}
}, async (error, response) => {
if (error) {
console.error(error);
return res(500);
}
if (response.hits.total === 0) {
return res([]);
}
const hits = response.hits.hits.map(hit => new mongo.ObjectID(hit._id));
const users = await User
.find({
_id: {
$in: hits
}
2017-01-17 03:11:22 +01:00
});
2016-12-28 23:49:51 +01:00
// Serialize
res(await Promise.all(users.map(async user =>
2018-02-02 00:21:30 +01:00
await pack(user, me, { detail: true }))));
2016-12-28 23:49:51 +01:00
});
}