Added rsaEncryption for message text field

Author: sedanurakcil

src/hooks/useSendMessage.ts

@@ -1,11 +1,11 @@
 import { useContext } from "react";
 import { IMessage } from "react-native-gifted-chat";
+import forge from "node-forge";
 import { sendMessage } from "../utils/socket";
 import { CommtContext } from "../context/Context";
 import { addMessage } from "../context/actions/messagesAction";
 import { updateLastMessage } from "../context/actions/roomsActions";
-import forge from "node-forge";
-import { aesEncrypt } from "../utils/encryption";
+import { aesEncrypt, rsaEncrypt } from "../utils/encryption";
 
 interface onSendMessageProps {
   message: IMessage;
@@ -16,9 +16,10 @@ interface onSendMessageProps {
 const useSendMessage = () => {
   const {
     state: {
-      users: { selfUser },
+      users: { selfUser, users },
+      rooms,
       app: {
-        configs: { secretKey, apiKey, subscriptionKey, projectName },
+        configs: { secretKey, apiKey, subscriptionKey, projectName, e2e },
       },
     },
     dispatch,
@@ -32,13 +33,38 @@ const useSendMessage = () => {
       senderId: message.user._id,
     };
 
-    // create iv and encrypt data for each message
+    let encryptedMessage = messageContent.text;
+
+    // If the e2e feature of the client is enabled and the message is not generated by the system
+    if (e2e && !message.system) {
+      const room = rooms.find((room) => room.roomId === roomId);
+
+      // Find the opposing user ID among one-to-one room participants
+      const oppositeUserId = room?.groupName
+        ? null
+        : room?.participants.find(
+            (id) => id !== selfUser?._id && !id.startsWith("system"),
+          );
+
+      const oppositeUserPck = users.find((user) => user._id === oppositeUserId)
+        ?.publicKey;
+
+      // If the opposite user has a public key, the message text is encrypted using RSA.
+      if (oppositeUserPck) {
+        encryptedMessage = rsaEncrypt({
+          message: encryptedMessage,
+          publicKey: oppositeUserPck,
+        });
+      }
+    }
+
+    // AES encryption is the standard encryption method and encrypts every messages. It encrypts data by creating IV
     const iv = forge.random.getBytesSync(16);
-    const encryptedMessage = aesEncrypt({
+    encryptedMessage = aesEncrypt({
       key: secretKey,
       iv,
       messageData: JSON.stringify({
-        message: messageContent,
+        message: { ...messageContent, text: encryptedMessage },
         roomId,
         chatRoomAuthId,
       }),