diff --git a/keyserver/src/fetchers/upload-fetchers.js b/keyserver/src/fetchers/upload-fetchers.js --- a/keyserver/src/fetchers/upload-fetchers.js +++ b/keyserver/src/fetchers/upload-fetchers.js @@ -115,10 +115,10 @@ return result.map(mediaFromRow); } -async function fetchMediaFromMediaMessageContent( +async function fetchUploadsForMessage( viewer: Viewer, mediaMessageContents: $ReadOnlyArray, -): Promise<$ReadOnlyArray> { +): Promise<$ReadOnlyArray> { const uploadIDs = []; for (const mediaContent of mediaMessageContents) { uploadIDs.push(mediaContent.uploadID); @@ -135,6 +135,14 @@ `; const [uploads] = await dbQuery(query); + return uploads; +} + +async function fetchMediaFromMediaMessageContent( + viewer: Viewer, + mediaMessageContents: $ReadOnlyArray, +): Promise<$ReadOnlyArray> { + const uploads = await fetchUploadsForMessage(viewer, mediaMessageContents); const uploadMap = _keyBy('uploadID')(uploads); const media: Media[] = [];