mirror of
https://github.com/overleaf/overleaf.git
synced 2024-11-14 20:40:17 -05:00
80 lines
3.2 KiB
CoffeeScript
80 lines
3.2 KiB
CoffeeScript
archiver = require "archiver"
|
|
async = require "async"
|
|
logger = require "logger-sharelatex"
|
|
ProjectEntityHandler = require "../Project/ProjectEntityHandler"
|
|
FileStoreHandler = require("../FileStore/FileStoreHandler")
|
|
Project = require("../../models/Project").Project
|
|
|
|
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) ->
|
|
Project.findById project_id, "name", (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()
|
|
|
|
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"
|
|
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()
|
|
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
|
|
stream.on "end", () ->
|
|
callback()
|
|
async.parallelLimit jobs, 5, callback
|