All files / app/storage storage.js

29.41% Statements 5/17
100% Branches 0/0
0% Functions 0/5
31.25% Lines 5/16

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61    1x 1x 1x   1x                                                                                                 1x          
// Google Cloud Storage helpers (upload + signed download URL).
// Authenticates automatically via GOOGLE_APPLICATION_CREDENTIALS.
const { Storage } = require('@google-cloud/storage');
const moment = require('moment');
const config = require('../config');
 
const storage = new Storage({ projectId: config.projectId });
 
// Part IV - Store the zip in GCS with the given object name.
// The source is piped straight into the GCS write stream, so nothing is
// buffered in memory on the way up.
function uploadStream(objectName, source, contentType) {
  const file = storage.bucket(config.storageBucket).file(objectName);
 
  const stream = file.createWriteStream({
    metadata: {
      contentType,
      cacheControl: 'private'
    },
    resumable: false
  });
 
  return new Promise((resolve, reject) => {
    source.on('error', reject);
    stream.on('error', reject);
    stream.on('finish', () => resolve('Ok'));
 
    source.pipe(stream);
  });
}
 
// Part V - Open a read stream on the stored zip so the API can pipe it straight
// to the client (streamed download) without buffering the whole file in memory.
function createReadStream(objectName) {
  return storage
    .bucket(config.storageBucket)
    .file(objectName)
    .createReadStream();
}
 
// Part V - Generate a temporary (2 days) signed download URL for the zip.
async function getDownloadUrl(objectName) {
  const options = {
    action: 'read',
    expires: moment().add(2, 'days').unix() * 1000
  };
 
  const signedUrls = await storage
    .bucket(config.storageBucket)
    .file(objectName)
    .getSignedUrl(options);
 
  return signedUrls[0];
}
 
module.exports = {
  uploadStream,
  createReadStream,
  getDownloadUrl
};