UserGetter.coffee 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. mongojs = require("../../infrastructure/mongojs")
  2. metrics = require('metrics-sharelatex')
  3. logger = require('logger-sharelatex')
  4. db = mongojs.db
  5. ObjectId = mongojs.ObjectId
  6. module.exports = UserGetter =
  7. getUser: (query, projection, callback = (error, user) ->) ->
  8. if query?.email?
  9. return callback(new Error("Don't use getUser to find user by email"), null)
  10. if arguments.length == 2
  11. callback = projection
  12. projection = {}
  13. if typeof query == "string"
  14. try
  15. query = _id: ObjectId(query)
  16. catch e
  17. return callback(null, null)
  18. else if query instanceof ObjectId
  19. query = _id: query
  20. db.users.findOne query, projection, callback
  21. getUserByMainEmail: (email, projection, callback = (error, user) ->) ->
  22. email = email.trim()
  23. if arguments.length == 2
  24. callback = projection
  25. projection = {}
  26. db.users.findOne email: email, projection, callback
  27. getUsers: (user_ids, projection, callback = (error, users) ->) ->
  28. try
  29. user_ids = user_ids.map (u) -> ObjectId(u.toString())
  30. catch error
  31. return callback error
  32. db.users.find { _id: { $in: user_ids} }, projection, callback
  33. getUserOrUserStubById: (user_id, projection, callback = (error, user) ->) ->
  34. try
  35. query = _id: ObjectId(user_id.toString())
  36. catch e
  37. return callback(new Error(e))
  38. db.users.findOne query, projection, (error, user) ->
  39. return callback(error) if error?
  40. return callback(null, user) if user?
  41. db.userstubs.findOne query, projection, callback
  42. [
  43. 'getUser',
  44. 'getUserByMainEmail',
  45. 'getUsers',
  46. 'getUserOrUserStubById'
  47. ].map (method) ->
  48. metrics.timeAsyncMethod UserGetter, method, 'mongo.UserGetter', logger