[FIX] Black screen when try to open a chat with a non-existent department

apps/meteor/app/livechat/imports/server/rest/departments.ts

@@ -12,6 +12,7 @@ import {
 	findDepartmentsBetweenIds,
 	findDepartmentAgents,
 } from '../../../server/api/lib/departments';
+import { DepartmentHelper } from '../../../server/lib/Departments';
 
 API.v1.addRoute(
 	'livechat/department',
@@ -133,15 +134,14 @@ API.v1.addRoute(
 
 			return API.v1.failure();
 		},
-		delete() {
+		async delete() {
 			check(this.urlParams, {
 				_id: String,
 			});
 
-			if (Livechat.removeDepartment(this.urlParams._id)) {
-				return API.v1.success();
-			}
-			return API.v1.failure();
+			await DepartmentHelper.removeDepartment(this.urlParams._id);
+
+			return API.v1.success();
 		},
 	},
 );

apps/meteor/app/livechat/server/lib/Departments.ts

@@ -0,0 +1,56 @@
+import { LivechatDepartment, LivechatDepartmentAgents, LivechatRooms } from '@rocket.chat/models';
+
+import { callbacks } from '../../../../lib/callbacks';
+import { Logger } from '../../../logger/server';
+
+class DepartmentHelperClass {
+	logger = new Logger('Omnichannel:DepartmentHelper');
+
+	async removeDepartment(departmentId: string) {
+		this.logger.debug(`Removing department: ${departmentId}`);
+
+		const department = await LivechatDepartment.findOneById(departmentId);
+		if (!department) {
+			this.logger.debug(`Department not found: ${departmentId}`);
+			throw new Error('error-department-not-found');
+		}
+
+		const { _id } = department;
+
+		const ret = await LivechatDepartment.removeById(_id);
+		if (ret.acknowledged !== true) {
+			this.logger.error(`Department record not removed: ${_id}. Result from db: ${ret}`);
+			throw new Error('error-failed-to-delete-department');
+		}
+		this.logger.debug(`Department record removed: ${_id}`);
+
+		const agentsIds: string[] = await LivechatDepartmentAgents.findAgentsByDepartmentId(department._id)
+			.cursor.map((agent) => agent.agentId)
+			.toArray();
+
+		this.logger.debug(
+			`Performing post-department-removal actions: ${_id}. Removing department agents, unsetting fallback department and removing department from rooms`,
+		);
+
+		const promiseResponses = await Promise.allSettled([
+			LivechatDepartmentAgents.removeByDepartmentId(_id),
+			LivechatDepartment.unsetFallbackDepartmentByDepartmentId(_id),
+			LivechatRooms.bulkRemoveDepartmentAndUnitsFromRooms(_id),
+		]);
+		promiseResponses.forEach((response, index) => {
+			if (response.status === 'rejected') {
+				this.logger.error(`Error while performing post-department-removal actions: ${_id}. Action No: ${index}. Error:`, response.reason);
+			}
+		});
+
+		this.logger.debug(`Post-department-removal actions completed: ${_id}. Notifying callbacks with department and agentsIds`);
+
+		Meteor.defer(() => {
+			callbacks.run('livechat.afterRemoveDepartment', { department, agentsIds });
+		});
+
+		return ret;
+	}
+}
+
+export const DepartmentHelper = new DepartmentHelperClass();

apps/meteor/app/livechat/server/lib/Livechat.js

@@ -1129,30 +1129,6 @@ export const Livechat = {
 		return true;
 	},
 
-	removeDepartment(_id) {
-		check(_id, String);
-
-		const department = LivechatDepartment.findOneById(_id, { fields: { _id: 1 } });
-
-		if (!department) {
-			throw new Meteor.Error('department-not-found', 'Department not found', {
-				method: 'livechat:removeDepartment',
-			});
-		}
-		const ret = LivechatDepartment.removeById(_id);
-		const agentsIds = LivechatDepartmentAgents.findByDepartmentId(_id)
-			.fetch()
-			.map((agent) => agent.agentId);
-		LivechatDepartmentAgents.removeByDepartmentId(_id);
-		LivechatDepartment.unsetFallbackDepartmentByDepartmentId(_id);
-		if (ret) {
-			Meteor.defer(() => {
-				callbacks.run('livechat.afterRemoveDepartment', { department, agentsIds });
-			});
-		}
-		return ret;
-	},
-
 	showConnecting() {
 		const { showConnecting } = RoutingManager.getConfig();
 		return showConnecting;

apps/meteor/app/livechat/server/methods/removeDepartment.js

@@ -1,19 +1,22 @@
 import { Meteor } from 'meteor/meteor';
+import { check } from 'meteor/check';
 
 import { hasPermission } from '../../../authorization';
-import { Livechat } from '../lib/Livechat';
 import { methodDeprecationLogger } from '../../../lib/server/lib/deprecationWarningLogger';
+import { DepartmentHelper } from '../lib/Departments';
 
 Meteor.methods({
 	'livechat:removeDepartment'(_id) {
 		methodDeprecationLogger.warn('livechat:removeDepartment will be deprecated in future versions of Rocket.Chat');
 
+		check(_id, String);
+
 		if (!Meteor.userId() || !hasPermission(Meteor.userId(), 'manage-livechat-departments')) {
 			throw new Meteor.Error('error-not-allowed', 'Not allowed', {
 				method: 'livechat:removeDepartment',
 			});
 		}
 
-		return Livechat.removeDepartment(_id);
+		return DepartmentHelper.removeDepartment(_id);
 	},
 });

apps/meteor/app/models/server/models/LivechatDepartment.js

@@ -152,18 +152,6 @@ export class LivechatDepartment extends Base {
 
 		return this.find(query, options);
 	}
-
-	unsetFallbackDepartmentByDepartmentId(_id) {
-		return this.update(
-			{ fallbackForwardDepartment: _id },
-			{
-				$unset: {
-					fallbackForwardDepartment: 1,
-				},
-			},
-			{ multi: true },
-		);
-	}
 }
 
 export default new LivechatDepartment();

apps/meteor/app/models/server/models/LivechatDepartmentAgents.js

@@ -52,10 +52,6 @@ export class LivechatDepartmentAgents extends Base {
 		this.remove({ departmentId, agentId });
 	}
 
-	removeByDepartmentId(departmentId) {
-		this.remove({ departmentId });
-	}
-
 	getNextAgentForDepartment(departmentId, ignoreAgentId, extraQuery) {
 		const agents = this.findByDepartmentId(departmentId).fetch();
 

apps/meteor/client/views/omnichannel/directory/chats/contextualBar/DepartmentField.tsx

@@ -0,0 +1,28 @@
+import { Box, Skeleton } from '@rocket.chat/fuselage';
+import { useTranslation } from '@rocket.chat/ui-contexts';
+import React from 'react';
+
+import Field from '../../../components/Field';
+import Info from '../../../components/Info';
+import Label from '../../../components/Label';
+import { useDepartmentInfo } from '../../hooks/useDepartmentInfo';
+
+type DepartmentFieldProps = {
+	departmentId: string;
+};
+
+const DepartmentField = ({ departmentId }: DepartmentFieldProps) => {
+	const t = useTranslation();
+	const { data, isLoading, isError } = useDepartmentInfo(departmentId);
+
+	return (
+		<Field>
+			<Label>{t('Department')}</Label>
+			{isLoading && <Skeleton />}
+			{isError && <Box color='danger'>{t('Something_went_wrong')}</Box>}
+			{!isLoading && !isError && <Info>{data?.department?.name || t('Department_not_found')}</Info>}
+		</Field>
+	);
+};
+
+export default DepartmentField;

apps/meteor/client/views/omnichannel/directory/hooks/useDepartmentInfo.ts

@@ -0,0 +1,10 @@
+import type { OperationResult } from '@rocket.chat/rest-typings';
+import { useEndpoint } from '@rocket.chat/ui-contexts';
+import type { UseQueryResult } from '@tanstack/react-query';
+import { useQuery } from '@tanstack/react-query';
+
+export const useDepartmentInfo = (departmentId: string): UseQueryResult<OperationResult<'GET', '/v1/livechat/department/:_id'>> => {
+	const deptInfo = useEndpoint('GET', `/v1/livechat/department/:_id`, { _id: departmentId });
+
+	return useQuery(['livechat/department', departmentId], () => deptInfo({}));
+};

apps/meteor/ee/app/livechat-enterprise/server/hooks/afterRemoveDepartment.ts

@@ -0,0 +1,30 @@
+import type { ILivechatAgent, ILivechatDepartmentRecord } from '@rocket.chat/core-typings';
+import { LivechatDepartment } from '@rocket.chat/models';
+
+import { callbacks } from '../../../../../lib/callbacks';
+import { cbLogger } from '../lib/logger';
+
+const afterRemoveDepartment = async (options: { department: ILivechatDepartmentRecord; agentsId: ILivechatAgent['_id'][] }) => {
+	cbLogger.debug(`Performing post-department-removal actions in EE: ${options?.department?._id}. Removing department from forward list`);
+	if (!options || !options.department) {
+		cbLogger.warn('No department found in options', options);
+		return options;
+	}
+
+	const { department } = options;
+
+	cbLogger.debug(`Removing department from forward list: ${department._id}`);
+	await LivechatDepartment.removeDepartmentFromForwardListById(department._id);
+	cbLogger.debug(`Removed department from forward list: ${department._id}`);
+
+	cbLogger.debug(`Post-department-removal actions completed in EE: ${department._id}`);
+
+	return options;
+};
+
+callbacks.add(
+	'livechat.afterRemoveDepartment',
+	(options) => Promise.await(afterRemoveDepartment(options)),
+	callbacks.priority.HIGH,
+	'livechat-after-remove-department',
+);

apps/meteor/ee/app/models/server/models/LivechatDepartment.js

@@ -55,8 +55,4 @@ overwriteClassOnLicense('livechat-enterprise', LivechatDepartment, {
 	},
 });
 
-LivechatDepartment.prototype.removeDepartmentFromForwardListById = function (_id) {
-	return this.update({ departmentsAllowedToForward: _id }, { $pull: { departmentsAllowedToForward: _id } }, { multi: true });
-};
-
 export default LivechatDepartment;

apps/meteor/ee/server/models/LivechatDepartment.ts

@@ -0,0 +1,6 @@
+import { registerModel } from '@rocket.chat/models';
+
+import { db } from '../../../server/database/utils';
+import { LivechatDepartmentEE } from './raw/LivechatDepartment';
+
+registerModel('ILivechatDepartmentModel', new LivechatDepartmentEE(db));

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

@@ -0,0 +1,15 @@
+import type { ILivechatDepartmentModel } from '@rocket.chat/model-typings';
+
+import { LivechatDepartmentRaw } from '../../../../server/models/raw/LivechatDepartment';
+
+declare module '@rocket.chat/model-typings' {
+	export interface ILivechatDepartmentModel {
+		removeDepartmentFromForwardListById(departmentId: string): Promise<void>;
+	}
+}
+
+export class LivechatDepartmentEE extends LivechatDepartmentRaw implements ILivechatDepartmentModel {
+	async removeDepartmentFromForwardListById(departmentId: string): Promise<void> {
+		await this.updateMany({ departmentsAllowedToForward: departmentId }, { $pull: { departmentsAllowedToForward: departmentId } });
+	}
+}

apps/meteor/ee/server/models/startup.ts

@@ -7,4 +7,5 @@ onLicense('livechat-enterprise', () => {
 	import('./LivechatUnit');
 	import('./LivechatUnitMonitors');
 	import('./LivechatRooms');
+	import('./LivechatDepartment');
 });

apps/meteor/lib/callbacks.ts

@@ -107,7 +107,7 @@ type ChainedCallbackSignatures = {
 		oldDepartmentId: ILivechatDepartmentRecord['_id'];
 	};
 	'livechat.afterInquiryQueued': (inquiry: ILivechatInquiryRecord) => ILivechatInquiryRecord;
-	'livechat.afterRemoveDepartment': (params: { departmentId: ILivechatDepartmentRecord['_id']; agentsId: ILivechatAgent['_id'][] }) => {
+	'livechat.afterRemoveDepartment': (params: { department: ILivechatDepartmentRecord; agentsId: ILivechatAgent['_id'][] }) => {
 		departmentId: ILivechatDepartmentRecord['_id'];
 		agentsId: ILivechatAgent['_id'][];
 	};

apps/meteor/packages/rocketchat-i18n/i18n/en.i18n.json

@@ -1880,6 +1880,7 @@
   "error-email-domain-blacklisted": "The email domain is blacklisted",
   "error-email-send-failed": "Error trying to send email: __message__",
   "error-essential-app-disabled": "Error: a Rocket.Chat App that is essential for this is disabled. Please contact your administrator",
+  "error-failed-to-delete-department": "Failed to delete department",
   "error-field-unavailable": "<strong>__field__</strong> is already in use :(",
   "error-file-too-large": "File is too large",
   "error-forwarding-chat": "Something went wrong while forwarding the chat, Please try again later.",
@@ -5550,4 +5551,4 @@
   "Theme_dark": "Dark",
   "Join_your_team": "Join your team",
   "Create_an_account": "Create an account"
-}
+}

apps/meteor/server/models/raw/LivechatDepartment.ts

@@ -110,4 +110,12 @@ export class LivechatDepartmentRaw extends BaseRaw<ILivechatDepartmentRecord> im
 
 		return this.updateMany(query, update);
 	}
+
+	unsetFallbackDepartmentByDepartmentId(departmentId: string): Promise<Document | UpdateResult> {
+		return this.updateMany({ fallbackDepartment: departmentId }, { $unset: { fallbackDepartment: 1 } });
+	}
+
+	removeDepartmentFromForwardListById(_departmentId: string): Promise<void> {
+		throw new Error('Method not implemented in Community Edition.');
+	}
 }

apps/meteor/server/models/raw/LivechatDepartmentAgents.ts

@@ -1,6 +1,6 @@
 import type { ILivechatDepartmentAgents, RocketChatRecordDeleted } from '@rocket.chat/core-typings';
 import type { FindPaginated, ILivechatDepartmentAgentsModel } from '@rocket.chat/model-typings';
-import type { Collection, FindCursor, Db, Filter, FindOptions } from 'mongodb';
+import type { Collection, FindCursor, Db, Filter, FindOptions, DeleteResult } from 'mongodb';
 
 import { BaseRaw } from './BaseRaw';
 
@@ -105,4 +105,8 @@ export class LivechatDepartmentAgentsRaw extends BaseRaw<ILivechatDepartmentAgen
 	findAgentsByAgentIdAndBusinessHourId(_agentId: string, _businessHourId: string): [] {
 		return [];
 	}
+
+	removeByDepartmentId(departmentId: string): Promise<DeleteResult> {
+		return this.deleteOne({ departmentId });
+	}
 }

apps/meteor/server/models/raw/LivechatRooms.js

@@ -1288,4 +1288,8 @@ export class LivechatRoomsRaw extends BaseRaw {
 			},
 		]);
 	}
+
+	bulkRemoveDepartmentAndUnitsFromRooms(departmentId) {
+		return this.updateMany({ departmentId }, { $unset: { departmentId: 1, departmentAncestors: 1 } });
+	}
 }