ProjectDuplicator.coffee 3.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. projectCreationHandler = require('./ProjectCreationHandler')
  2. projectEntityHandler = require('./ProjectEntityHandler')
  3. projectLocator = require('./ProjectLocator')
  4. projectOptionsHandler = require('./ProjectOptionsHandler')
  5. DocumentUpdaterHandler = require("../DocumentUpdater/DocumentUpdaterHandler")
  6. DocstoreManager = require "../Docstore/DocstoreManager"
  7. ProjectGetter = require("./ProjectGetter")
  8. _ = require('underscore')
  9. async = require('async')
  10. logger = require("logger-sharelatex")
  11. module.exports =
  12. duplicate: (owner, originalProjectId, newProjectName, callback)->
  13. DocumentUpdaterHandler.flushProjectToMongo originalProjectId, (err) ->
  14. return callback(err) if err?
  15. ProjectGetter.getProject originalProjectId, {compiler:true, rootFolder:true, rootDoc_id:true}, (err, originalProject) ->
  16. return callback(err) if err?
  17. projectCreationHandler.createBlankProject owner._id, newProjectName, (err, newProject)->
  18. return callback(err) if err?
  19. projectLocator.findRootDoc {project:originalProject}, (err, originalRootDoc)->
  20. return callback(err) if err?
  21. DocstoreManager.getAllDocs originalProjectId, (err, docContentsArray) ->
  22. return callback(err) if err?
  23. docContents = {}
  24. for docContent in docContentsArray
  25. docContents[docContent._id] = docContent
  26. projectOptionsHandler.setCompiler newProject._id, originalProject.compiler
  27. setRootDoc = _.once (doc_id)->
  28. projectEntityHandler.setRootDoc newProject, doc_id
  29. copyDocs = (originalFolder, newParentFolder, callback)->
  30. jobs = originalFolder.docs.map (doc)->
  31. return (callback)->
  32. content = docContents[doc._id.toString()]
  33. return callback(new Error("doc_id not found: #{doc._id}")) if !content?
  34. projectEntityHandler.addDoc newProject, newParentFolder._id, doc.name, content.lines, (err, newDoc)->
  35. if err?
  36. logger.err err:err, originalProjectId:originalProjectId, newProjectName:newProjectName, "error adding doc"
  37. return callback(err)
  38. if originalRootDoc? and newDoc.name == originalRootDoc.name
  39. setRootDoc newDoc._id
  40. callback()
  41. async.series jobs, callback
  42. copyFiles = (originalFolder, newParentFolder, callback)->
  43. jobs = originalFolder.fileRefs.map (file)->
  44. return (callback)->
  45. projectEntityHandler.copyFileFromExistingProject newProject, newParentFolder._id, originalProject._id, file, callback
  46. async.parallelLimit jobs, 5, callback
  47. copyFolder = (folder, desFolder, callback)->
  48. jobs = folder.folders.map (childFolder)->
  49. return (callback)->
  50. projectEntityHandler.addFolder newProject, desFolder._id, childFolder.name, (err, newFolder)->
  51. copyFolder childFolder, newFolder, callback
  52. jobs.push (cb)->
  53. copyDocs folder, desFolder, cb
  54. jobs.push (cb)->
  55. copyFiles folder, desFolder, cb
  56. async.series jobs, callback
  57. copyFolder originalProject.rootFolder[0], newProject.rootFolder[0], (err)->
  58. if err?
  59. logger.err err:err, originalProjectId:originalProjectId, newProjectName:newProjectName, "error cloning project"
  60. callback(err, newProject)