overleaf/services/web/app/coffee/Features/ConnectedUsers/ConnectedUsersManager.coffee

63 lines
1.9 KiB
CoffeeScript
Raw Normal View History

_ = require("underscore")
async = require("async")
Settings = require('settings-sharelatex')
logger = require("logger-sharelatex")
redis = require('redis')
rclient = redis.createClient(Settings.redis.web.port, Settings.redis.web.host)
rclient.auth(Settings.redis.web.password)
ONE_HOUR_IN_S = 60 * 60
buildProjectSetKey = (project_id)-> return "users_in_project:#{project_id}"
buildUserKey = (project_id, user_id)-> return "connected_user:#{project_id}:#{user_id}"
module.exports =
markUserAsConnected: (project_id, user_id, callback = (err)->)->
logger.log project_id:project_id, user_id:user_id, "marking user as connected"
async.series [
(cb)->
rclient.sadd buildProjectSetKey(project_id), user_id, cb
(cb)->
ttl = ONE_HOUR_IN_S * 6
rclient.setex buildUserKey(project_id, user_id), ttl, new Date(), cb
], (err)->
if err?
logger.err err:err, project_id:project_id, user_id:user_id, "problem marking user as connected"
callback(err)
markUserAsDisconnected: (project_id, user_id, callback)->
logger.log project_id:project_id, user_id:user_id, "marking user as disconnected"
async.series [
(cb)->
rclient.srem buildProjectSetKey(project_id), user_id, cb
(cb)->
rclient.del buildUserKey(project_id, user_id), cb
], callback
_getConnectedUser: (project_id, user_id, callback)->
rclient.get buildUserKey(project_id, user_id), (err, result)->
if !result?
connected = false
else
connected = true
callback err, {connected:connected, user_id:user_id}
getConnectedUsers: (project_id, callback)->
self = @
rclient.smembers buildProjectSetKey(project_id), (err, results)->
jobs = results.map (user_id)->
(cb)->
self._getConnectedUser(project_id, user_id, cb)
async.series jobs, (err, users)->
users = _.filter users, (user)->
user.connected
user_ids = _.map users, (user)->
user.user_id
callback err, user_ids