RocketChat/Rocket.Chat

import type { IMessage } from '@rocket.chat/core-typings';
import { isE2EEMessage } from '@rocket.chat/core-typings';
import { useSafely } from '@rocket.chat/fuselage-hooks';
import { useTranslation } from '@rocket.chat/ui-contexts';
import { useState, useEffect } from 'react';

import { e2e } from '../../app/e2e/client/rocketchat.e2e';

export const useDecryptedMessage = (message: IMessage): string => {
    const t = useTranslation();
    const [decryptedMessage, setDecryptedMessage] = useSafely(useState(t('E2E_message_encrypted_placeholder')));

    useEffect(() => {
        if (!isE2EEMessage(message)) {
            return;
        }

        e2e.decryptMessage(message).then((decryptedMsg) => {
            if (decryptedMsg.msg) {
                setDecryptedMessage(decryptedMsg.msg);
            }
        });
    }, [message, t, setDecryptedMessage]);

    return isE2EEMessage(message) ? decryptedMessage : message.msg;
};