ProjectCreationHandler.coffee 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130
  1. logger = require('logger-sharelatex')
  2. async = require("async")
  3. metrics = require('metrics-sharelatex')
  4. Settings = require('settings-sharelatex')
  5. ObjectId = require('mongoose').Types.ObjectId
  6. Project = require('../../models/Project').Project
  7. Folder = require('../../models/Folder').Folder
  8. ProjectEntityUpdateHandler = require('./ProjectEntityUpdateHandler')
  9. ProjectDetailsHandler = require('./ProjectDetailsHandler')
  10. HistoryManager = require('../History/HistoryManager')
  11. User = require('../../models/User').User
  12. fs = require('fs')
  13. Path = require "path"
  14. _ = require "underscore"
  15. AnalyticsManger = require("../Analytics/AnalyticsManager")
  16. module.exports = ProjectCreationHandler =
  17. createBlankProject : (owner_id, projectName, attributes, callback = (error, project) ->)->
  18. metrics.inc("project-creation")
  19. if arguments.length == 3
  20. callback = attributes
  21. attributes = null
  22. ProjectDetailsHandler.validateProjectName projectName, (error) ->
  23. return callback(error) if error?
  24. logger.log owner_id:owner_id, projectName:projectName, "creating blank project"
  25. if attributes?
  26. ProjectCreationHandler._createBlankProject owner_id, projectName, attributes, (error, project) ->
  27. return callback(error) if error?
  28. AnalyticsManger.recordEvent(
  29. owner_id, 'project-imported', { projectId: project._id, attributes: attributes }
  30. )
  31. callback(error, project)
  32. else
  33. HistoryManager.initializeProject (error, history) ->
  34. return callback(error) if error?
  35. attributes = overleaf: history: id: history?.overleaf_id
  36. ProjectCreationHandler._createBlankProject owner_id, projectName, attributes, (error, project) ->
  37. return callback(error) if error?
  38. AnalyticsManger.recordEvent(
  39. owner_id, 'project-created', { projectId: project._id }
  40. )
  41. callback(error, project)
  42. _createBlankProject : (owner_id, projectName, attributes, callback = (error, project) ->)->
  43. rootFolder = new Folder {'name':'rootFolder'}
  44. attributes.owner_ref = new ObjectId(owner_id)
  45. attributes.name = projectName
  46. project = new Project attributes
  47. Object.assign(project, attributes)
  48. if Settings.apis?.project_history?.displayHistoryForNewProjects
  49. project.overleaf.history.display = true
  50. if Settings.currentImageName?
  51. # avoid clobbering any imageName already set in attributes (e.g. importedImageName)
  52. project.imageName ?= Settings.currentImageName
  53. project.rootFolder[0] = rootFolder
  54. User.findById owner_id, "ace.spellCheckLanguage", (err, user)->
  55. project.spellCheckLanguage = user.ace.spellCheckLanguage
  56. project.save (err)->
  57. return callback(err) if err?
  58. callback err, project
  59. createProjectFromSnippet : (owner_id, projectName, docLines, callback = (error, project) ->)->
  60. @createBlankProject owner_id, projectName, (error, project)->
  61. return callback(error) if error?
  62. ProjectCreationHandler._createRootDoc project, owner_id, docLines, callback
  63. createBasicProject : (owner_id, projectName, callback = (error, project) ->)->
  64. self = @
  65. @createBlankProject owner_id, projectName, (error, project)->
  66. return callback(error) if error?
  67. self._buildTemplate "mainbasic.tex", owner_id, projectName, (error, docLines)->
  68. return callback(error) if error?
  69. ProjectCreationHandler._createRootDoc project, owner_id, docLines, callback
  70. createExampleProject: (owner_id, projectName, callback = (error, project) ->)->
  71. self = @
  72. @createBlankProject owner_id, projectName, (error, project)->
  73. return callback(error) if error?
  74. async.series [
  75. (callback) ->
  76. self._buildTemplate "main.tex", owner_id, projectName, (error, docLines)->
  77. return callback(error) if error?
  78. ProjectCreationHandler._createRootDoc project, owner_id, docLines, callback
  79. (callback) ->
  80. self._buildTemplate "references.bib", owner_id, projectName, (error, docLines)->
  81. return callback(error) if error?
  82. ProjectEntityUpdateHandler.addDoc project._id, project.rootFolder[0]._id, "references.bib", docLines, owner_id, (error, doc)->
  83. callback(error)
  84. (callback) ->
  85. universePath = Path.resolve(__dirname + "/../../../templates/project_files/universe.jpg")
  86. ProjectEntityUpdateHandler.addFile project._id, project.rootFolder[0]._id, "universe.jpg", universePath, null, owner_id, callback
  87. ], (error) ->
  88. callback(error, project)
  89. _createRootDoc: (project, owner_id, docLines, callback = (error, project) ->)->
  90. ProjectEntityUpdateHandler.addDoc project._id, project.rootFolder[0]._id, "main.tex", docLines, owner_id, (error, doc)->
  91. if error?
  92. logger.err err:error, "error adding root doc when creating project"
  93. return callback(error)
  94. ProjectEntityUpdateHandler.setRootDoc project._id, doc._id, (error) ->
  95. callback(error, project)
  96. _buildTemplate: (template_name, user_id, project_name, callback = (error, output) ->)->
  97. User.findById user_id, "first_name last_name", (error, user)->
  98. return callback(error) if error?
  99. monthNames = [ "January", "February", "March", "April", "May", "June", "July", "August", "September", "October", "November", "December" ]
  100. templatePath = Path.resolve(__dirname + "/../../../templates/project_files/#{template_name}")
  101. fs.readFile templatePath, (error, template) ->
  102. return callback(error) if error?
  103. data =
  104. project_name: project_name
  105. user: user
  106. year: new Date().getUTCFullYear()
  107. month: monthNames[new Date().getUTCMonth()]
  108. output = _.template(template.toString(), data)
  109. callback null, output.split("\n")
  110. metrics.timeAsyncMethod(
  111. ProjectCreationHandler, 'createBlankProject',
  112. 'mongo.ProjectCreationHandler',
  113. logger
  114. )