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

import $ from 'cafy';
const escapeRegexp = require('escape-regexp');
import User, { pack, ILocalUser, validateUsername, IUser } from '../../../../models/user';
import getParams from '../../get-params';

export const meta = {
	desc: {
		ja: 'ユーザーを検索します。'
	},

	requireCredential: false,

	params: {
		query: $.str.note({
			desc: {
				ja: 'クエリ'
			}
		}),

		offset: $.num.optional.min(0).note({
			default: 0,
			desc: {
				ja: 'オフセット'
			}
		}),

		limit: $.num.optional.range(1, 100).note({
			default: 10,
			desc: {
				ja: '取得する数'
			}
		}),

		localOnly: $.bool.optional.note({
			default: false,
			desc: {
				ja: 'ローカルユーザーのみ検索対象にするか否か'
			}
		}),
	},
};

/**
 * Search a user
 */
export default (params: any, me: ILocalUser) => new Promise(async (res, rej) => {
	const [ps, psErr] = getParams(meta, params);
	if (psErr) return rej(psErr);

	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()))
			}, {
				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);
	}

	// Serialize
	res(await Promise.all(users.map(user => pack(user, me, { detail: true }))));
});