Rocket.Chat/client/startup/userRoles.ts

import { Meteor } from 'meteor/meteor';
import { Tracker } from 'meteor/tracker';

import { UserRoles, RoomRoles, ChatMessage } from '../../app/models/client';
import { Notifications } from '../../app/notifications/client';
import { IRocketChatRecord } from '../../definition/IRocketChatRecord';
import { IRole, IUser } from '../../definition/IUser';
import { handleError } from '../lib/utils/handleError';

Meteor.startup(() => {
	Tracker.autorun(() => {
		if (Meteor.userId()) {
			Meteor.call('getUserRoles', (error: Error, results: IRocketChatRecord[]) => {
				if (error) {
					return handleError(error);
				}

				for (const record of results) {
					UserRoles.upsert({ _id: record._id }, record);
				}
			});

			Notifications.onLogged(
				'roles-change',
				(role: {
					type: 'added' | 'removed' | 'changed';
					_id: IRole['_id'];
					u: Partial<IUser>;
					scope: IRole['scope'];
				}) => {
					if (role.type === 'added') {
						if (role.scope) {
							RoomRoles.upsert(
								{ 'rid': role.scope, 'u._id': role.u._id },
								{ $setOnInsert: { u: role.u }, $addToSet: { roles: role._id } },
							);
						} else {
							UserRoles.upsert(
								{ _id: role.u._id },
								{ $addToSet: { roles: role._id }, $set: { username: role.u.username } },
							);
							ChatMessage.update(
								{ 'u._id': role.u._id },
								{ $addToSet: { roles: role._id } },
								{ multi: true },
							);
						}

						return;
					}

					if (role.type === 'removed') {
						if (role.scope) {
							RoomRoles.update(
								{ 'rid': role.scope, 'u._id': role.u._id },
								{ $pull: { roles: role._id } },
							);
						} else {
							UserRoles.update({ _id: role.u._id }, { $pull: { roles: role._id } });
							ChatMessage.update(
								{ 'u._id': role.u._id },
								{ $pull: { roles: role._id } },
								{ multi: true },
							);
						}

						return;
					}

					if (role.type === 'changed') {
						ChatMessage.update({ roles: role._id }, { $inc: { rerender: 1 } }, { multi: true });
					}
				},
			);
		}
	});
});