From 25b96378c7a5e3f8b970dfc988708a3354965521 Mon Sep 17 00:00:00 2001
From: Pedro Lopez <pedroslopez@me.com>
Date: Sun, 15 Mar 2020 15:29:56 -0400
Subject: [PATCH] fix: properly handle unresolved media

---
 src/structures/Message.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/structures/Message.js b/src/structures/Message.js
index 35643d3..e675814 100644
--- a/src/structures/Message.js
+++ b/src/structures/Message.js
@@ -186,7 +186,7 @@ class Message extends Base {
             return undefined;
         }
 
-        const {data, mimetype, filename} = await this.client.pupPage.evaluate(async (msgId) => {
+        const result = await this.client.pupPage.evaluate(async (msgId) => {
             const msg = window.Store.Msg.get(msgId);
             
             if(msg.mediaData.mediaStage != 'RESOLVED') {
@@ -211,7 +211,8 @@ class Message extends Base {
 
         }, this.id._serialized);
 
-        return new MessageMedia(mimetype, data, filename);
+        if(!result) return undefined;
+        return new MessageMedia(result.mimetype, result.data, result.filename);
     }
 
     /**