overleaf/services/web/app/coffee/Features/Downloads/ProjectZipStreamManager.coffee

81 lines
3.2 KiB
CoffeeScript
Raw Normal View History

2014-02-12 05:23:40 -05:00
archiver = require "archiver"
async = require "async"
logger = require "logger-sharelatex"
ProjectEntityHandler = require "../Project/ProjectEntityHandler"
ProjectGetter = require('../Project/ProjectGetter')
2014-02-12 05:23:40 -05:00
FileStoreHandler = require("../FileStore/FileStoreHandler")
module.exports = ProjectZipStreamManager =
createZipStreamForMultipleProjects: (project_ids, callback = (error, stream) ->) ->
# We'll build up a zip file that contains multiple zip files
archive = archiver("zip")
archive.on "error", (err)->
logger.err err:err, project_ids:project_ids, "something went wrong building archive of project"
callback null, archive
logger.log project_ids: project_ids, "creating zip stream of multiple projects"
jobs = []
for project_id in project_ids or []
do (project_id) ->
jobs.push (callback) ->
ProjectGetter.getProject project_id, name: true, (error, project) ->
return callback(error) if error?
logger.log project_id: project_id, name: project.name, "appending project to zip stream"
ProjectZipStreamManager.createZipStreamForProject project_id, (error, stream) ->
return callback(error) if error?
archive.append stream, name: "#{project.name}.zip"
stream.on "end", () ->
logger.log project_id: project_id, name: project.name, "zip stream ended"
callback()
async.series jobs, () ->
logger.log project_ids: project_ids, "finished creating zip stream of multiple projects"
archive.finalize()
2014-02-12 05:23:40 -05:00
createZipStreamForProject: (project_id, callback = (error, stream) ->) ->
archive = archiver("zip")
# return stream immediately before we start adding things to it
archive.on "error", (err)->
logger.err err:err, project_id:project_id, "something went wrong building archive of project"
2014-02-12 05:23:40 -05:00
callback(null, archive)
@addAllDocsToArchive project_id, archive, (error) =>
if error?
logger.error err: error, project_id: project_id, "error adding docs to zip stream"
@addAllFilesToArchive project_id, archive, (error) =>
if error?
logger.error err: error, project_id: project_id, "error adding files to zip stream"
archive.finalize()
addAllDocsToArchive: (project_id, archive, callback = (error) ->) ->
ProjectEntityHandler.getAllDocs project_id, (error, docs) ->
return callback(error) if error?
jobs = []
for path, doc of docs
do (path, doc) ->
path = path.slice(1) if path[0] == "/"
jobs.push (callback) ->
logger.log project_id: project_id, "Adding doc"
archive.append doc.lines.join("\n"), name: path
callback()
2014-02-12 05:23:40 -05:00
async.series jobs, callback
addAllFilesToArchive: (project_id, archive, callback = (error) ->) ->
ProjectEntityHandler.getAllFiles project_id, (error, files) ->
return callback(error) if error?
jobs = []
for path, file of files
do (path, file) ->
jobs.push (callback) ->
FileStoreHandler.getFileStream project_id, file._id, {}, (error, stream) ->
if error?
logger.err err:error, project_id:project_id, file_id:file._id, "something went wrong adding file to zip archive"
return callback(err)
path = path.slice(1) if path[0] == "/"
archive.append stream, name: path
2014-05-16 06:26:58 -04:00
stream.on "end", () ->
callback()
async.parallelLimit jobs, 5, callback