ProjectCreationHandler.coffee 3.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  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. ProjectEntityHandler = require('./ProjectEntityHandler')
  9. User = require('../../models/User').User
  10. fs = require('fs')
  11. Path = require "path"
  12. _ = require "underscore"
  13. module.exports = ProjectCreationHandler =
  14. createBlankProject : (owner_id, projectName, callback = (error, project) ->)->
  15. metrics.inc("project-creation")
  16. logger.log owner_id:owner_id, projectName:projectName, "creating blank project"
  17. rootFolder = new Folder {'name':'rootFolder'}
  18. project = new Project
  19. owner_ref : new ObjectId(owner_id)
  20. name : projectName
  21. if Settings.currentImageName?
  22. project.imageName = Settings.currentImageName
  23. project.rootFolder[0] = rootFolder
  24. User.findById owner_id, "ace.spellCheckLanguage", (err, user)->
  25. project.spellCheckLanguage = user.ace.spellCheckLanguage
  26. project.save (err)->
  27. return callback(err) if err?
  28. callback err, project
  29. createBasicProject : (owner_id, projectName, callback = (error, project) ->)->
  30. self = @
  31. @createBlankProject owner_id, projectName, (error, project)->
  32. return callback(error) if error?
  33. self._buildTemplate "mainbasic.tex", owner_id, projectName, (error, docLines)->
  34. return callback(error) if error?
  35. ProjectEntityHandler.addDoc project._id, project.rootFolder[0]._id, "main.tex", docLines, (error, doc)->
  36. if error?
  37. logger.err err:error, "error adding doc when creating basic project"
  38. return callback(error)
  39. ProjectEntityHandler.setRootDoc project._id, doc._id, (error) ->
  40. callback(error, project)
  41. createExampleProject: (owner_id, projectName, callback = (error, project) ->)->
  42. self = @
  43. @createBlankProject owner_id, projectName, (error, project)->
  44. return callback(error) if error?
  45. async.series [
  46. (callback) ->
  47. self._buildTemplate "main.tex", owner_id, projectName, (error, docLines)->
  48. return callback(error) if error?
  49. ProjectEntityHandler.addDoc project._id, project.rootFolder[0]._id, "main.tex", docLines, (error, doc)->
  50. return callback(error) if error?
  51. ProjectEntityHandler.setRootDoc project._id, doc._id, callback
  52. (callback) ->
  53. self._buildTemplate "references.bib", owner_id, projectName, (error, docLines)->
  54. return callback(error) if error?
  55. ProjectEntityHandler.addDoc project._id, project.rootFolder[0]._id, "references.bib", docLines, (error, doc)->
  56. callback(error)
  57. (callback) ->
  58. universePath = Path.resolve(__dirname + "/../../../templates/project_files/universe.jpg")
  59. ProjectEntityHandler.addFile project._id, project.rootFolder[0]._id, "universe.jpg", universePath, callback
  60. ], (error) ->
  61. callback(error, project)
  62. _buildTemplate: (template_name, user_id, project_name, callback = (error, output) ->)->
  63. User.findById user_id, "first_name last_name", (error, user)->
  64. return callback(error) if error?
  65. monthNames = [ "January", "February", "March", "April", "May", "June", "July", "August", "September", "October", "November", "December" ]
  66. templatePath = Path.resolve(__dirname + "/../../../templates/project_files/#{template_name}")
  67. fs.readFile templatePath, (error, template) ->
  68. return callback(error) if error?
  69. data =
  70. project_name: project_name
  71. user: user
  72. year: new Date().getUTCFullYear()
  73. month: monthNames[new Date().getUTCMonth()]
  74. output = _.template(template.toString(), data)
  75. callback null, output.split("\n")
  76. metrics.timeAsyncMethod(
  77. ProjectCreationHandler, 'createBlankProject',
  78. 'mongo.ProjectCreationHandler',
  79. logger
  80. )