forked from rDrama/rDrama
90 lines
2.1 KiB
Python
90 lines
2.1 KiB
Python
import sys
|
|
from files.helpers.const import SITE, SITE_FULL
|
|
|
|
if "load_chat" in sys.argv or SITE == 'localhost':
|
|
import time
|
|
from files.helpers.wrappers import auth_required
|
|
from files.helpers.sanitize import sanitize
|
|
from datetime import datetime
|
|
from flask_socketio import SocketIO, emit
|
|
from files.__main__ import app, limiter, cache
|
|
from flask import render_template, make_response, send_from_directory, abort
|
|
import sys
|
|
import atexit
|
|
|
|
socketio = SocketIO(app, async_mode='gevent', cors_allowed_origins=[SITE_FULL])
|
|
typing = []
|
|
online = []
|
|
messages = cache.get('chat') or []
|
|
|
|
|
|
@app.get("/chat")
|
|
@auth_required
|
|
def chat( v):
|
|
return render_template("chat.html", v=v, messages=messages)
|
|
|
|
|
|
@app.get('/static/chat.js')
|
|
@limiter.exempt
|
|
def chatjs():
|
|
resp = make_response(send_from_directory('assets', 'js/chat.js'))
|
|
return resp
|
|
|
|
|
|
@socketio.on('speak')
|
|
@limiter.limit("3/second;10/minute")
|
|
@auth_required
|
|
def speak(data, v):
|
|
if v.is_banned: abort(403)
|
|
global messages
|
|
text = data[:1000].strip()
|
|
if not text: abort(403)
|
|
|
|
data={
|
|
"avatar": v.profile_url,
|
|
"username":v.username,
|
|
"namecolor":v.namecolor,
|
|
"text":text,
|
|
"text_html":sanitize(text),
|
|
}
|
|
|
|
messages.append(data)
|
|
messages = messages[-500:]
|
|
emit('speak', data, broadcast=True)
|
|
return '', 204
|
|
|
|
@socketio.on('connect')
|
|
@auth_required
|
|
def connect(v):
|
|
if v.username not in online:
|
|
online.append(v.username)
|
|
emit("online", online, broadcast=True)
|
|
|
|
emit('typing', typing)
|
|
return '', 204
|
|
|
|
@socketio.on('disconnect')
|
|
@auth_required
|
|
def disconnect(v):
|
|
if v.username in online:
|
|
online.remove(v.username)
|
|
emit("online", online, broadcast=True)
|
|
|
|
if v.username in typing: typing.remove(v.username)
|
|
emit('typing', typing, broadcast=True)
|
|
return '', 204
|
|
|
|
@socketio.on('typing')
|
|
@auth_required
|
|
def typing_indicator(data, v):
|
|
|
|
if data and v.username not in typing: typing.append(v.username)
|
|
elif not data and v.username in typing: typing.remove(v.username)
|
|
|
|
emit('typing', typing, broadcast=True)
|
|
return '', 204
|
|
|
|
|
|
def close_running_threads():
|
|
cache.set('chat', messages)
|
|
atexit.register(close_running_threads) |