upstream-rocketchat/apps/meteor/server/meteor-methods/platform/requestSubscriptionKeys.ts

import { api } from '@rocket.chat/core-services';
import type { ServerMethods } from '@rocket.chat/ddp-client';
import { Subscriptions, Rooms } from '@rocket.chat/models';
import { Meteor } from 'meteor/meteor';

import { methodDeprecationLogger } from '../../lib/deprecationWarningLogger';

declare module '@rocket.chat/ddp-client' {
	// eslint-disable-next-line @typescript-eslint/naming-convention
	interface ServerMethods {
		'e2e.requestSubscriptionKeys'(): boolean;
	}
}

export const requestSubscriptionKeysMethod = async (userId: string): Promise<void> => {
	// Get all encrypted rooms that the user is subscribed to and has no E2E key yet
	const subscriptions = await Subscriptions.findByUserIdWithoutE2E(userId).toArray();
	const roomIds = subscriptions.map((subscription) => subscription.rid);

	// For all subscriptions without E2E key, get the rooms that have encryption enabled
	const query = {
		e2eKeyId: {
			$exists: true,
		},
		_id: {
			$in: roomIds,
		},
	};

	const rooms = Rooms.find(query);
	await rooms.forEach((room) => {
		void api.broadcast('notify.e2e.keyRequest', room._id, room.e2eKeyId);
	});
};

Meteor.methods<ServerMethods>({
	async 'e2e.requestSubscriptionKeys'() {
		methodDeprecationLogger.method('e2e.requestSubscriptionKeys', '9.0.0', '/v1/e2e.requestSubscriptionKeys');

		const userId = Meteor.userId();
		if (!userId) {
			throw new Meteor.Error('error-invalid-user', 'Invalid user', {
				method: 'requestSubscriptionKeys',
			});
		}

		await requestSubscriptionKeysMethod(userId);

		return true;
	},
});