MarseyWorld/files/routes/chat.py

import sys
if "load_chat" in sys.argv:
	from files.helpers.const import SITE_FULL
	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, r
	from flask import render_template
	import sys
	import atexit

	socketio = SocketIO(app, async_mode='gevent', cors_allowed_origins=[SITE_FULL])
	typing = []
	online = []
	messages = r.get('chat')

	@app.get("/chat")
	@auth_required
	def chat( v):
		return render_template("chat.html", v=v, messages=messages)

	@socketio.on('speak')
	@limiter.limit("5/second;30/minute")
	@auth_required
	def speak(data, v):
		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[:20]
		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():
		r.set('chat', messages)
	atexit.register(close_running_threads)