upstream-rocketchat/apps/meteor/ee/server/models/raw/Users.ts

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