fix: destructure message before responding (#73)

fix: destructuring of message before responding
KILTprotocol · Aug 7, 2020 · 33718dd · 33718dd
1 parent d08fd4d
commit 33718dd
Showing 1 changed file with 32 additions and 2 deletions.
diff --git a/src/messaging/mongodb-messaging.service.ts b/src/messaging/mongodb-messaging.service.ts
@@ -20,13 +20,43 @@ export class MongoDbMessagingService implements MessagingService {
   public async findBySenderAddress(
     senderAddress: IEncryptedMessage['senderAddress']
   ): Promise<IEncryptedMessage[]> {
-    return await this.messageModel.find({ senderAddress }).exec()
+    return (await this.messageModel.find({ senderAddress }).exec()).map(
+      (singleDBMessage: MessageDB) => {
+        return {
+          message: singleDBMessage.message,
+          nonce: singleDBMessage.nonce,
+          createdAt: singleDBMessage.createdAt,
+          hash: singleDBMessage.hash,
+          signature: singleDBMessage.signature,
+          receiverAddress: singleDBMessage.receiverAddress,
+          senderAddress: singleDBMessage.senderAddress,
+          senderBoxPublicKey: singleDBMessage.senderBoxPublicKey,
+          messageId: singleDBMessage.messageId,
+          receivedAt: singleDBMessage.receivedAt,
+        } as IEncryptedMessage
+      }
+    ) as IEncryptedMessage[]
   }
 
   public async findByReceiverAddress(
     receiverAddress: IEncryptedMessage['receiverAddress']
   ): Promise<IEncryptedMessage[]> {
-    return await this.messageModel.find({ receiverAddress }).exec()
+    return (await this.messageModel.find({ receiverAddress }).exec()).map(
+      (singleDBMessage: MessageDB) => {
+        return {
+          message: singleDBMessage.message,
+          nonce: singleDBMessage.nonce,
+          createdAt: singleDBMessage.createdAt,
+          hash: singleDBMessage.hash,
+          signature: singleDBMessage.signature,
+          receiverAddress: singleDBMessage.receiverAddress,
+          senderAddress: singleDBMessage.senderAddress,
+          senderBoxPublicKey: singleDBMessage.senderBoxPublicKey,
+          messageId: singleDBMessage.messageId,
+          receivedAt: singleDBMessage.receivedAt,
+        } as IEncryptedMessage
+      }
+    ) as IEncryptedMessage[]
   }
 
   public async remove(