mirror of
https://github.com/misskey-dev/misskey
synced 2024-12-22 18:48:34 +09:00
100 lines
2.3 KiB
TypeScript
100 lines
2.3 KiB
TypeScript
/**
|
|
* Module dependencies
|
|
*/
|
|
import * as mongo from 'mongodb';
|
|
import it from 'cafy';
|
|
import User from '../../models/user';
|
|
import serialize from '../../serializers/user';
|
|
import config from '../../../conf';
|
|
const escapeRegexp = require('escape-regexp');
|
|
|
|
/**
|
|
* Search a user
|
|
*
|
|
* @param {any} params
|
|
* @param {any} me
|
|
* @return {Promise<any>}
|
|
*/
|
|
module.exports = (params, me) => new Promise(async (res, rej) => {
|
|
// Get 'query' parameter
|
|
const [query, queryError] = it(params.query).expect.string().required().trim().validate(x => x != '').get();
|
|
if (queryError) return rej('invalid query param');
|
|
|
|
// Get 'offset' parameter
|
|
const [offset = 0, offsetErr] = it(params.offset).expect.number().min(0).get();
|
|
if (offsetErr) return rej('invalid offset param');
|
|
|
|
// Get 'max' parameter
|
|
const [max = 10, maxErr] = it(params.max).expect.number().range(1, 30).get();
|
|
if (maxErr) return rej('invalid max param');
|
|
|
|
// If Elasticsearch is available, search by it
|
|
// 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: [{
|
|
username_lower: new RegExp(escapedQuery.toLowerCase())
|
|
}, {
|
|
name: new RegExp(escapedQuery)
|
|
}]
|
|
}, {
|
|
limit: max
|
|
});
|
|
|
|
// Serialize
|
|
res(await Promise.all(users.map(async user =>
|
|
await serialize(user, me, { detail: true }))));
|
|
}
|
|
|
|
// 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
|
|
}
|
|
});
|
|
|
|
// Serialize
|
|
res(await Promise.all(users.map(async user =>
|
|
await serialize(user, me, { detail: true }))));
|
|
});
|
|
}
|