From 954433366eddbb7464d21eeda400e1fbd6775c8d Mon Sep 17 00:00:00 2001
From: Daniel Gasienica <daniel@signal.org>
Date: Wed, 25 Apr 2018 11:16:16 -0400
Subject: [PATCH] Load documents for media gallery

---
 js/views/conversation_view.js | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

diff --git a/js/views/conversation_view.js b/js/views/conversation_view.js
index 21f876017c53..945488a846c7 100644
--- a/js/views/conversation_view.js
+++ b/js/views/conversation_view.js
@@ -577,17 +577,24 @@
       //   events up to its parent elements in the DOM.
       this.closeMenu();
 
-      const media = await Signal.Backbone.Conversation.fetchVisualMediaAttachments({
-        conversationId: this.model.get('id'),
-        WhisperMessageCollection: Whisper.MessageCollection,
+      const conversationId = this.model.get('id');
+      const WhisperMessageCollection = Whisper.MessageCollection;
+      const rawMedia = await Signal.Backbone.Conversation.fetchVisualMediaAttachments({
+        conversationId,
+        WhisperMessageCollection,
       });
+      const documents = await Signal.Backbone.Conversation.fetchFileAttachments({
+        conversationId,
+        WhisperMessageCollection,
+      });
+
       const loadMessages = Signal.Components.PropTypes.Message
         .loadWithObjectURL(Signal.Migrations.loadMessage);
-      const mediaWithObjectURLs = await loadMessages(media);
+      const media = await loadMessages(rawMedia);
 
       const mediaGalleryProps = {
-        media: mediaWithObjectURLs,
-        documents: [],
+        media,
+        documents,
         onItemClick: ({ message }) => {
           const lightboxProps = {
             imageURL: message.objectURL,