diff --git a/keyserver/src/services/blob.js b/keyserver/src/services/blob.js index 04dca32a3..228409f15 100644 --- a/keyserver/src/services/blob.js +++ b/keyserver/src/services/blob.js @@ -1,75 +1,88 @@ // @flow import blobService from 'lib/facts/blob-service.js'; import { getBlobFetchableURL, makeBlobServiceEndpointURL, } from 'lib/utils/blob-service.js'; import { getMessageForException } from 'lib/utils/errors.js'; async function uploadBlob(blob: Blob, hash: string): Promise { const formData = new FormData(); formData.append('blob_hash', hash); formData.append('blob_data', blob); const uploadBlobResponse = await fetch( makeBlobServiceEndpointURL(blobService.httpEndpoints.UPLOAD_BLOB), { method: blobService.httpEndpoints.UPLOAD_BLOB.method, body: formData, }, ); if (!uploadBlobResponse.ok) { const { status, statusText } = uploadBlobResponse; throw new Error(`Payload upload failed with HTTP ${status}: ${statusText}`); } } async function assignHolder(holder: string, hash: string): Promise { const assignHolderResponse = await fetch( makeBlobServiceEndpointURL(blobService.httpEndpoints.ASSIGN_HOLDER), { method: blobService.httpEndpoints.ASSIGN_HOLDER.method, body: JSON.stringify({ holder, blob_hash: hash, }), headers: { 'content-type': 'application/json', }, }, ); if (!assignHolderResponse.ok) { const { status, statusText } = assignHolderResponse; throw new Error( `Holder assignment failed with HTTP ${status}: ${statusText}`, ); } } async function upload(blob: Blob, hash: string, holder: string): Promise { try { await Promise.all([assignHolder(holder, hash), uploadBlob(blob, hash)]); } catch (e) { throw new Error( `Payload upload failed with: ${ getMessageForException(e) ?? 'unknown error' }`, ); } } -async function download(hash: string): Promise { +async function download(hash: string): Promise< + | { + +found: false, + } + | { + +found: true, + +blob: Blob, + }, +> { const url = getBlobFetchableURL(hash); const response = await fetch(url, { method: blobService.httpEndpoints.GET_BLOB.method, headers: { 'content-type': 'application/json', }, }); - return response.blob(); + + if (!response.ok) { + return { found: false }; + } + const blob = await response.blob(); + return { found: true, blob }; } export { upload, uploadBlob, assignHolder, download };