1 files changed, 78 insertions, 62 deletions
diff --git a/packages/backend/src/server/api/endpoints/users/search-by-username-and-host.ts b/packages/backend/src/server/api/endpoints/users/search-by-username-and-host.ts
index 6e5bc46bb5..f13df3ee9d 100644
--- a/packages/backend/src/server/api/endpoints/users/search-by-username-and-host.ts
+++ b/packages/backend/src/server/api/endpoints/users/search-by-username-and-host.ts
@@ -1,8 +1,11 @@
 import { Brackets } from 'typeorm';
-import { Followings, Users } from '@/models/index.js';
+import { Inject, Injectable } from '@nestjs/common';
+import type { UsersRepository, FollowingsRepository } from '@/models/index.js';
 import { USER_ACTIVE_THRESHOLD } from '@/const.js';
-import { User } from '@/models/entities/user.js';
-import define from '../../define.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';
 
 export const meta = {
 	tags: ['users'],
@@ -39,78 +42,91 @@ export const paramDef = {
 // TODO: avatar,bannerをJOINしたいけどエラーになる
 
 // eslint-disable-next-line import/no-default-export
-export default define(meta, paramDef, async (ps, me) => {
-	const activeThreshold = new Date(Date.now() - (1000 * 60 * 60 * 24 * 30)); // 30日
+@Injectable()
+export default class extends Endpoint<typeof meta, typeof paramDef> {
+	constructor(
+		@Inject(DI.usersRepository)
+		private usersRepository: UsersRepository,
 
-	if (ps.host) {
-		const q = Users.createQueryBuilder('user')
-			.where('user.isSuspended = FALSE')
-			.andWhere('user.host LIKE :host', { host: ps.host.toLowerCase() + '%' });
+		@Inject(DI.followingsRepository)
+		private followingsRepository: FollowingsRepository,
 
-		if (ps.username) {
-			q.andWhere('user.usernameLower LIKE :username', { username: ps.username.toLowerCase() + '%' });
-		}
+		private userEntityService: UserEntityService,
+	) {
+		super(meta, paramDef, async (ps, me) => {
+			const activeThreshold = new Date(Date.now() - (1000 * 60 * 60 * 24 * 30)); // 30日
 
-		q.andWhere('user.updatedAt IS NOT NULL');
-		q.orderBy('user.updatedAt', 'DESC');
+			if (ps.host) {
+				const q = this.usersRepository.createQueryBuilder('user')
+					.where('user.isSuspended = FALSE')
+					.andWhere('user.host LIKE :host', { host: ps.host.toLowerCase() + '%' });
 
-		const users = await q.take(ps.limit).getMany();
+				if (ps.username) {
+					q.andWhere('user.usernameLower LIKE :username', { username: ps.username.toLowerCase() + '%' });
+				}
 
-		return await Users.packMany(users, me, { detail: ps.detail });
-	} else if (ps.username) {
-		let users: User[] = [];
+				q.andWhere('user.updatedAt IS NOT NULL');
+				q.orderBy('user.updatedAt', 'DESC');
 
-		if (me) {
-			const followingQuery = Followings.createQueryBuilder('following')
-				.select('following.followeeId')
-				.where('following.followerId = :followerId', { followerId: me.id });
+				const users = await q.take(ps.limit).getMany();
 
-			const query = Users.createQueryBuilder('user')
-				.where(`user.id IN (${ followingQuery.getQuery() })`)
-				.andWhere('user.id != :meId', { meId: me.id })
-				.andWhere('user.isSuspended = FALSE')
-				.andWhere('user.usernameLower LIKE :username', { username: ps.username.toLowerCase() + '%' })
-				.andWhere(new Brackets(qb => { qb
-					.where('user.updatedAt IS NULL')
-					.orWhere('user.updatedAt > :activeThreshold', { activeThreshold: activeThreshold });
-				}));
+				return await this.userEntityService.packMany(users, me, { detail: ps.detail });
+			} else if (ps.username) {
+				let users: User[] = [];
 
-			query.setParameters(followingQuery.getParameters());
+				if (me) {
+					const followingQuery = this.followingsRepository.createQueryBuilder('following')
+						.select('following.followeeId')
+						.where('following.followerId = :followerId', { followerId: me.id });
 
-			users = await query
-				.orderBy('user.usernameLower', 'ASC')
-				.take(ps.limit)
-				.getMany();
+					const query = this.usersRepository.createQueryBuilder('user')
+						.where(`user.id IN (${ followingQuery.getQuery() })`)
+						.andWhere('user.id != :meId', { meId: me.id })
+						.andWhere('user.isSuspended = FALSE')
+						.andWhere('user.usernameLower LIKE :username', { username: ps.username.toLowerCase() + '%' })
+						.andWhere(new Brackets(qb => { qb
+							.where('user.updatedAt IS NULL')
+							.orWhere('user.updatedAt > :activeThreshold', { activeThreshold: activeThreshold });
+						}));
 
-			if (users.length < ps.limit) {
-				const otherQuery = await Users.createQueryBuilder('user')
-					.where(`user.id NOT IN (${ followingQuery.getQuery() })`)
-					.andWhere('user.id != :meId', { meId: me.id })
-					.andWhere('user.isSuspended = FALSE')
-					.andWhere('user.usernameLower LIKE :username', { username: ps.username.toLowerCase() + '%' })
-					.andWhere('user.updatedAt IS NOT NULL');
+					query.setParameters(followingQuery.getParameters());
 
-				otherQuery.setParameters(followingQuery.getParameters());
+					users = await query
+						.orderBy('user.usernameLower', 'ASC')
+						.take(ps.limit)
+						.getMany();
 
-				const otherUsers = await otherQuery
-					.orderBy('user.updatedAt', 'DESC')
-					.take(ps.limit - users.length)
-					.getMany();
+					if (users.length < ps.limit) {
+						const otherQuery = await this.usersRepository.createQueryBuilder('user')
+							.where(`user.id NOT IN (${ followingQuery.getQuery() })`)
+							.andWhere('user.id != :meId', { meId: me.id })
+							.andWhere('user.isSuspended = FALSE')
+							.andWhere('user.usernameLower LIKE :username', { username: ps.username.toLowerCase() + '%' })
+							.andWhere('user.updatedAt IS NOT NULL');
 
-				users = users.concat(otherUsers);
+						otherQuery.setParameters(followingQuery.getParameters());
+
+						const otherUsers = await otherQuery
+							.orderBy('user.updatedAt', 'DESC')
+							.take(ps.limit - users.length)
+							.getMany();
+
+						users = users.concat(otherUsers);
+					}
+				} else {
+					users = await this.usersRepository.createQueryBuilder('user')
+						.where('user.isSuspended = FALSE')
+						.andWhere('user.usernameLower LIKE :username', { username: ps.username.toLowerCase() + '%' })
+						.andWhere('user.updatedAt IS NOT NULL')
+						.orderBy('user.updatedAt', 'DESC')
+						.take(ps.limit - users.length)
+						.getMany();
+				}
+
+				return await this.userEntityService.packMany(users, me, { detail: !!ps.detail });
 			}
-		} else {
-			users = await Users.createQueryBuilder('user')
-				.where('user.isSuspended = FALSE')
-				.andWhere('user.usernameLower LIKE :username', { username: ps.username.toLowerCase() + '%' })
-				.andWhere('user.updatedAt IS NOT NULL')
-				.orderBy('user.updatedAt', 'DESC')
-				.take(ps.limit - users.length)
-				.getMany();
-		}
 
-		return await Users.packMany(users, me, { detail: !!ps.detail });
+			return [];
+		});
 	}
-
-	return [];
-});
+}