96 lines
1.9 KiB
Python
96 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
SocketIO events.
|
|
"""
|
|
import time
|
|
|
|
import bleach
|
|
from flask import session
|
|
from flask_socketio import SocketIO, emit, join_room
|
|
|
|
import database as db
|
|
import tools
|
|
|
|
socketio = SocketIO()
|
|
|
|
@socketio.on('joined')
|
|
def joined(data):
|
|
"""
|
|
Sent by clients when they enter a room.
|
|
"""
|
|
room = data["room"]
|
|
join_room(room)
|
|
|
|
|
|
@socketio.on('message')
|
|
def message(data):
|
|
"""
|
|
Sent by a client when the user entered a new message.
|
|
"""
|
|
room = data["room"]
|
|
|
|
message = data["message"]
|
|
date = int(time.time())
|
|
data["date"] = date
|
|
|
|
message = message.strip()
|
|
if not message:
|
|
return
|
|
tags = ["b", "code", "i", "s"]
|
|
message = bleach.clean(message, tags=tags)
|
|
lines = []
|
|
for line in message.splitlines():
|
|
if line.startswith(">"):
|
|
line = '<span class="greenText">' + line + '</span>'
|
|
lines.append(line)
|
|
message = "<br />".join(lines)
|
|
message = tools.handle_img(message)
|
|
data["message"] = message
|
|
|
|
db.log_chat_message(data)
|
|
emit("message", data, room=room)
|
|
|
|
|
|
@socketio.on("new_post")
|
|
def new_post(data):
|
|
"""
|
|
Called when the QM makes a new post.
|
|
"""
|
|
room = data["room"]
|
|
res = db.get_quest_meta(quest_id=room)
|
|
if not res:
|
|
return
|
|
if session.get("user_id") != res[3]:
|
|
return
|
|
|
|
post = data["post"]
|
|
post = bleach.clean(post.strip())
|
|
post = post.replace("\n", "<br />")
|
|
post = tools.handle_img(post)
|
|
data["post"] = [post]
|
|
db.insert_quest_post(room, post, int(time.time()))
|
|
|
|
emit("new_post", data, room=room)
|
|
|
|
|
|
@socketio.on("update_post")
|
|
def update_post(data):
|
|
"""
|
|
Called when the QM edits and saves a post.
|
|
"""
|
|
room = data["room"]
|
|
res = db.get_quest_meta(quest_id=room)
|
|
if not res:
|
|
return
|
|
if session.get("user_id") != res[3]:
|
|
return
|
|
|
|
post = data["post"]
|
|
post = post.strip().replace("<br>", "<br />")
|
|
post = tools.handle_img(post)
|
|
data["post"] = post
|
|
|
|
post_id = data["post_id"]
|
|
db.update_quest_post(post_id, post)
|
|
emit("update_post", data, room=room)
|