import { Brackets } from 'typeorm'; import { Inject, Injectable } from '@nestjs/common'; import type { UsersRepository, FollowingsRepository } from '@/models/index.js'; import type { Config } from '@/config.js'; import type { User } from '@/models/entities/User.js'; import { Endpoint } from '@/server/api/endpoint-base.js'; import { UserEntityService } from '@/core/entities/UserEntityService.js'; import { DI } from '@/di-symbols.js'; import { sqlLikeEscape } from '@/misc/sql-like-escape.js'; export const meta = { tags: ['users'], requireCredential: false, description: 'Search for a user by username and/or host.', res: { type: 'array', optional: false, nullable: false, items: { type: 'object', optional: false, nullable: false, ref: 'User', }, }, } as const; export const paramDef = { type: 'object', properties: { limit: { type: 'integer', minimum: 1, maximum: 100, default: 10 }, detail: { type: 'boolean', default: true }, username: { type: 'string', nullable: true }, host: { type: 'string', nullable: true }, }, anyOf: [ { required: ['username'] }, { required: ['host'] }, ], } as const; // eslint-disable-next-line import/no-default-export @Injectable() export default class extends Endpoint { constructor( @Inject(DI.config) private config: Config, @Inject(DI.usersRepository) private usersRepository: UsersRepository, @Inject(DI.followingsRepository) private followingsRepository: FollowingsRepository, private userEntityService: UserEntityService, ) { super(meta, paramDef, async (ps, me) => { const setUsernameAndHostQuery = (query = this.usersRepository.createQueryBuilder('user')) => { if (ps.username) { query.andWhere('user.usernameLower LIKE :username', { username: sqlLikeEscape(ps.username.toLowerCase()) + '%' }); } if (ps.host) { if (ps.host === this.config.hostname || ps.host === '.') { query.andWhere('user.host IS NULL'); } else { query.andWhere('user.host LIKE :host', { host: sqlLikeEscape(ps.host.toLowerCase()) + '%', }); } } return query; }; const activeThreshold = new Date(Date.now() - (1000 * 60 * 60 * 24 * 30)); // 30日 let users: User[] = []; if (me) { const followingQuery = this.followingsRepository.createQueryBuilder('following') .select('following.followeeId') .where('following.followerId = :followerId', { followerId: me.id }); const query = setUsernameAndHostQuery() .andWhere(`user.id IN (${ followingQuery.getQuery() })`) .andWhere('user.id != :meId', { meId: me.id }) .andWhere('user.isSuspended = FALSE') .andWhere(new Brackets(qb => { qb .where('user.updatedAt IS NULL') .orWhere('user.updatedAt > :activeThreshold', { activeThreshold: activeThreshold }); })); query.setParameters(followingQuery.getParameters()); users = await query .orderBy('user.usernameLower', 'ASC') .take(ps.limit) .getMany(); if (users.length < ps.limit) { const otherQuery = setUsernameAndHostQuery() .andWhere(`user.id NOT IN (${ followingQuery.getQuery() })`) .andWhere('user.isSuspended = FALSE') .andWhere('user.updatedAt IS NOT NULL'); otherQuery.setParameters(followingQuery.getParameters()); const otherUsers = await otherQuery .orderBy('user.updatedAt', 'DESC') .take(ps.limit - users.length) .getMany(); users = users.concat(otherUsers); } } else { const query = setUsernameAndHostQuery() .andWhere('user.isSuspended = FALSE') .andWhere('user.updatedAt IS NOT NULL'); users = await query .orderBy('user.updatedAt', 'DESC') .take(ps.limit - users.length) .getMany(); } return await this.userEntityService.packMany(users, me, { detail: !!ps.detail }); }); } }