snapshot of https://github.com/RocketChat/Rocket.Chat.git (develop, 2026-09-16; blobs > 131072B omitted)

apps/meteor/ee/server/models/raw/Users.ts

@@ -0,0 +1,102 @@
+import type { RocketChatRecordDeleted, IUser, AvailableAgentsAggregation } from '@rocket.chat/core-typings';
+import { queryStatusAgentOnline, UsersRaw } from '@rocket.chat/models';
+import type { Db, Collection, Filter } from 'mongodb';
+
+import { readSecondaryPreferred } from '../../../../server/database/readSecondaryPreferred';
+
+declare module '@rocket.chat/model-typings' {
+	interface IUsersModel {
+		getUnavailableAgents(
+			departmentId: string,
+			customFilter: Filter<AvailableAgentsAggregation>,
+			enabledWhenIdle?: boolean,
+			enabledWhenOffline?: boolean,
+		): Promise<Pick<AvailableAgentsAggregation, 'username'>[]>;
+	}
+}
+
+export class UsersEE extends UsersRaw {
+	constructor(db: Db, trash?: Collection<RocketChatRecordDeleted<IUser>>) {
+		super(db, trash);
+	}
+
+	override getUnavailableAgents(
+		departmentId: string,
+		customFilter: Filter<AvailableAgentsAggregation>,
+		enabledWhenIdle = false,
+		enabledWhenOffline = false,
+	): Promise<Pick<AvailableAgentsAggregation, 'username'>[]> {
+		// if department is provided, remove the agents that are not from the selected department
+		const departmentFilter = departmentId
+			? [
+					{
+						$lookup: {
+							from: 'rocketchat_livechat_department_agents',
+							let: { userId: '$_id' },
+							pipeline: [
+								{
+									$match: {
+										$expr: {
+											$and: [{ $eq: ['$$userId', '$agentId'] }, { $eq: ['$departmentId', departmentId] }],
+										},
+									},
+								},
+							],
+							as: 'department',
+						},
+					},
+					{
+						$match: { department: { $size: 1 } },
+					},
+				]
+			: [];
+
+		return this.col
+			.aggregate<AvailableAgentsAggregation>(
+				[
+					{
+						$match: queryStatusAgentOnline({}, enabledWhenIdle, enabledWhenOffline),
+					},
+					...departmentFilter,
+					{
+						$lookup: {
+							from: 'rocketchat_subscription',
+							localField: '_id',
+							foreignField: 'u._id',
+							pipeline: [{ $match: { $and: [{ t: 'l' }, { open: true }, { onHold: { $ne: true } }] } }],
+							as: 'subs',
+						},
+					},
+					{
+						$project: {
+							'agentId': '$_id',
+							'maxChatsForAgent': { $convert: { input: '$livechat.maxNumberSimultaneousChat', to: 'double', onError: 0, onNull: 0 } },
+							'username': 1,
+							...(departmentId
+								? {
+										'queueInfo.chatsForDepartment': {
+											$size: {
+												$filter: {
+													input: '$subs',
+													as: 'sub',
+													cond: {
+														$and: [{ $eq: ['$$sub.department', departmentId] }],
+													},
+												},
+											},
+										},
+									}
+								: {}),
+							'queueInfo.chats': {
+								$size: '$subs',
+							},
+						},
+					},
+					...(customFilter ? [customFilter] : []),
+					{ $project: { username: 1 } },
+				],
+				{ allowDiskUse: true, readPreference: readSecondaryPreferred() },
+			)
+			.toArray();
+	}
+}