import { Abac } from '@rocket.chat/core-services';
import type { IRoom, IRoomAbacRedaction } from '@rocket.chat/core-typings';
import { License } from '@rocket.chat/license';
import { Users } from '@rocket.chat/models';

import { scopeAdminRoomsForAbac } from '../../../../server/api/lib/scopeAdminRoomsForAbac';
import { isABACManagedRoom } from '../../../../server/lib/authorization/isABACManagedRoom';

const redact = <T extends Pick<IRoom, 't' | 'abacAttributes'>>(rooms: T[]): Array<T & IRoomAbacRedaction> =>
	rooms.map((room) => (isABACManagedRoom(room) ? { ...room, abacAttributes: [], abacAttributesRedacted: true } : room));

scopeAdminRoomsForAbac.patch(async (next, rooms, uid) => {
	const managed = License.hasModule('abac') ? rooms.filter(isABACManagedRoom) : [];
	if (!managed.length) {
		return next(rooms, uid);
	}

	try {
		const user = await Users.findOneById(uid, { projection: { _id: 1, username: 1, name: 1 } });
		if (!user) {
			return redact(rooms);
		}

		const scoped = await Abac.scopeRoomsForAdmin(managed, user);
		const scopedById = new Map(scoped.map((room) => [room._id, room]));
		return rooms.map((room) => scopedById.get(room._id) ?? room);
	} catch (err) {
		return redact(rooms);
	}
});