decrypt message text with rsa in handleMessage

Author: sedanurakcil

src/utils/SocketController.tsx

@@ -11,7 +11,7 @@ import {
 } from "../context/actions/roomsActions";
 import { updateTypingUsers } from "../context/actions/typingUsersAction";
 import { IndicatorProps } from "../context/reducers/appReducer";
-import { aesDecrypt } from "./encryption";
+import { aesDecrypt, rsaDecrypt } from "./encryption";
 import { handleLogger } from "../service";
 import * as events from "./events";
 
@@ -28,6 +28,7 @@ const SocketController = () => {
           subscriptionKey,
           secretKey,
           projectName,
+          e2e,
         },
       },
     },
@@ -224,9 +225,24 @@ const SocketController = () => {
 
       let { message } = decryptedMessage;
       const { roomId } = decryptedMessage;
+      const room = rooms.find(({ roomId: Id }) => Id === roomId);
+
+      // If the e2e feature of the client is enabled, the active user has a private key, the room is a one-to-one room (not a group), and the message is not generated by the system
+      if (
+        e2e &&
+        selfUser?.privateKey &&
+        !room?.groupAvatar &&
+        !message.system
+      ) {
+        // decrypt message text using RSA
+        const decryptedMsg = rsaDecrypt({
+          encryptedMsg: message.text,
+          privateKey: selfUser.privateKey,
+        });
+        message = { ...message, text: decryptedMsg };
+      }
 
       message = { ...message, user: { _id: message.senderId } };
-      const room = rooms.find(({ roomId: Id }) => Id === roomId);
 
       // if message belongs community room, add avatar and name to user object
       if (room && room.groupAvatar) {