overleaf/services/web/app/src/Features/Downloads/ProjectZipStreamManager.js

/* eslint-disable
    camelcase,
    handle-callback-err,
    max-len,
    no-undef,
*/
// TODO: This file was created by bulk-decaffeinate.
// Fix any style issues and re-enable lint.
/*
 * decaffeinate suggestions:
 * DS101: Remove unnecessary use of Array.from
 * DS102: Remove unnecessary code created because of implicit returns
 * DS207: Consider shorter variations of null checks
 * Full docs: https://github.com/decaffeinate/decaffeinate/blob/master/docs/suggestions.md
 */
let ProjectZipStreamManager
const archiver = require('archiver')
const async = require('async')
const logger = require('logger-sharelatex')
const ProjectEntityHandler = require('../Project/ProjectEntityHandler')
const ProjectGetter = require('../Project/ProjectGetter')
const FileStoreHandler = require('../FileStore/FileStoreHandler')

module.exports = ProjectZipStreamManager = {
  createZipStreamForMultipleProjects(project_ids, callback) {
    // We'll build up a zip file that contains multiple zip files

    if (callback == null) {
      callback = function(error, stream) {}
    }
    const archive = archiver('zip')
    archive.on('error', err =>
      logger.err(
        { err, project_ids },
        'something went wrong building archive of project'
      )
    )
    callback(null, archive)

    logger.log({ project_ids }, 'creating zip stream of multiple projects')

    const jobs = []
    for (let project_id of Array.from(project_ids || [])) {
      ;(project_id =>
        jobs.push(callback =>
          ProjectGetter.getProject(project_id, { name: true }, function(
            error,
            project
          ) {
            if (error != null) {
              return callback(error)
            }
            logger.log(
              { project_id, name: project.name },
              'appending project to zip stream'
            )
            return ProjectZipStreamManager.createZipStreamForProject(
              project_id,
              function(error, stream) {
                if (error != null) {
                  return callback(error)
                }
                archive.append(stream, { name: `${project.name}.zip` })
                return stream.on('end', function() {
                  logger.log(
                    { project_id, name: project.name },
                    'zip stream ended'
                  )
                  return callback()
                })
              }
            )
          })
        ))(project_id)
    }

    return async.series(jobs, function() {
      logger.log(
        { project_ids },
        'finished creating zip stream of multiple projects'
      )
      return archive.finalize()
    })
  },

  createZipStreamForProject(project_id, callback) {
    if (callback == null) {
      callback = function(error, stream) {}
    }
    const archive = archiver('zip')
    // return stream immediately before we start adding things to it
    archive.on('error', err =>
      logger.err(
        { err, project_id },
        'something went wrong building archive of project'
      )
    )
    callback(null, archive)
    return this.addAllDocsToArchive(project_id, archive, error => {
      if (error != null) {
        logger.error(
          { err: error, project_id },
          'error adding docs to zip stream'
        )
      }
      return this.addAllFilesToArchive(project_id, archive, error => {
        if (error != null) {
          logger.error(
            { err: error, project_id },
            'error adding files to zip stream'
          )
        }
        return archive.finalize()
      })
    })
  },

  addAllDocsToArchive(project_id, archive, callback) {
    if (callback == null) {
      callback = function(error) {}
    }
    return ProjectEntityHandler.getAllDocs(project_id, function(error, docs) {
      if (error != null) {
        return callback(error)
      }
      const jobs = []
      for (let path in docs) {
        const doc = docs[path]
        ;(function(path, doc) {
          if (path[0] === '/') {
            path = path.slice(1)
          }
          return jobs.push(function(callback) {
            logger.log({ project_id }, 'Adding doc')
            archive.append(doc.lines.join('\n'), { name: path })
            return callback()
          })
        })(path, doc)
      }
      return async.series(jobs, callback)
    })
  },

  addAllFilesToArchive(project_id, archive, callback) {
    if (callback == null) {
      callback = function(error) {}
    }
    return ProjectEntityHandler.getAllFiles(project_id, function(error, files) {
      if (error != null) {
        return callback(error)
      }
      const jobs = []
      for (let path in files) {
        const file = files[path]
        ;((path, file) =>
          jobs.push(callback =>
            FileStoreHandler.getFileStream(project_id, file._id, {}, function(
              error,
              stream
            ) {
              if (error != null) {
                logger.warn(
                  { err: error, project_id, file_id: file._id },
                  'something went wrong adding file to zip archive'
                )
                return callback(err)
              }
              if (path[0] === '/') {
                path = path.slice(1)
              }
              archive.append(stream, { name: path })
              return stream.on('end', () => callback())
            })
          ))(path, file)
      }
      return async.parallelLimit(jobs, 5, callback)
    })
  }
}