ProjectDuplicator.coffee 2.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  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. Project = require("../../models/Project").Project
  9. _ = require('underscore')
  10. async = require('async')
  11. module.exports =
  12. duplicate: (owner, originalProjectId, newProjectName, callback)->
  13. DocumentUpdaterHandler.flushProjectToMongo originalProjectId, (err) ->
  14. return callback(err) if err?
  15. ProjectGetter.getProject originalProjectId, (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._id, newParentFolder._id, doc.name, content.lines, (err, newDoc)->
  35. if originalRootDoc? and newDoc.name == originalRootDoc.name
  36. setRootDoc newDoc._id
  37. callback()
  38. async.series jobs, callback
  39. copyFiles = (originalFolder, newParentFolder, callback)->
  40. jobs = originalFolder.fileRefs.map (file)->
  41. return (callback)->
  42. projectEntityHandler.copyFileFromExistingProject newProject, newParentFolder._id, originalProject._id, file, callback
  43. async.parallelLimit jobs, 5, callback
  44. copyFolder = (folder, desFolder, callback)->
  45. jobs = folder.folders.map (childFolder)->
  46. return (callback)->
  47. projectEntityHandler.addFolder newProject._id, desFolder._id, childFolder.name, (err, newFolder)->
  48. copyFolder childFolder, newFolder, callback
  49. jobs.push (cb)->
  50. copyDocs folder, desFolder, cb
  51. jobs.push (cb)->
  52. copyFiles folder, desFolder, cb
  53. async.series jobs, callback
  54. copyFolder originalProject.rootFolder[0], newProject.rootFolder[0], ->
  55. callback(err, newProject)