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

import type { ILivechatInquiryRecord, ILivechatPriority } from '@rocket.chat/core-typings';
import { DEFAULT_SLA_CONFIG, LivechatPriorityWeight } from '@rocket.chat/core-typings';
import type { ILivechatInquiryModel } from '@rocket.chat/model-typings';
import { LivechatInquiryRaw } from '@rocket.chat/models';
import type { UpdateResult, Document, WithId } from 'mongodb';

declare module '@rocket.chat/model-typings' {
	interface ILivechatInquiryModel {
		setSlaForRoom(rid: string, sla: { estimatedWaitingTimeQueue: number; slaId: string }): Promise<null | WithId<ILivechatInquiryRecord>>;
		unsetSlaForRoom(rid: string): Promise<null | WithId<ILivechatInquiryRecord>>;
		bulkUnsetSla(roomIds: string[]): Promise<Document | UpdateResult>;
		setPriorityForRoom(rid: string, priority: Pick<ILivechatPriority, '_id' | 'sortItem'>): Promise<null | WithId<ILivechatInquiryRecord>>;
		unsetPriorityForRoom(rid: string): Promise<null | WithId<ILivechatInquiryRecord>>;
	}
}

// Note: Expect a circular dependency error here 😓
export class LivechatInquiryRawEE extends LivechatInquiryRaw implements ILivechatInquiryModel {
	override setSlaForRoom(
		rid: string,
		sla: { estimatedWaitingTimeQueue: number; slaId: string },
	): Promise<null | WithId<ILivechatInquiryRecord>> {
		const { estimatedWaitingTimeQueue, slaId } = sla;

		return this.findOneAndUpdate(
			{ rid },
			{
				$set: {
					slaId,
					estimatedWaitingTimeQueue,
				},
			},
		);
	}

	override unsetSlaForRoom(rid: string): Promise<null | WithId<ILivechatInquiryRecord>> {
		return this.findOneAndUpdate(
			{ rid },
			{
				$unset: {
					slaId: 1,
				},
				$set: {
					estimatedWaitingTimeQueue: DEFAULT_SLA_CONFIG.ESTIMATED_WAITING_TIME_QUEUE,
				},
			},
		);
	}

	override bulkUnsetSla(roomIds: string[]): Promise<Document | UpdateResult> {
		return this.updateMany(
			{
				rid: { $in: roomIds },
			},
			{
				$unset: {
					slaId: 1,
				},
				$set: {
					estimatedWaitingTimeQueue: DEFAULT_SLA_CONFIG.ESTIMATED_WAITING_TIME_QUEUE,
				},
			},
		);
	}

	override setPriorityForRoom(
		rid: string,
		priority: Pick<ILivechatPriority, '_id' | 'sortItem'>,
	): Promise<null | WithId<ILivechatInquiryRecord>> {
		return this.findOneAndUpdate(
			{ rid },
			{
				$set: {
					priorityId: priority._id,
					priorityWeight: priority.sortItem,
				},
			},
		);
	}

	override unsetPriorityForRoom(rid: string): Promise<null | WithId<ILivechatInquiryRecord>> {
		return this.findOneAndUpdate(
			{ rid },
			{
				$unset: {
					priorityId: 1,
				},
				$set: {
					priorityWeight: LivechatPriorityWeight.NOT_SPECIFIED,
				},
			},
		);
	}
}