def start_control(message):
if 'observer' not in session:
socketio.emit('terminate', {}, namespace='/observer', room=request.sid)
return
self_key = 'da:control:sid:' + str(request.sid)
key = 'da:control:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid'])
existing_sid = rr.get(key)
if existing_sid is None or existing_sid.decode() == request.sid:
#sys.stderr.write('Controlling ' + key + '\n')
pipe = rr.pipeline()
pipe.set(self_key, key)
pipe.expire(self_key, 12)
pipe.set(key, request.sid)
pipe.expire(key, 12)
pipe.execute()
int_key = 'da:interviewsession:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid'])
int_sid = rr.get(int_key)
if int_sid is not None:
int_sid = int_sid.decode()
rr.publish(int_sid, json.dumps(dict(messagetype='controllerstart')))
else:
sys.stderr.write('That key ' + key + ' is already taken\n')
key = 'da:session:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid'])
#rr.publish('da:monitor', json.dumps(dict(messagetype='abortcontroller', key=key)))
socketio.emit('abortcontrolling', {'key': key}, namespace='/observer', room=request.sid)
def monitor_chat_message(data):
if 'monitor' not in session:
socketio.emit('terminate', {}, namespace='/monitor', room=request.sid)
return
key = data.get('key', None)
#sys.stderr.write("Key is " + str(key) + "\n")
if key is None:
sys.stderr.write("No key provided\n")
return
m = re.match(r'da:session:uid:(.*):i:(.*):userid:(.*)', key)
if not m:
sys.stderr.write("Invalid key provided\n")
return
session_id = m.group(1)
yaml_filename = m.group(2)
chat_user_id = m.group(3)
key = 'da:interviewsession:uid:' + str(session_id) + ':i:' + str(yaml_filename) + ':userid:' + str(chat_user_id)
sid = rr.get(key)
if sid is None:
sys.stderr.write("No sid for monitor chat message with key " + str(key) + "\n")
return
sid = sid.decode()
secret = secrets.get(sid, None)
if secret is not None:
secret = str(secret)
#obtain_lock(session_id, yaml_filename)
try:
steps, user_dict, encrypted = fetch_user_dict(session_id, yaml_filename, secret=secret)
except Exception as err:
#release_lock(session_id, yaml_filename)
sys.stderr.write("monitor_chat_message: could not get dictionary: " + text_type(err) + "\n")
return
#release_lock(session_id, yaml_filename)
nowtime = datetime.datetime.utcnow()
if encrypted:
message = encrypt_phrase(data['data'], secret)
else:
message = pack_phrase(data['data'])
user_id = session.get('user_id', None)
if user_id is not None:
user_id = int(user_id)
person = UserModel.query.filter_by(id=user_id).first()
chat_mode = user_dict['_internal']['livehelp']['mode']
m = re.match('t([0-9]+)', chat_user_id)
if m:
temp_owner_id = m.group(1)
owner_id = None
else:
temp_owner_id = None
owner_id = chat_user_id
if chat_mode in ['peer', 'peerhelp']:
open_to_peer = True
else:
open_to_peer = False
record = ChatLog(filename=yaml_filename, key=session_id, message=message, encrypted=encrypted, modtime=nowtime, user_id=user_id, temp_owner_id=temp_owner_id, owner_id=owner_id, open_to_peer=open_to_peer)
db.session.add(record)
db.session.commit()
modtime = nice_utc_date(nowtime)
rr.publish(sid, json.dumps(dict(origin='client', messagetype='chat', sid=request.sid, yaml_filename=yaml_filename, uid=session_id, user_id=chat_user_id, message=dict(id=record.id, user_id=record.user_id, first_name=person.first_name, last_name=person.last_name, email=person.email, modtime=modtime, message=data['data'], roles=[role.name for role in person.roles], mode=chat_mode))))
def on_observe(message):
if 'observer' not in session:
socketio.emit('terminate', {}, namespace='/observer', room=request.sid)
return
if request.sid not in threads:
key = 'da:input:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid'])
#sys.stderr.write('Observing ' + key + '\n')
threads[request.sid] = socketio.start_background_task(target=observer_thread, sid=request.sid, key=key)
def on_observe(message):
if 'observer' not in session:
socketio.emit('terminate', {}, namespace='/observer', room=request.sid)
return
if request.sid not in threads:
key = 'da:input:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid'])
#sys.stderr.write('Observing ' + key + '\n')
threads[request.sid] = socketio.start_background_task(target=observer_thread, sid=request.sid, key=key)
def monitor_chat_message(data):
if 'monitor' not in session:
socketio.emit('terminate', {}, namespace='/monitor', room=request.sid)
return
key = data.get('key', None)
#sys.stderr.write("Key is " + str(key) + "\n")
if key is None:
sys.stderr.write("No key provided\n")
return
m = re.match(r'da:session:uid:(.*):i:(.*):userid:(.*)', key)
if not m:
sys.stderr.write("Invalid key provided\n")
return
session_id = m.group(1)
yaml_filename = m.group(2)
chat_user_id = m.group(3)
key = 'da:interviewsession:uid:' + str(session_id) + ':i:' + str(yaml_filename) + ':userid:' + str(chat_user_id)
sid = rr.get(key)
if sid is None:
sys.stderr.write("No sid for monitor chat message with key " + str(key) + "\n")
return
secret = secrets.get(sid, None)
if secret is not None:
secret = str(secret)
#obtain_lock(session_id, yaml_filename)
try:
steps, user_dict, encrypted = fetch_user_dict(session_id, yaml_filename, secret=secret)
except Exception as err:
#release_lock(session_id, yaml_filename)
sys.stderr.write("monitor_chat_message: could not get dictionary: " + unicode(err) + "\n")
return
#release_lock(session_id, yaml_filename)
nowtime = datetime.datetime.utcnow()
if encrypted:
message = encrypt_phrase(data['data'], secret)
else:
message = pack_phrase(data['data'])
user_id = session.get('user_id', None)
if user_id is not None:
user_id = int(user_id)
person = UserModel.query.filter_by(id=user_id).first()
chat_mode = user_dict['_internal']['livehelp']['mode']
m = re.match('t([0-9]+)', chat_user_id)
if m:
temp_owner_id = m.group(1)
owner_id = None
else:
temp_owner_id = None
owner_id = chat_user_id
if chat_mode in ['peer', 'peerhelp']:
open_to_peer = True
else:
open_to_peer = False
record = ChatLog(filename=yaml_filename, key=session_id, message=message, encrypted=encrypted, modtime=nowtime, user_id=user_id, temp_owner_id=temp_owner_id, owner_id=owner_id, open_to_peer=open_to_peer)
db.session.add(record)
db.session.commit()
modtime = nice_utc_date(nowtime)
rr.publish(sid, json.dumps(dict(origin='client', messagetype='chat', sid=request.sid, yaml_filename=yaml_filename, uid=session_id, user_id=chat_user_id, message=dict(id=record.id, user_id=record.user_id, first_name=person.first_name, last_name=person.last_name, email=person.email, modtime=modtime, message=data['data'], roles=[role.name for role in person.roles], mode=chat_mode))))
def monitor_chat_log(data):
if 'monitor' not in session:
socketio.emit('terminate', {}, namespace='/monitor', room=request.sid)
return
key = data.get('key', None)
#sys.stderr.write("Key is " + str(key) + "\n")
if key is None:
sys.stderr.write("No key provided\n")
return
m = re.match(r'da:session:uid:(.*):i:(.*):userid:(.*)', key)
if not m:
sys.stderr.write("Invalid key provided\n")
return
session_id = m.group(1)
yaml_filename = m.group(2)
chat_user_id = m.group(3)
key = 'da:interviewsession:uid:' + str(session_id) + ':i:' + str(yaml_filename) + ':userid:' + str(chat_user_id)
sid = rr.get(key)
if sid is None:
sys.stderr.write("No sid for monitor chat message with key " + str(key) + "\n")
return
sid = sid.decode()
secret = secrets.get(sid, None)
if secret is not None:
secret = str(secret)
#obtain_lock(session_id, yaml_filename)
try:
steps, user_dict, encrypted = fetch_user_dict(session_id, yaml_filename, secret=secret)
except Exception as err:
#release_lock(session_id, yaml_filename)
sys.stderr.write("monitor_chat_log: could not get dictionary: " + text_type(err) + "\n")
return
#release_lock(session_id, yaml_filename)
chat_mode = user_dict['_internal']['livehelp']['mode']
m = re.match('t([0-9]+)', chat_user_id)
if m:
temp_user_id = m.group(1)
user_id = None
else:
temp_user_id = None
user_id = chat_user_id
self_user_id = session.get('user_id', None)
if user_id is not None:
user_id = int(user_id)
if temp_user_id is not None:
temp_user_id = int(temp_user_id)
if self_user_id is not None:
self_user_id = int(self_user_id)
messages = get_chat_log(chat_mode, yaml_filename, session_id, user_id, temp_user_id, secret, self_user_id, None)
socketio.emit('chat_log', {'uid': session_id, 'i': yaml_filename, 'userid': chat_user_id, 'mode': chat_mode, 'data': messages}, namespace='/monitor', room=request.sid)
def monitor_unblock(data):
if 'monitor' not in session:
socketio.emit('terminate', {}, namespace='/monitor', room=request.sid)
return
key = data.get('key', None)
if key is None:
sys.stderr.write("No key provided\n")
return
sys.stderr.write("Unblocking\n")
rr.delete(re.sub(r'^da:session:', 'da:block:', key))
sid = rr.get(re.sub(r'^da:session:', 'da:interviewsession:', key))
if sid is not None:
rr.publish(sid, json.dumps(dict(messagetype='chatpartner', sid=request.sid)))
socketio.emit('unblock', {'key': key}, namespace='/monitor', room=request.sid)
def monitor_unblock(data):
if 'monitor' not in session:
socketio.emit('terminate', {}, namespace='/monitor', room=request.sid)
return
key = data.get('key', None)
if key is None:
sys.stderr.write("No key provided\n")
return
sys.stderr.write("Unblocking\n")
rr.delete(re.sub(r'^da:session:', 'da:block:', key))
sid = rr.get(re.sub(r'^da:session:', 'da:interviewsession:', key))
if sid is not None:
rr.publish(sid, json.dumps(dict(messagetype='chatpartner', sid=request.sid)))
socketio.emit('unblock', {'key': key}, namespace='/monitor', room=request.sid)
def monitor_chat_log(data):
if 'monitor' not in session:
socketio.emit('terminate', {}, namespace='/monitor', room=request.sid)
return
key = data.get('key', None)
#sys.stderr.write("Key is " + str(key) + "\n")
if key is None:
sys.stderr.write("No key provided\n")
return
m = re.match(r'da:session:uid:(.*):i:(.*):userid:(.*)', key)
if not m:
sys.stderr.write("Invalid key provided\n")
return
session_id = m.group(1)
yaml_filename = m.group(2)
chat_user_id = m.group(3)
key = 'da:interviewsession:uid:' + str(session_id) + ':i:' + str(yaml_filename) + ':userid:' + str(chat_user_id)
sid = rr.get(key)
if sid is None:
sys.stderr.write("No sid for monitor chat message with key " + str(key) + "\n")
return
secret = secrets.get(sid, None)
if secret is not None:
secret = str(secret)
#obtain_lock(session_id, yaml_filename)
try:
steps, user_dict, encrypted = fetch_user_dict(session_id, yaml_filename, secret=secret)
except Exception as err:
#release_lock(session_id, yaml_filename)
sys.stderr.write("monitor_chat_log: could not get dictionary: " + unicode(err) + "\n")
return
#release_lock(session_id, yaml_filename)
chat_mode = user_dict['_internal']['livehelp']['mode']
m = re.match('t([0-9]+)', chat_user_id)
if m:
temp_user_id = m.group(1)
user_id = None
else:
temp_user_id = None
user_id = chat_user_id
self_user_id = session.get('user_id', None)
if user_id is not None:
user_id = int(user_id)
if temp_user_id is not None:
temp_user_id = int(temp_user_id)
if self_user_id is not None:
self_user_id = int(self_user_id)
messages = get_chat_log(chat_mode, yaml_filename, session_id, user_id, temp_user_id, secret, self_user_id, None)
socketio.emit('chat_log', {'uid': session_id, 'i': yaml_filename, 'userid': chat_user_id, 'mode': chat_mode, 'data': messages}, namespace='/monitor', room=request.sid)
def on_monitor_connect():
if 'monitor' not in session:
socketio.emit('terminate', {}, namespace='/monitor', room=request.sid)
return
sys.stderr.write('Client connected on monitor and will join room monitor\n')
key = 'da:monitor:' + str(request.sid)
pipe = rr.pipeline()
pipe.set(key, 1)
pipe.expire(key, 60)
pipe.execute()
join_room('monitor')
join_room(request.sid)
user_id = session.get('user_id', None)
if request.sid not in threads:
threads[request.sid] = socketio.start_background_task(target=monitor_thread, sid=request.sid, user_id=user_id)
def on_monitor_connect():
if 'monitor' not in session:
socketio.emit('terminate', {}, namespace='/monitor', room=request.sid)
return
sys.stderr.write('Client connected on monitor and will join room monitor\n')
key = 'da:monitor:' + str(request.sid)
pipe = rr.pipeline()
pipe.set(key, 1)
pipe.expire(key, 60)
pipe.execute()
join_room('monitor')
join_room(request.sid)
user_id = session.get('user_id', None)
if request.sid not in threads:
threads[request.sid] = socketio.start_background_task(target=monitor_thread, sid=request.sid, user_id=user_id)
def chat_log(message):
user_dict = get_dict()
if user_dict is None:
return
chat_mode = user_dict['_internal']['livehelp']['mode']
yaml_filename = session.get('i', None)
session_id = session.get('uid', None)
user_id = session.get('user_id', None)
if user_id is None:
temp_user_id = session.get('tempuser', None)
else:
temp_user_id = None
secret = request.cookies.get('secret', None)
messages = get_chat_log(chat_mode, yaml_filename, session_id, user_id, temp_user_id, secret, user_id, temp_user_id)
socketio.emit('chat_log', {'data': messages}, namespace='/interview', room=request.sid)
def monitor_block(data):
if 'monitor' not in session:
socketio.emit('terminate', {}, namespace='/monitor', room=request.sid)
return
key = data.get('key', None)
if key is None:
sys.stderr.write("No key provided\n")
return
rr.set(re.sub(r'^da:session:', 'da:block:', key), 1)
sid = rr.get(re.sub(r'^da:session:', 'da:interviewsession:', key))
if sid is not None:
rr.publish(sid, json.dumps(dict(messagetype='block', sid=request.sid)))
sys.stderr.write("Blocking\n")
else:
sys.stderr.write("Could not block because could not get sid\n")
socketio.emit('block', {'key': key}, namespace='/monitor', room=request.sid)
def monitor_block(data):
if 'monitor' not in session:
socketio.emit('terminate', {}, namespace='/monitor', room=request.sid)
return
key = data.get('key', None)
if key is None:
sys.stderr.write("No key provided\n")
return
rr.set(re.sub(r'^da:session:', 'da:block:', key), 1)
sid = rr.get(re.sub(r'^da:session:', 'da:interviewsession:', key))
if sid is not None:
rr.publish(sid, json.dumps(dict(messagetype='block', sid=request.sid)))
sys.stderr.write("Blocking\n")
else:
sys.stderr.write("Could not block because could not get sid\n")
socketio.emit('block', {'key': key}, namespace='/monitor', room=request.sid)
def stop_control(message):
if 'observer' not in session:
socketio.emit('terminate', {}, namespace='/observer', room=request.sid)
return
self_key = 'da:control:sid:' + str(request.sid)
key = 'da:control:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid'])
sys.stderr.write('Stop controlling ' + key + '\n')
existing_sid = rr.get(key)
pipe = rr.pipeline()
pipe.delete(self_key)
if existing_sid is not None and existing_sid == request.sid:
pipe.delete(key)
pipe.execute()
sid = rr.get('da:interviewsession:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid']))
if sid is not None:
rr.publish(sid, json.dumps(dict(messagetype='controllerexit', sid=request.sid)))
else:
pipe.execute()
def stop_control(message):
if 'observer' not in session:
socketio.emit('terminate', {}, namespace='/observer', room=request.sid)
return
self_key = 'da:control:sid:' + str(request.sid)
key = 'da:control:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid'])
sys.stderr.write('Stop controlling ' + key + '\n')
existing_sid = rr.get(key)
pipe = rr.pipeline()
pipe.delete(self_key)
if existing_sid is not None and existing_sid == request.sid:
pipe.delete(key)
pipe.execute()
sid = rr.get('da:interviewsession:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid']))
if sid is not None:
sys.stderr.write("Calling controllerexit 1");
rr.publish(sid, json.dumps(dict(messagetype='controllerexit', sid=request.sid)))
else:
pipe.execute()
def observer_changes(message):
sys.stderr.write('observerChanges\n')
if 'observer' not in session:
socketio.emit('terminate', {}, namespace='/observer', room=request.sid)
return
sid = rr.get('da:interviewsession:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid']))
if sid is None:
key = 'da:session:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid'])
sys.stderr.write('observerChanges: sid is none.\n')
if rr.get(key) is None:
sys.stderr.write('observerChanges: session has gone away for good. Sending stopcontrolling.\n')
socketio.emit('stopcontrolling', {'key': key}, namespace='/observer', room=request.sid)
else:
socketio.emit('noconnection', {'key': key}, namespace='/observer', room=request.sid)
else:
sid = sid.decode()
sys.stderr.write('observerChanges: sid exists at ' + time.strftime("%Y-%m-%d %H:%M:%S") + '\n')
rr.publish(sid, json.dumps(dict(messagetype='controllerchanges', sid=request.sid, clicked=message.get('clicked', None), parameters=message['parameters'])))
# sid=request.sid, yaml_filename=str(message['i']), uid=str(message['uid']), user_id=str(message['userid'])
self_key = 'da:control:sid:' + str(request.sid)
key = 'da:control:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid'])
#sys.stderr.write('Controlling ' + key + '\n')
pipe = rr.pipeline()
pipe.set(self_key, key)
pipe.expire(key, 12)
pipe.set(key, request.sid)
pipe.expire(key, 12)
pipe.execute()
def observer_changes(message):
sys.stderr.write('observerChanges\n')
if 'observer' not in session:
socketio.emit('terminate', {}, namespace='/observer', room=request.sid)
return
sid = rr.get('da:interviewsession:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid']))
if sid is None:
key = 'da:session:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid'])
sys.stderr.write('observerChanges: sid is none.\n')
if rr.get(key) is None:
sys.stderr.write('observerChanges: session has gone away for good. Sending stopcontrolling.\n')
socketio.emit('stopcontrolling', {'key': key}, namespace='/observer', room=request.sid)
else:
socketio.emit('noconnection', {'key': key}, namespace='/observer', room=request.sid)
else:
sys.stderr.write('observerChanges: sid exists at ' + time.strftime("%Y-%m-%d %H:%M:%S") + '\n')
rr.publish(sid, json.dumps(dict(messagetype='controllerchanges', sid=request.sid, clicked=message.get('clicked', None), parameters=message['parameters'])))
# sid=request.sid, yaml_filename=str(message['i']), uid=str(message['uid']), user_id=str(message['userid'])
self_key = 'da:control:sid:' + str(request.sid)
key = 'da:control:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid'])
#sys.stderr.write('Controlling ' + key + '\n')
pipe = rr.pipeline()
pipe.set(self_key, key)
pipe.expire(key, 12)
pipe.set(key, request.sid)
pipe.expire(key, 12)
pipe.execute()
def chat_log(message):
user_dict = get_dict()
if user_dict is None:
return
chat_mode = user_dict['_internal']['livehelp']['mode']
yaml_filename = session.get('i', None)
session_id = session.get('uid', None)
user_id = session.get('user_id', None)
if user_id is None:
temp_user_id = session.get('tempuser', None)
else:
temp_user_id = None
if user_id is not None:
user_id = int(user_id)
if temp_user_id is not None:
temp_user_id = int(temp_user_id)
secret = request.cookies.get('secret', None)
if secret is not None:
secret = str(secret)
#sys.stderr.write("chat_log: " + str(repr(user_id)) + " " + str(repr(temp_user_id)) + "\n")
messages = get_chat_log(chat_mode, yaml_filename, session_id, user_id, temp_user_id, secret, user_id, temp_user_id)
socketio.emit('chat_log', {'data': messages}, namespace='/wsinterview', room=request.sid)
def chat_log(message):
user_dict = get_dict()
if user_dict is None:
return
chat_mode = user_dict['_internal']['livehelp']['mode']
yaml_filename = session.get('i', None)
session_id = session.get('uid', None)
user_id = session.get('user_id', None)
if user_id is None:
temp_user_id = session.get('tempuser', None)
else:
temp_user_id = None
if user_id is not None:
user_id = int(user_id)
if temp_user_id is not None:
temp_user_id = int(temp_user_id)
secret = request.cookies.get('secret', None)
if secret is not None:
secret = str(secret)
#sys.stderr.write("chat_log: " + str(repr(user_id)) + " " + str(repr(temp_user_id)) + "\n")
messages = get_chat_log(chat_mode, yaml_filename, session_id, user_id, temp_user_id, secret, user_id, temp_user_id)
socketio.emit('chat_log', {'data': messages}, namespace='/wsinterview', room=request.sid)
def start_control(message):
if 'observer' not in session:
socketio.emit('terminate', {}, namespace='/observer', room=request.sid)
return
self_key = 'da:control:sid:' + str(request.sid)
key = 'da:control:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid'])
existing_sid = rr.get(key)
if existing_sid is None or existing_sid == request.sid:
#sys.stderr.write('Controlling ' + key + '\n')
pipe = rr.pipeline()
pipe.set(self_key, key)
pipe.expire(self_key, 12)
pipe.set(key, request.sid)
pipe.expire(key, 12)
pipe.execute()
int_key = 'da:interviewsession:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid'])
int_sid = rr.get(int_key)
if int_sid is not None:
rr.publish(int_sid, json.dumps(dict(messagetype='controllerstart')))
else:
sys.stderr.write('That key ' + key + ' is already taken\n')
key = 'da:session:uid:' + str(message['uid']) + ':i:' + str(message['i']) + ':userid:' + str(message['userid'])
#rr.publish('da:monitor', json.dumps(dict(messagetype='abortcontroller', key=key)))
socketio.emit('abortcontrolling', {'key': key}, namespace='/observer', room=request.sid)
def observer_thread(sid=None, key=None):
with app.app_context():
sys.stderr.write("Started observer thread for " + str(sid) + "\n")
r = redis.StrictRedis(host=redis_host, port=redis_port, db=redis_offset)
pubsub = r.pubsub()
pubsub.subscribe([key, sid])
for item in pubsub.listen():
sys.stderr.write("2\n" + repr(item) + "\n")
if item['type'] != 'message':
continue
#sys.stderr.write("observer sid: " + str(sid) + ":\n")
data = None
try:
data = json.loads(item['data'].decode())
except:
sys.stderr.write(" observer JSON parse error: " + item['data'].decode() + "\n")
continue
if 'message' in data and data['message'] == "KILL" and (('sid' in data and data['sid'] == sid) or 'sid' not in data):
pubsub.unsubscribe()
sys.stderr.write(" observer unsubscribed and finished for " + str(sid) + "\n")
break
elif 'message' in data:
if data['message'] == "newpage":
#sys.stderr.write(" Got new page for observer\n")
try:
obj = json.loads(r.get(data['key']).decode())
except:
sys.stderr.write(" newpage JSON parse error\n")
continue
socketio.emit('newpage', {'obj': obj}, namespace='/observer', room=sid)
elif data['message'] == "start_being_controlled":
#sys.stderr.write(" got start_being_controlled message with key " + str(data['key']) + "\n")
socketio.emit('start_being_controlled', {'key': data['key']}, namespace='/observer', room=sid)
else:
#sys.stderr.write(" Got parameters for observer\n")
socketio.emit('pushchanges', {'parameters': data}, namespace='/observer', room=sid)
sys.stderr.write(' exiting observer thread for sid ' + str(sid) + '\n')
def observer_thread(sid=None, key=None):
with app.app_context():
sys.stderr.write("Started observer thread for " + str(sid) + "\n")
r = redis.StrictRedis(host=redis_host, db=0)
pubsub = r.pubsub()
pubsub.subscribe([key, sid])
for item in pubsub.listen():
if item['type'] != 'message':
continue
#sys.stderr.write("observer sid: " + str(sid) + ":\n")
data = None
try:
data = json.loads(item['data'])
except:
sys.stderr.write(" observer JSON parse error: " + str(item['data']) + "\n")
continue
if 'message' in data and data['message'] == "KILL" and (('sid' in data and data['sid'] == sid) or 'sid' not in data):
pubsub.unsubscribe()
sys.stderr.write(" observer unsubscribed and finished for " + str(sid) + "\n")
break
elif 'message' in data:
if data['message'] == "newpage":
#sys.stderr.write(" Got new page for observer\n")
try:
obj = json.loads(r.get(data['key']))
except:
sys.stderr.write(" newpage JSON parse error\n")
continue
socketio.emit('newpage', {'obj': obj}, namespace='/observer', room=sid)
elif data['message'] == "start_being_controlled":
#sys.stderr.write(" got start_being_controlled message with key " + str(data['key']) + "\n")
socketio.emit('start_being_controlled', {'key': data['key']}, namespace='/observer', room=sid)
else:
#sys.stderr.write(" Got parameters for observer\n")
socketio.emit('pushchanges', {'parameters': data}, namespace='/observer', room=sid)
sys.stderr.write(' exiting observer thread for sid ' + str(sid) + '\n')
def on_interview_reconnect(data):
sys.stderr.write("Client reconnected on interview\n")
interview_connect()
rr.publish('da:monitor', json.dumps(dict(messagetype='refreshsessions')))
socketio.emit('reconnected', {}, namespace='/wsinterview', room=request.sid)
def background_thread(sid=None, user_id=None, temp_user_id=None):
if user_id is not None:
user_id = int(user_id)
if temp_user_id is not None:
temp_user_id = int(temp_user_id)
with app.app_context():
sys.stderr.write("Started client thread for " + str(sid) + " who is " + str(user_id) + " or " + str(temp_user_id) + "\n")
if user_id is None:
person = None
user_is_temp = True
else:
person = UserModel.query.filter_by(id=user_id).first()
user_is_temp = False
if person is not None and person.timezone is not None:
the_timezone = pytz.timezone(person.timezone)
else:
the_timezone = pytz.timezone(get_default_timezone())
r = redis.StrictRedis(host=redis_host, db=0)
partners = set()
pubsub = r.pubsub()
pubsub.subscribe([sid])
for item in pubsub.listen():
if item['type'] != 'message':
continue
#sys.stderr.write("sid: " + str(sid) + ":\n")
data = None
try:
data = json.loads(item['data'])
except:
sys.stderr.write(" JSON parse error: " + str(item['data']) + "\n")
continue
if data.get('message', None) == "KILL" and (('sid' in data and data['sid'] == sid) or 'sid' not in data):
pubsub.unsubscribe(sid)
sys.stderr.write(" interview unsubscribed and finished for " + str(sid) + "\n")
break
else:
sys.stderr.write(" Got something for sid " + str(sid) + " from " + data.get('origin', "Unknown origin") + "\n")
if 'messagetype' in data:
if data['messagetype'] == 'chat':
#sys.stderr.write(" Emitting interview chat message: " + str(data['message']['message']) + "\n")
if (user_is_temp is True and str(temp_user_id) == str(data['message'].get('temp_user_id', None))) or (user_is_temp is False and str(user_id) == str(data['message'].get('user_id', None))):
data['message']['is_self'] = True
else:
data['message']['is_self'] = False
socketio.emit('chatmessage', {'i': data['yaml_filename'], 'uid': data['uid'], 'userid': data['user_id'], 'data': data['message']}, namespace='/wsinterview', room=sid)
elif data['messagetype'] == 'chatready':
pubsub.subscribe(data['sid'])
partners.add(data['sid'])
sys.stderr.write("chatready 2")
socketio.emit('chatready', {}, namespace='/wsinterview', room=sid)
elif data['messagetype'] == 'departure':
if data['sid'] in partners:
partners.remove(data['sid'])
socketio.emit('departure', {'numpartners': len(partners)}, namespace='/wsinterview', room=sid)
elif data['messagetype'] == 'block':
if data['sid'] in partners:
partners.remove(data['sid'])
socketio.emit('departure', {'numpartners': len(partners)}, namespace='/wsinterview', room=sid)
elif data['messagetype'] == 'chatpartner':
partners.add(data['sid'])
elif data['messagetype'] == 'controllerchanges':
socketio.emit('controllerchanges', {'parameters': data['parameters'], 'clicked': data['clicked']}, namespace='/wsinterview', room=sid)
elif data['messagetype'] == 'controllerstart':
socketio.emit('controllerstart', {}, namespace='/wsinterview', room=sid)
elif data['messagetype'] == 'controllerexit':
socketio.emit('controllerexit', {}, namespace='/wsinterview', room=sid)
# elif data['messagetype'] == "newpage":
# sys.stderr.write(" Got new page for interview\n")
# try:
# obj = json.loads(r.get(data['key']))
# except:
# sys.stderr.write(" newpage JSON parse error\n")
# continue
# socketio.emit('newpage', {'obj': obj}, namespace='/wsinterview', room=sid)
sys.stderr.write(' exiting interview thread for sid ' + str(sid) + '\n')
def interview_connect():
session_id = session.get('uid', None)
if session_id is not None:
user_dict, is_encrypted = get_dict_encrypt()
if is_encrypted:
secret = request.cookies.get('secret', None)
else:
secret = None
if secret is not None:
secret = str(secret)
if user_dict is None:
sys.stderr.write("user_dict did not exist.\n")
socketio.emit('terminate', {}, namespace='/wsinterview', room=request.sid)
return
chat_info = user_dict['_internal']['livehelp']
if chat_info['availability'] == 'unavailable':
sys.stderr.write("Socket started but chat is unavailable.\n")
socketio.emit('terminate', {}, namespace='/wsinterview', room=request.sid)
return
#sys.stderr.write('chat info is ' + str(chat_info) + "\n")
if user_dict['_internal']['livehelp']['mode'] in ['peer', 'peerhelp']:
peer_ok = True
else:
peer_ok = False
yaml_filename = session.get('i', None)
the_user_id = session.get('user_id', 't' + str(session.get('tempuser', None)))
if request.sid not in threads:
#sys.stderr.write('Starting thread for sid ' + str(request.sid) + "\n")
threads[request.sid] = socketio.start_background_task(target=background_thread, sid=request.sid, user_id=session.get('user_id', None), temp_user_id=session.get('tempuser', None))
channel_up = wait_for_channel(rr, request.sid)
if not channel_up:
sys.stderr.write("Channel did not come up.\n")
socketio.emit('terminate', {}, namespace='/wsinterview', room=request.sid)
return
lkey = 'da:ready:uid:' + str(session_id) + ':i:' + str(yaml_filename) + ':userid:' + str(the_user_id)
#sys.stderr.write("Searching: " + lkey + "\n")
if rr.exists(lkey):
lkey_exists = True
else:
lkey_exists = False
if lkey_exists is False and peer_ok is False:
sys.stderr.write("Key does not exist: " + lkey + ".\n")
#socketio.emit('terminate', {}, namespace='/wsinterview', room=request.sid)
#return
failed_to_find_partner = True
found_help = False
if lkey_exists:
partner_keys = rr.lrange(lkey, 0, -1)
#sys.stderr.write("partner_keys is: " + str(type(partner_keys)) + " " + str(partner_keys) + "\n")
if partner_keys is None and not peer_ok:
sys.stderr.write("No partner keys: " + lkey + ".\n")
socketio.emit('terminate', {}, namespace='/wsinterview', room=request.sid)
return
rr.delete(lkey)
for pkey in partner_keys:
pkey = pkey.decode()
#sys.stderr.write("Considering: " + pkey + "\n")
partner_sid = rr.get(pkey)
if partner_sid is not None:
partner_sid = partner_sid.decode()
if re.match(r'^da:monitor:available:.*', pkey):
is_help = True
else:
is_help = False
if is_help and found_help:
continue
#sys.stderr.write("Trying to pub to " + str(partner_sid) + " from " + str(pkey) + "\n")
listeners = rr.publish(partner_sid, json.dumps(dict(messagetype='chatready', uid=session_id, i=yaml_filename, userid=the_user_id, secret=secret, sid=request.sid)))
#sys.stderr.write("Listeners: " + str(listeners) + "\n")
if re.match(r'^da:interviewsession.*', pkey):
rr.publish(request.sid, json.dumps(dict(messagetype='chatready', sid=partner_sid)))
else:
rr.publish(request.sid, json.dumps(dict(messagetype='chatpartner', sid=partner_sid)))
if listeners > 0:
if is_help:
found_help = True
failed_to_find_partner = False
if failed_to_find_partner and peer_ok is False:
sys.stderr.write("Unable to reach any potential chat partners.\n")
#socketio.emit('terminate', {}, namespace='/wsinterview', room=request.sid)
#return
key = 'da:interviewsession:uid:' + str(session_id) + ':i:' + str(yaml_filename) + ':userid:' + str(the_user_id)
rr.set(key, request.sid)
def update_monitor(message):
if 'monitor' not in session:
socketio.emit('terminate', {}, namespace='/monitor', room=request.sid)
return
#sys.stderr.write('received message from ' + str(request.sid) + "\n")
available_for_chat = message['available_for_chat']
new_subscribed_roles = message['subscribed_roles']
new_phone_partners = message['phone_partners_to_add']
term_phone_partners = message['phone_partners_to_terminate']
phone_number = message['phone_number']
phone_number_key = 'da:monitor:phonenumber:' + str(session['user_id'])
if phone_number is None or phone_number == '':
rr.delete(phone_number_key)
else:
pipe = rr.pipeline()
pipe.set(phone_number_key, phone_number)
pipe.expire(phone_number_key, 2592000)
pipe.execute()
phone_partners = dict()
prefix = 'da:phonecode:monitor:' + str(session['user_id']) + ':uid:'
for key in term_phone_partners:
the_code = rr.get(key)
if the_code is not None:
the_code = the_code.decode()
rr.delete(re.sub(r'da:session:uid:', prefix, key))
rr.delete('da:callforward:' + the_code)
if phone_number is None or phone_number == '':
for key in rr.keys(prefix + '*'):
key = key.decode()
the_code = rr.get(key)
if the_code is not None:
the_code = the_code.decode()
rr.delete(key)
rr.delete('da:callforward:' + the_code)
else:
codes_in_use = set()
for key in rr.keys('da:callforward:*'):
key = key.decode()
code = re.sub(r'^da:callforward:', '', key)
codes_in_use.add(code)
for key in rr.keys(prefix + '*'):
key = key.decode()
phone_partners[re.sub(r'^da:phonecode:monitor:[0-9]*:uid:', 'da:session:uid:', key)] = 1
for key in new_phone_partners:
if key in phone_partners:
continue
times = 0
ok = False
while times < 1000:
times += 1
code = "%04d" % random.randint(1000, 9999)
if code in codes_in_use:
continue
ok = True
the_code = code
new_key = re.sub(r'^da:session:uid:', prefix, key)
code_key = 'da:callforward:' + str(code)
pipe = rr.pipeline()
pipe.set(new_key, code)
pipe.set(code_key, phone_number)
pipe.expire(new_key, 300)
pipe.expire(code_key, 300)
pipe.execute()
phone_partners[key] = 1
break
if times >= 1000:
logmessage("update_monitor: could not get a random integer")
#sys.stderr.write('subscribed roles are type ' + str(type(new_subscribed_roles)) + " which is " + str(new_subscribed_roles) + "\n")
monitor_key = 'da:monitor:' + str(request.sid)
pipe = rr.pipeline()
pipe.set(monitor_key, 1)
pipe.expire(monitor_key, 60)
pipe.execute()
key = 'da:monitor:available:' + str(session['user_id'])
key_exists = rr.exists(key)
chat_partners = dict()
for cp_key in rr.hgetall('da:monitor:chatpartners:' + str(session['user_id'])):
cp_key = cp_key.decode()
if rr.get(cp_key) is None:
rr.hdel('da:monitor:chatpartners:' + str(session['user_id']), cp_key)
else:
chat_partners[re.sub('^da:interviewsession:uid:', r'da:session:uid:', cp_key)] = 1
#sys.stderr.write('daAvailableForChat is ' + str(available_for_chat) + " for key " + key + "\n")
if available_for_chat:
pipe = rr.pipeline()
pipe.set(key, request.sid)
pipe.expire(key, 60)
pipe.execute()
elif key_exists:
#sys.stderr.write("Deleting shit\n")
pipe = rr.pipeline()
pipe.delete(key)
for avail_key in rr.keys('da:monitor:role:*:userid:' + str(session['user_id'])):
pipe.delete(avail_key.decode())
pipe.execute()
avail_roles = list()
for key in rr.keys('da:chat:roletype:*'):
avail_roles.append(re.sub(r'^da:chat:roletype:', r'', key.decode()))
sub_role_key = 'da:monitor:userrole:' + str(session['user_id'])
if rr.exists(sub_role_key):
subscribed_roles = decode_dict(rr.hgetall(sub_role_key))
else:
subscribed_roles = dict()
del_mon_role_keys = list()
for role_key in [k for k in new_subscribed_roles.keys()]:
if role_key not in avail_roles:
#sys.stderr.write("role_key is " + str(role_key) + " which is " + str(type(role_key)) + "\n")
del new_subscribed_roles[role_key]
for role_key in [k for k in subscribed_roles.keys()]:
if role_key not in avail_roles:
rr.hdel(sub_role_key, role_key)
del_mon_role_keys.append('da:monitor:role:' + role_key + ':userid:' + str(session['user_id']))
for role_key in [k for k in new_subscribed_roles.keys()]:
if role_key not in subscribed_roles:
rr.hset(sub_role_key, role_key, 1)
subscribed_roles[role_key] = 1
for role_key in [k for k in subscribed_roles.keys()]:
if role_key not in new_subscribed_roles:
rr.hdel(sub_role_key, role_key)
del_mon_role_keys.append('da:monitor:role:' + role_key + ':userid:' + str(session['user_id']))
del subscribed_roles[role_key]
if len(del_mon_role_keys):
pipe = rr.pipeline()
for key in del_mon_role_keys:
pipe.delete(key)
pipe.execute()
if available_for_chat and len(subscribed_roles):
pipe = rr.pipeline()
for role_key in [k for k in subscribed_roles.keys()]:
key = 'da:monitor:role:' + role_key + ':userid:' + str(session['user_id'])
pipe.set(key, 1)
pipe.expire(key, 60)
pipe.execute()
keylist = list()
for key in rr.keys('da:session:*'):
keylist.append(key.decode())
sessions = dict()
for key in keylist:
try:
sessobj = fix_pickle_obj(rr.get(key))
except:
sys.stderr.write('error parsing value of ' + str(key) + " which was " + repr(rr.get(key)) + "\n")
continue
if sessobj.get('chatstatus', None) != 'off':
html = rr.get(re.sub(r'^da:session:', 'da:html:', key))
if html is not None:
html = html.decode()
obj = json.loads(html)
sessobj['browser_title'] = obj.get('browser_title', 'not available')
if rr.exists(re.sub(r'^da:session:', 'da:block:', key)):
sessobj['blocked'] = True
else:
sessobj['blocked'] = False
sessions[key] = sessobj
socketio.emit('updatemonitor', {'available_for_chat': available_for_chat, 'subscribedRoles': subscribed_roles, 'sessions': sessions, 'availRoles': sorted(avail_roles), 'chatPartners': chat_partners, 'phonePartners': phone_partners}, namespace='/monitor', room=request.sid)
def monitor_thread(sid=None, user_id=None):
with app.app_context():
sys.stderr.write("Started monitor thread for " + str(sid) + " who is " + str(user_id) + "\n")
if user_id is not None:
person = UserModel.query.filter_by(id=user_id).first()
else:
person = None
if person is not None and person.timezone is not None:
the_timezone = pytz.timezone(person.timezone)
else:
the_timezone = pytz.timezone(get_default_timezone())
r = redis.StrictRedis(host=redis_host, db=0)
listening_sids = set()
pubsub = r.pubsub()
pubsub.subscribe(['da:monitor', sid])
for item in pubsub.listen():
if item['type'] != 'message':
continue
#sys.stderr.write("monitor sid: " + str(sid) + ":\n")
data = None
try:
data = json.loads(item['data'])
except:
sys.stderr.write(" monitor JSON parse error: " + str(item['data']) + "\n")
continue
if 'message' in data and data['message'] == "KILL":
if item['channel'] == str(sid):
sys.stderr.write(" monitor unsubscribed from all\n")
pubsub.unsubscribe()
for interview_sid in listening_sids:
r.publish(interview_sid, json.dumps(dict(messagetype='departure', sid=sid)))
break
elif item['channel'] != 'da:monitor':
pubsub.unsubscribe(item['channel'])
if data['sid'] in listening_sids:
listening_sids.remove(data['sid'])
sys.stderr.write(" monitor unsubscribed from " + str(item['channel']) + "\n")
continue
else:
sys.stderr.write(" Got something for monitor\n")
if 'messagetype' in data:
#if data['messagetype'] == 'abortcontroller':
# socketio.emit('abortcontroller', {'key': data['key']}, namespace='/monitor', room=sid)
if data['messagetype'] == 'sessionupdate':
#sys.stderr.write(" Got a session update: " + str(data['session']) + "\n")
#sys.stderr.write(" Got a session update\n")
socketio.emit('sessionupdate', {'key': data['key'], 'session': data['session']}, namespace='/monitor', room=sid)
if data['messagetype'] == 'chatready':
pubsub.subscribe(data['sid'])
listening_sids.add(data['sid'])
secrets[data['sid']] = data['secret']
r.hset('da:monitor:chatpartners:' + str(user_id), 'da:interviewsession:uid:' + str(data['uid']) + ':i:' + str(data['i']) + ':userid:' + str(data['userid']), 1)
if str(data['userid']).startswith('t'):
name = word("anonymous visitor") + ' ' + str(data['userid'])[1:]
else:
person = UserModel.query.filter_by(id=data['userid']).first()
if person.first_name:
name = str(person.first_name) + ' ' + str(person.last_name)
else:
name = str(person.email)
sys.stderr.write("chatready 1")
socketio.emit('chatready', {'uid': data['uid'], 'i': data['i'], 'userid': data['userid'], 'name': name}, namespace='/monitor', room=sid)
if data['messagetype'] == 'block':
pubsub.unsubscribe(item['channel'])
if item['channel'] in listening_sids:
listening_sids.remove(item['channel'])
sys.stderr.write(" monitor unsubscribed from " + str(item['channel']) + "\n")
if data['messagetype'] == 'refreshsessions':
socketio.emit('refreshsessions', {}, namespace='/monitor', room=sid)
if data['messagetype'] == 'chat':
#sys.stderr.write(" Emitting monitor chat message: " + str(data['message']['message']) + "\n")
if str(user_id) == str(data['message'].get('user_id', None)):
data['message']['is_self'] = True
else:
data['message']['is_self'] = False
socketio.emit('chatmessage', {'i': data['yaml_filename'], 'uid': data['uid'], 'userid': data['user_id'], 'data': data['message']}, namespace='/monitor', room=sid)
if data['messagetype'] == 'chatstop':
sys.stderr.write(" Chat termination for sid " + data['sid'] + "\n")
pubsub.unsubscribe(data['sid'])
if data['sid'] in secrets:
del secrets[data['sid']]
r.hdel('da:monitor:chatpartners:' + str(user_id), 'da:interviewsession:uid:' + str(data['uid']) + ':i:' + str(data['i']) + ':userid:' + data['userid'])
socketio.emit('chatstop', {'uid': data['uid'], 'i': data['i'], 'userid': data['userid']}, namespace='/monitor', room=sid)
sys.stderr.write(' exiting monitor thread for sid ' + str(sid) + '\n')
def interview_connect():
session_id = session.get('uid', None)
if session_id is not None:
user_dict, is_encrypted = get_dict_encrypt()
if is_encrypted:
secret = request.cookies.get('secret', None)
else:
secret = None
if secret is not None:
secret = str(secret)
if user_dict is None:
sys.stderr.write("user_dict did not exist.\n")
socketio.emit('terminate', {}, namespace='/wsinterview', room=request.sid)
return
chat_info = user_dict['_internal']['livehelp']
if chat_info['availability'] == 'unavailable':
sys.stderr.write("Socket started but chat is unavailable.\n")
socketio.emit('terminate', {}, namespace='/wsinterview', room=request.sid)
return
#sys.stderr.write('chat info is ' + str(chat_info) + "\n")
if user_dict['_internal']['livehelp']['mode'] in ['peer', 'peerhelp']:
peer_ok = True
else:
peer_ok = False
yaml_filename = session.get('i', None)
the_user_id = session.get('user_id', 't' + str(session.get('tempuser', None)))
if request.sid not in threads:
#sys.stderr.write('Starting thread for sid ' + str(request.sid) + "\n")
threads[request.sid] = socketio.start_background_task(target=background_thread, sid=request.sid, user_id=session.get('user_id', None), temp_user_id=session.get('tempuser', None))
channel_up = wait_for_channel(rr, request.sid)
if not channel_up:
sys.stderr.write("Channel did not come up.\n")
socketio.emit('terminate', {}, namespace='/wsinterview', room=request.sid)
return
lkey = 'da:ready:uid:' + str(session_id) + ':i:' + str(yaml_filename) + ':userid:' + str(the_user_id)
#sys.stderr.write("Searching: " + lkey + "\n")
if rr.exists(lkey):
lkey_exists = True
else:
lkey_exists = False
if lkey_exists is False and peer_ok is False:
sys.stderr.write("Key does not exist: " + lkey + ".\n")
#socketio.emit('terminate', {}, namespace='/wsinterview', room=request.sid)
#return
failed_to_find_partner = True
found_help = False
if lkey_exists:
partner_keys = rr.lrange(lkey, 0, -1)
#sys.stderr.write("partner_keys is: " + str(type(partner_keys)) + " " + str(partner_keys) + "\n")
if partner_keys is None and not peer_ok:
sys.stderr.write("No partner keys: " + lkey + ".\n")
socketio.emit('terminate', {}, namespace='/wsinterview', room=request.sid)
return
rr.delete(lkey)
for pkey in partner_keys:
#sys.stderr.write("Considering: " + pkey + "\n")
partner_sid = rr.get(pkey)
if partner_sid is not None:
if re.match(r'^da:monitor:available:.*', pkey):
is_help = True
else:
is_help = False
if is_help and found_help:
continue
#sys.stderr.write("Trying to pub to " + str(partner_sid) + " from " + str(pkey) + "\n")
listeners = rr.publish(partner_sid, json.dumps(dict(messagetype='chatready', uid=session_id, i=yaml_filename, userid=the_user_id, secret=secret, sid=request.sid)))
#sys.stderr.write("Listeners: " + str(listeners) + "\n")
if re.match(r'^da:interviewsession.*', pkey):
rr.publish(request.sid, json.dumps(dict(messagetype='chatready', sid=partner_sid)))
else:
rr.publish(request.sid, json.dumps(dict(messagetype='chatpartner', sid=partner_sid)))
if listeners > 0:
if is_help:
found_help = True
failed_to_find_partner = False
if failed_to_find_partner and peer_ok is False:
sys.stderr.write("Unable to reach any potential chat partners.\n")
#socketio.emit('terminate', {}, namespace='/wsinterview', room=request.sid)
#return
key = 'da:interviewsession:uid:' + str(session_id) + ':i:' + str(yaml_filename) + ':userid:' + str(the_user_id)
rr.set(key, request.sid)
def on_interview_reconnect(data):
sys.stderr.write("Client reconnected on interview\n")
interview_connect()
rr.publish('da:monitor', json.dumps(dict(messagetype='refreshsessions')))
socketio.emit('reconnected', {}, namespace='/wsinterview', room=request.sid)
def handle_message(message):
socketio.emit('mymessage', {'data': "Hello"}, namespace='/wsinterview', room=request.sid)
def background_thread(sid=None, user_id=None, temp_user_id=None):
if user_id is not None:
user_id = int(user_id)
if temp_user_id is not None:
temp_user_id = int(temp_user_id)
with app.app_context():
sys.stderr.write("Started client thread for " + str(sid) + " who is " + str(user_id) + " or " + str(temp_user_id) + "\n")
if user_id is None:
person = None
user_is_temp = True
else:
person = UserModel.query.filter_by(id=user_id).first()
user_is_temp = False
if person is not None and person.timezone is not None:
the_timezone = pytz.timezone(person.timezone)
else:
the_timezone = pytz.timezone(get_default_timezone())
r = redis.StrictRedis(host=redis_host, port=redis_port, db=redis_offset)
partners = set()
pubsub = r.pubsub()
pubsub.subscribe([sid])
for item in pubsub.listen():
sys.stderr.write("0\n" + repr(item) + "\n")
if item['type'] != 'message':
continue
#sys.stderr.write("sid: " + str(sid) + ":\n")
data = None
try:
data = json.loads(item['data'].decode())
except:
sys.stderr.write(" JSON parse error: " + str(item['data'].decode()) + "\n")
continue
if data.get('message', None) == "KILL" and (('sid' in data and data['sid'] == sid) or 'sid' not in data):
pubsub.unsubscribe(sid)
sys.stderr.write(" interview unsubscribed and finished for " + str(sid) + "\n")
break
else:
sys.stderr.write(" Got something for sid " + str(sid) + " from " + data.get('origin', "Unknown origin") + "\n")
if 'messagetype' in data:
if data['messagetype'] == 'chat':
#sys.stderr.write(" Emitting interview chat message: " + str(data['message']['message']) + "\n")
if (user_is_temp is True and str(temp_user_id) == str(data['message'].get('temp_user_id', None))) or (user_is_temp is False and str(user_id) == str(data['message'].get('user_id', None))):
data['message']['is_self'] = True
else:
data['message']['is_self'] = False
socketio.emit('chatmessage', {'i': data['yaml_filename'], 'uid': data['uid'], 'userid': data['user_id'], 'data': data['message']}, namespace='/wsinterview', room=sid)
elif data['messagetype'] == 'chatready':
pubsub.subscribe(data['sid'])
partners.add(data['sid'])
sys.stderr.write("chatready 2")
socketio.emit('chatready', {}, namespace='/wsinterview', room=sid)
elif data['messagetype'] == 'departure':
if data['sid'] in partners:
partners.remove(data['sid'])
socketio.emit('departure', {'numpartners': len(partners)}, namespace='/wsinterview', room=sid)
elif data['messagetype'] == 'block':
if data['sid'] in partners:
partners.remove(data['sid'])
socketio.emit('departure', {'numpartners': len(partners)}, namespace='/wsinterview', room=sid)
elif data['messagetype'] == 'chatpartner':
partners.add(data['sid'])
elif data['messagetype'] == 'controllerchanges':
socketio.emit('controllerchanges', {'parameters': data['parameters'], 'clicked': data['clicked']}, namespace='/wsinterview', room=sid)
elif data['messagetype'] == 'controllerstart':
socketio.emit('controllerstart', {}, namespace='/wsinterview', room=sid)
elif data['messagetype'] == 'controllerexit':
socketio.emit('controllerexit', {}, namespace='/wsinterview', room=sid)
# elif data['messagetype'] == "newpage":
# sys.stderr.write(" Got new page for interview\n")
# try:
# obj = json.loads(r.get(data['key']))
# except:
# sys.stderr.write(" newpage JSON parse error\n")
# continue
# socketio.emit('newpage', {'obj': obj}, namespace='/wsinterview', room=sid)
sys.stderr.write(' exiting interview thread for sid ' + str(sid) + '\n')
def on_observer_connect():
if 'observer' not in session:
socketio.emit('terminate', {}, namespace='/observer', room=request.sid)
return
join_room(request.sid)
def terminate_monitor_connection():
sys.stderr.write("terminate_monitor_connection\n")
# hopefully the disconnect will be triggered
# if request.sid in threads:
# rr.publish(request.sid, json.dumps(dict(origin='client', message='KILL', sid=request.sid)))
socketio.emit('terminate', {}, namespace='/monitor', room=request.sid)
def monitor_thread(sid=None, user_id=None):
with app.app_context():
sys.stderr.write("Started monitor thread for " + str(sid) + " who is " + str(user_id) + "\n")
if user_id is not None:
person = UserModel.query.filter_by(id=user_id).first()
else:
person = None
if person is not None and person.timezone is not None:
the_timezone = pytz.timezone(person.timezone)
else:
the_timezone = pytz.timezone(get_default_timezone())
r = redis.StrictRedis(host=redis_host, port=redis_port, db=redis_offset)
listening_sids = set()
pubsub = r.pubsub()
pubsub.subscribe(['da:monitor', sid])
for item in pubsub.listen():
sys.stderr.write("1\n" + repr(item) + "\n")
if item['type'] != 'message':
continue
#sys.stderr.write("monitor sid: " + str(sid) + ":\n")
data = None
try:
data = json.loads(item['data'].decode())
except:
sys.stderr.write(" monitor JSON parse error: " + item['data'].decode() + "\n")
continue
if 'message' in data and data['message'] == "KILL":
if item['channel'].decode() == str(sid):
sys.stderr.write(" monitor unsubscribed from all\n")
pubsub.unsubscribe()
for interview_sid in listening_sids:
r.publish(interview_sid, json.dumps(dict(messagetype='departure', sid=sid)))
break
elif item['channel'].decode() != 'da:monitor':
pubsub.unsubscribe(item['channel'].decode())
if data['sid'] in listening_sids:
listening_sids.remove(data['sid'])
sys.stderr.write(" monitor unsubscribed from " + item['channel'].decode() + "\n")
continue
else:
sys.stderr.write(" Got something for monitor\n")
if 'messagetype' in data:
#if data['messagetype'] == 'abortcontroller':
# socketio.emit('abortcontroller', {'key': data['key']}, namespace='/monitor', room=sid)
if data['messagetype'] == 'sessionupdate':
#sys.stderr.write(" Got a session update: " + str(data['session']) + "\n")
#sys.stderr.write(" Got a session update\n")
socketio.emit('sessionupdate', {'key': data['key'], 'session': data['session']}, namespace='/monitor', room=sid)
if data['messagetype'] == 'chatready':
pubsub.subscribe(data['sid'])
listening_sids.add(data['sid'])
secrets[data['sid']] = data['secret']
r.hset('da:monitor:chatpartners:' + str(user_id), 'da:interviewsession:uid:' + str(data['uid']) + ':i:' + str(data['i']) + ':userid:' + str(data['userid']), 1)
if str(data['userid']).startswith('t'):
name = word("anonymous visitor") + ' ' + str(data['userid'])[1:]
else:
person = UserModel.query.filter_by(id=data['userid']).first()
if person.first_name:
name = str(person.first_name) + ' ' + str(person.last_name)
else:
name = str(person.email)
sys.stderr.write("chatready 1")
socketio.emit('chatready', {'uid': data['uid'], 'i': data['i'], 'userid': data['userid'], 'name': name}, namespace='/monitor', room=sid)
if data['messagetype'] == 'block':
pubsub.unsubscribe(item['channel'].decode())
if item['channel'].decode() in listening_sids:
listening_sids.remove(item['channel'].decode())
sys.stderr.write(" monitor unsubscribed from " + item['channel'].decode() + "\n")
if data['messagetype'] == 'refreshsessions':
socketio.emit('refreshsessions', {}, namespace='/monitor', room=sid)
if data['messagetype'] == 'chat':
#sys.stderr.write(" Emitting monitor chat message: " + str(data['message']['message']) + "\n")
if str(user_id) == str(data['message'].get('user_id', None)):
data['message']['is_self'] = True
else:
data['message']['is_self'] = False
socketio.emit('chatmessage', {'i': data['yaml_filename'], 'uid': data['uid'], 'userid': data['user_id'], 'data': data['message']}, namespace='/monitor', room=sid)
if data['messagetype'] == 'chatstop':
sys.stderr.write(" Chat termination for sid " + data['sid'] + "\n")
pubsub.unsubscribe(data['sid'])
if data['sid'] in secrets:
del secrets[data['sid']]
r.hdel('da:monitor:chatpartners:' + str(user_id), 'da:interviewsession:uid:' + str(data['uid']) + ':i:' + str(data['i']) + ':userid:' + data['userid'])
socketio.emit('chatstop', {'uid': data['uid'], 'i': data['i'], 'userid': data['userid']}, namespace='/monitor', room=sid)
sys.stderr.write(' exiting monitor thread for sid ' + str(sid) + '\n')
def handle_message(message):
socketio.emit('mymessage', {'data': "Hello"}, namespace='/wsinterview', room=request.sid)
def update_monitor(message):
if 'monitor' not in session:
socketio.emit('terminate', {}, namespace='/monitor', room=request.sid)
return
#sys.stderr.write('received message from ' + str(request.sid) + "\n")
available_for_chat = message['available_for_chat']
new_subscribed_roles = message['subscribed_roles']
new_phone_partners = message['phone_partners_to_add']
term_phone_partners = message['phone_partners_to_terminate']
phone_number = message['phone_number']
phone_number_key = 'da:monitor:phonenumber:' + str(session['user_id'])
if phone_number is None or phone_number == '':
rr.delete(phone_number_key)
else:
pipe = rr.pipeline()
pipe.set(phone_number_key, phone_number)
pipe.expire(phone_number_key, 2592000)
pipe.execute()
phone_partners = dict()
prefix = 'da:phonecode:monitor:' + str(session['user_id']) + ':uid:'
for key in term_phone_partners:
the_code = rr.get(key)
if the_code is not None:
rr.delete(re.sub(r'da:session:uid:', prefix, key))
rr.delete('da:callforward:' + str(the_code))
if phone_number is None or phone_number == '':
for key in rr.keys(prefix + '*'):
the_code = rr.get(key)
if the_code is not None:
rr.delete(key)
rr.delete('da:callforward:' + str(the_code))
else:
codes_in_use = set()
for key in rr.keys('da:callforward:*'):
code = re.sub(r'^da:callforward:', '', key)
codes_in_use.add(code)
for key in rr.keys(prefix + '*'):
phone_partners[re.sub(r'^da:phonecode:monitor:[0-9]*:uid:', 'da:session:uid:', key)] = 1
for key in new_phone_partners:
if key in phone_partners:
continue
times = 0
ok = False
while times < 1000:
times += 1
code = "%04d" % random.randint(1000, 9999)
if code in codes_in_use:
continue
ok = True
the_code = code
new_key = re.sub(r'^da:session:uid:', prefix, key)
code_key = 'da:callforward:' + str(code)
pipe = rr.pipeline()
pipe.set(new_key, code)
pipe.set(code_key, phone_number)
pipe.expire(new_key, 300)
pipe.expire(code_key, 300)
pipe.execute()
phone_partners[key] = 1
break
if times >= 1000:
logmessage("update_monitor: could not get a random integer")
#sys.stderr.write('subscribed roles are type ' + str(type(new_subscribed_roles)) + " which is " + str(new_subscribed_roles) + "\n")
monitor_key = 'da:monitor:' + str(request.sid)
pipe = rr.pipeline()
pipe.set(monitor_key, 1)
pipe.expire(monitor_key, 60)
pipe.execute()
key = 'da:monitor:available:' + str(session['user_id'])
key_exists = rr.exists(key)
chat_partners = dict()
for cp_key in rr.hgetall('da:monitor:chatpartners:' + str(session['user_id'])):
if rr.get(cp_key) is None:
rr.hdel('da:monitor:chatpartners:' + str(session['user_id']), cp_key)
else:
chat_partners[re.sub('^da:interviewsession:uid:', r'da:session:uid:', cp_key)] = 1
#sys.stderr.write('daAvailableForChat is ' + str(available_for_chat) + " for key " + key + "\n")
if available_for_chat:
pipe = rr.pipeline()
pipe.set(key, request.sid)
pipe.expire(key, 60)
pipe.execute()
elif key_exists:
#sys.stderr.write("Deleting shit\n")
pipe = rr.pipeline()
pipe.delete(key)
for avail_key in rr.keys('da:monitor:role:*:userid:' + str(session['user_id'])):
pipe.delete(avail_key)
pipe.execute()
avail_roles = list()
for key in rr.keys('da:chat:roletype:*'):
avail_roles.append(re.sub(r'^da:chat:roletype:', r'', key))
sub_role_key = 'da:monitor:userrole:' + str(session['user_id'])
if rr.exists(sub_role_key):
subscribed_roles = rr.hgetall(sub_role_key)
else:
subscribed_roles = dict()
del_mon_role_keys = list()
for role_key in new_subscribed_roles.keys():
if role_key not in avail_roles:
#sys.stderr.write("role_key is " + str(role_key) + " which is " + str(type(role_key)) + "\n")
del new_subscribed_roles[role_key]
for role_key in subscribed_roles.keys():
if role_key not in avail_roles:
rr.hdel(sub_role_key, role_key)
del_mon_role_keys.append('da:monitor:role:' + role_key + ':userid:' + str(session['user_id']))
for role_key in new_subscribed_roles.keys():
if role_key not in subscribed_roles:
rr.hset(sub_role_key, role_key, 1)
subscribed_roles[role_key] = 1
for role_key in subscribed_roles.keys():
if role_key not in new_subscribed_roles:
rr.hdel(sub_role_key, role_key)
del_mon_role_keys.append('da:monitor:role:' + role_key + ':userid:' + str(session['user_id']))
del subscribed_roles[role_key]
if len(del_mon_role_keys):
pipe = rr.pipeline()
for key in del_mon_role_keys:
pipe.delete(key)
pipe.execute()
if available_for_chat and len(subscribed_roles):
pipe = rr.pipeline()
for role_key in subscribed_roles.keys():
key = 'da:monitor:role:' + role_key + ':userid:' + str(session['user_id'])
pipe.set(key, 1)
pipe.expire(key, 60)
pipe.execute()
keylist = list()
for key in rr.keys('da:session:*'):
keylist.append(key)
sessions = dict()
for key in keylist:
try:
sessobj = pickle.loads(rr.get(key))
except:
sys.stderr.write('error parsing value of ' + str(key) + " which was " + str(rr.get(key)) + "\n")
continue
if sessobj.get('chatstatus', None) != 'off':
html = rr.get(re.sub(r'^da:session:', 'da:html:', key))
if html is not None:
obj = json.loads(html)
sessobj['browser_title'] = obj.get('browser_title', 'not available')
if rr.exists(re.sub(r'^da:session:', 'da:block:', key)):
sessobj['blocked'] = True
else:
sessobj['blocked'] = False
sessions[key] = sessobj
socketio.emit('updatemonitor', {'available_for_chat': available_for_chat, 'subscribedRoles': subscribed_roles, 'sessions': sessions, 'availRoles': sorted(avail_roles), 'chatPartners': chat_partners, 'phonePartners': phone_partners}, namespace='/monitor', room=request.sid)
def terminate_observer_connection():
sys.stderr.write("terminate_observer_connection\n")
# hopefully the disconnect will be triggered
# if request.sid in threads:
# rr.publish(request.sid, json.dumps(dict(origin='client', message='KILL', sid=request.sid)))
socketio.emit('terminate', {}, namespace='/observer', room=request.sid)
def on_observer_connect():
if 'observer' not in session:
socketio.emit('terminate', {}, namespace='/observer', room=request.sid)
return
join_room(request.sid)
