diff --git a/src/app/features/room/message/EncryptedContent.tsx b/src/app/features/room/message/EncryptedContent.tsx index bf0fd199f..f3e614178 100644 --- a/src/app/features/room/message/EncryptedContent.tsx +++ b/src/app/features/room/message/EncryptedContent.tsx @@ -1,5 +1,6 @@ import { MatrixEvent, MatrixEventEvent, MatrixEventHandlerMap } from 'matrix-js-sdk'; import React, { ReactNode, useEffect, useState } from 'react'; +import { MessageEvent } from '../../../../types/matrix/room'; type EncryptedContentProps = { mEvent: MatrixEvent; @@ -7,11 +8,12 @@ type EncryptedContentProps = { }; export function EncryptedContent({ mEvent, children }: EncryptedContentProps) { - const [, toggleDecrypted] = useState(!mEvent.isBeingDecrypted()); + const [, toggleEncrypted] = useState(mEvent.getType() === MessageEvent.RoomMessageEncrypted); useEffect(() => { - const handleDecrypted: MatrixEventHandlerMap[MatrixEventEvent.Decrypted] = () => { - toggleDecrypted((s) => !s); + toggleEncrypted(mEvent.getType() === MessageEvent.RoomMessageEncrypted); + const handleDecrypted: MatrixEventHandlerMap[MatrixEventEvent.Decrypted] = (event) => { + toggleEncrypted(event.getType() === MessageEvent.RoomMessageEncrypted); }; mEvent.on(MatrixEventEvent.Decrypted, handleDecrypted); return () => {