import type { IOmnichannelRoom } from '@rocket.chat/core-typings';
import type { FilterOperators } from 'mongodb';

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

callbacks.add(
	'livechat.applyRoomRestrictions',
	async (
		originalQuery: FilterOperators<IOmnichannelRoom> = {},
		{
			unitsFilter,
			userId,
		}: {
			unitsFilter?: string[];
			userId?: string;
		} = {},
	) => {
		return restrictQuery({ originalQuery, unitsFilter, userId });
	},
	callbacks.priority.HIGH,
	'livechat-apply-room-restrictions',
);