coriolis-web/server/api/DownloadZipApi.js

/*
Copyright (C) 2019  Cloudbase Solutions SRL
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as
published by the Free Software Foundation, either version 3 of the
License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

// @flow
import AdmZip from 'adm-zip'
import stream from 'stream'

import type { ZipContent } from '../../src/types/ZipContent'

export default (router: any) => {
  router.post('/download-zip', (req, res) => {
    try {
      let contents: ZipContent[] = req.body.contents
      if (!contents || !contents.length || !contents[0].filename || typeof contents[0].content !== 'string') {
        throw new Error()
      }
      let zip = new AdmZip()
      contents.forEach(content => {
        zip.addFile(content.filename, Buffer.alloc(content.content.length, content.content))
      })
      let zipBuffer = zip.toBuffer()
      let readStream = new stream.PassThrough()
      readStream.end(zipBuffer)
      res.set('Content-Disposition', 'attachment; filename=contents.zip')
      res.set('Content-Type', 'text/plain')
      readStream.pipe(res)
    } catch (err) {
      console.error(err)
      res.status(500).json({ error: { message: 'Invalid request body for download zip API' } })
    }
  })
}