RocketChat/Rocket.Chat

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

declare module '@rocket.chat/ddp-client' {
    // eslint-disable-next-line @typescript-eslint/naming-convention
    interface ServerMethods {
        'e2e.setUserPublicAndPrivateKeys'({ public_key, private_key }: { public_key: string; private_key: string; force?: boolean }): void;
    }
}

Meteor.methods<ServerMethods>({
    async 'e2e.setUserPublicAndPrivateKeys'(keyPair) {
        const userId = Meteor.userId();

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

        if (!keyPair.force) {
            const keys = await Users.fetchKeysByUserId(userId);

            if (keys.private_key && keys.public_key) {
                throw new Meteor.Error('error-keys-already-set', 'Keys already set', {
                    method: 'e2e.setUserPublicAndPrivateKeys',
                });
            }
        }

        await Users.setE2EPublicAndPrivateKeysByUserId(userId, {
            private_key: keyPair.private_key,
            public_key: keyPair.public_key,
        });

        const subscribedRoomIds = await Rooms.getSubscribedRoomIdsWithoutE2EKeys(userId);
        await Rooms.addUserIdToE2EEQueueByRoomIds(subscribedRoomIds, userId);
    },
});