import type { ILivechatDepartment, IOmnichannelRoom } from '@rocket.chat/core-typings';
import { LivechatRooms, LivechatDepartment } from '@rocket.chat/models';

import { callbacks } from '../../../../server/lib/callbacks';
import { cbLogger } from '../../lib/omnichannel/logger';

callbacks.add(
	'livechat.afterForwardChatToDepartment',
	async (options) => {
		const { rid, newDepartmentId } = options;

		const room = await LivechatRooms.findOneById<Pick<IOmnichannelRoom, '_id' | 'departmentAncestors'>>(rid, {
			projection: { departmentAncestors: 1 },
		});
		if (!room) {
			return options;
		}
		await LivechatRooms.unsetPredictedVisitorAbandonmentByRoomId(room._id);

		const department = await LivechatDepartment.findOneById<Pick<ILivechatDepartment, '_id' | 'ancestors'>>(newDepartmentId, {
			projection: { ancestors: 1 },
		});
		if (!department) {
			return options;
		}

		const { departmentAncestors } = room;
		const { ancestors } = department;
		if (!ancestors && !departmentAncestors) {
			return options;
		}

		cbLogger.debug({
			msg: 'Updating department ancestors for room',
			departmentId: newDepartmentId,
			roomId: rid,
		});
		await LivechatRooms.updateDepartmentAncestorsById(room._id, ancestors);

		return options;
	},
	callbacks.priority.MEDIUM,
	'livechat-after-forward-room-to-department',
);