ProjectGetter.coffee 3.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. mongojs = require("../../infrastructure/mongojs")
  2. metrics = require("metrics-sharelatex")
  3. db = mongojs.db
  4. ObjectId = mongojs.ObjectId
  5. async = require "async"
  6. Project = require("../../models/Project").Project
  7. logger = require("logger-sharelatex")
  8. LockManager = require("../../infrastructure/LockManager")
  9. module.exports = ProjectGetter =
  10. EXCLUDE_DEPTH: 8
  11. getProjectWithoutDocLines: (project_id, callback=(error, project) ->) ->
  12. excludes = {}
  13. for i in [1..ProjectGetter.EXCLUDE_DEPTH]
  14. excludes["rootFolder#{Array(i).join(".folders")}.docs.lines"] = 0
  15. ProjectGetter.getProject project_id, excludes, callback
  16. getProjectWithOnlyFolders: (project_id, callback=(error, project) ->) ->
  17. excludes = {}
  18. for i in [1..ProjectGetter.EXCLUDE_DEPTH]
  19. excludes["rootFolder#{Array(i).join(".folders")}.docs"] = 0
  20. excludes["rootFolder#{Array(i).join(".folders")}.fileRefs"] = 0
  21. ProjectGetter.getProject project_id, excludes, callback
  22. getProject: (project_id, projection, callback = (error, project) ->) ->
  23. if !project_id?
  24. return callback("no project_id provided")
  25. if projection?.rootFolder
  26. ProjectEntityMongoUpdateHandler = require './ProjectEntityMongoUpdateHandler'
  27. lockKey = ProjectEntityMongoUpdateHandler.getProjectMongoLockKey project_id
  28. LockManager.runWithLock lockKey,
  29. (cb) -> ProjectGetter.getProjectWithoutLock project_id, projection, cb
  30. callback
  31. else
  32. ProjectGetter.getProjectWithoutLock project_id, projection, callback
  33. getProjectWithoutLock: (project_id, projection, callback = (error, project) ->) ->
  34. if typeof(projection) == "function"
  35. callback = projection
  36. if typeof project_id == "string"
  37. query = _id: ObjectId(project_id)
  38. else if project_id instanceof ObjectId
  39. query = _id: project_id
  40. else if project_id?.toString().length == 24 # sometimes mongoose ids are hard to identify, this will catch them
  41. query = _id: ObjectId(project_id.toString())
  42. else
  43. err = new Error("malformed get request")
  44. logger.log project_id:project_id, err:err, type:typeof(project_id), "malformed get request"
  45. return callback(err)
  46. db.projects.find query, projection, (err, project) ->
  47. if err?
  48. logger.err err:err, query:query, projection:projection, "error getting project"
  49. return callback(err)
  50. callback(null, project?[0])
  51. findAllUsersProjects: (
  52. user_id,
  53. fields,
  54. callback = (error, projects={owned: [], readAndWrite: [], readOnly: [], tokenReadAndWrite: [], tokenReadOnly: []}) ->
  55. ) ->
  56. CollaboratorsHandler = require "../Collaborators/CollaboratorsHandler"
  57. Project.find {owner_ref: user_id}, fields, (error, ownedProjects) ->
  58. return callback(error) if error?
  59. CollaboratorsHandler.getProjectsUserIsMemberOf user_id, fields, (error, projects) ->
  60. return callback(error) if error?
  61. result = {
  62. owned: ownedProjects || [],
  63. readAndWrite: projects.readAndWrite || [],
  64. readOnly: projects.readOnly || [],
  65. tokenReadAndWrite: projects.tokenReadAndWrite || [],
  66. tokenReadOnly: projects.tokenReadOnly || []
  67. }
  68. callback null, result
  69. [
  70. 'getProject',
  71. 'getProjectWithoutDocLines'
  72. ].map (method) ->
  73. metrics.timeAsyncMethod(ProjectGetter, method, 'mongo.ProjectGetter', logger)