overleaf/services/web/app/coffee/Features/Project/ProjectDuplicator.coffee

97 lines
4 KiB
CoffeeScript
Raw Normal View History

2014-02-12 05:23:40 -05:00
projectCreationHandler = require('./ProjectCreationHandler')
ProjectEntityUpdateHandler = require('./ProjectEntityUpdateHandler')
2014-02-12 05:23:40 -05:00
projectLocator = require('./ProjectLocator')
projectOptionsHandler = require('./ProjectOptionsHandler')
DocumentUpdaterHandler = require("../DocumentUpdater/DocumentUpdaterHandler")
DocstoreManager = require "../Docstore/DocstoreManager"
2016-02-26 03:32:00 -05:00
ProjectGetter = require("./ProjectGetter")
2014-02-12 05:23:40 -05:00
_ = require('underscore')
async = require('async')
logger = require("logger-sharelatex")
2014-02-12 05:23:40 -05:00
2016-03-10 06:58:05 -05:00
module.exports = ProjectDuplicator =
_copyDocs: (owner_id, newProject, originalRootDoc, originalFolder, desFolder, docContents, callback)->
2016-03-10 06:58:05 -05:00
setRootDoc = _.once (doc_id)->
ProjectEntityUpdateHandler.setRootDoc newProject._id, doc_id
docs = originalFolder.docs or []
jobs = docs.map (doc)->
2016-03-10 07:43:36 -05:00
return (cb)->
if !doc?._id?
return callback()
2016-03-10 06:58:05 -05:00
content = docContents[doc._id.toString()]
ProjectEntityUpdateHandler.addDoc newProject._id, desFolder._id, doc.name, content.lines, owner_id, (err, newDoc)->
2016-03-10 06:58:05 -05:00
if err?
logger.err err:err, "error copying doc"
return callback(err)
if originalRootDoc? and newDoc.name == originalRootDoc.name
setRootDoc newDoc._id
2016-03-10 07:43:36 -05:00
cb()
2016-03-10 06:58:05 -05:00
async.series jobs, callback
_copyFiles: (owner_id, newProject, originalProject_id, originalFolder, desFolder, callback)->
fileRefs = originalFolder.fileRefs or []
jobs = fileRefs.map (file)->
2016-03-10 06:58:05 -05:00
return (cb)->
ProjectEntityUpdateHandler.copyFileFromExistingProjectWithProject newProject, desFolder._id, originalProject_id, file, owner_id, cb
async.parallelLimit jobs, 5, callback
2016-03-10 06:58:05 -05:00
_copyFolderRecursivly: (owner_id, newProject_id, originalProject_id, originalRootDoc, originalFolder, desFolder, docContents, callback)->
ProjectGetter.getProject newProject_id, {rootFolder:true, name:true}, (err, newProject)->
2016-03-10 06:58:05 -05:00
if err?
logger.err project_id:newProject_id, "could not get project"
return callback(err)
folders = originalFolder.folders or []
2016-03-10 06:58:05 -05:00
jobs = folders.map (childFolder)->
2016-03-10 06:58:05 -05:00
return (cb)->
if !childFolder?._id?
return cb()
ProjectEntityUpdateHandler.addFolder newProject._id, desFolder?._id, childFolder.name, (err, newFolder)->
2016-03-10 06:58:05 -05:00
return cb(err) if err?
ProjectDuplicator._copyFolderRecursivly owner_id, newProject_id, originalProject_id, originalRootDoc, childFolder, newFolder, docContents, cb
2016-03-10 06:58:05 -05:00
jobs.push (cb)->
ProjectDuplicator._copyFiles owner_id, newProject, originalProject_id, originalFolder, desFolder, cb
2016-03-10 06:58:05 -05:00
jobs.push (cb)->
ProjectDuplicator._copyDocs owner_id, newProject, originalRootDoc, originalFolder, desFolder, docContents, cb
2016-03-10 06:58:05 -05:00
async.series jobs, callback
duplicate: (owner, originalProject_id, newProjectName, callback)->
jobs =
flush: (cb)->
DocumentUpdaterHandler.flushProjectToMongo originalProject_id, cb
originalProject: (cb)->
ProjectGetter.getProject originalProject_id, {compiler:true, rootFolder:true, rootDoc_id:true}, cb
newProject: (cb)->
projectCreationHandler.createBlankProject owner._id, newProjectName, cb
originalRootDoc: (cb)->
projectLocator.findRootDoc {project_id:originalProject_id}, cb
docContentsArray: (cb)->
DocstoreManager.getAllDocs originalProject_id, cb
async.series jobs, (err, results)->
if err?
logger.err err:err, originalProject_id:originalProject_id, "error duplicating project"
return callback(err)
{originalProject, newProject, originalRootDoc, docContentsArray} = results
originalRootDoc = originalRootDoc?[0]
2016-03-10 06:58:05 -05:00
docContents = {}
for docContent in docContentsArray
docContents[docContent._id] = docContent
projectOptionsHandler.setCompiler newProject._id, originalProject.compiler, ->
ProjectDuplicator._copyFolderRecursivly owner._id, newProject._id, originalProject_id, originalRootDoc, originalProject.rootFolder[0], newProject.rootFolder[0], docContents, ->
2016-03-10 06:58:05 -05:00
if err?
logger.err err:err, originalProject_id:originalProject_id, newProjectName:newProjectName, "error cloning project"
callback(err, newProject)