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

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

@@ -0,0 +1,94 @@
+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,
+				},
+			},
+		);
+	}
+}