154 lines
3.1 KiB
Python
154 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
SocketIO events.
|
|
"""
|
|
import re
|
|
import time
|
|
import random
|
|
|
|
import bleach
|
|
from flask import session
|
|
from flask_socketio import SocketIO, emit, join_room
|
|
|
|
import tools
|
|
import database as db
|
|
|
|
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)
|
|
if message.startswith("/dice") or message.startswith("/roll"):
|
|
roll_msg = handle_dice(data)
|
|
message += '<hr class="msgSrvHr" />' + roll_msg
|
|
|
|
data["message"] = message
|
|
|
|
db.log_chat_message(data)
|
|
emit("message", data, room=room)
|
|
|
|
|
|
|
|
|
|
def handle_dice(message):
|
|
"""
|
|
Handle /dice or /roll messages.
|
|
"""
|
|
room = data["room"]
|
|
reg = re.search(r"(\d+)d(\d+)([+-]\d+)?", message)
|
|
if not reg:
|
|
return
|
|
try:
|
|
diceNum, diceSides, diceMod = map(int, reg.groups())
|
|
assert 0 < diceNum < 100
|
|
assert 0 < diceSides < 100
|
|
assert -1000 < diceMod < 1000
|
|
except (ValueError, AssertionError):
|
|
return
|
|
dice = [random.randint(1, diceSides) for _ in range(diceNum)]
|
|
total = sum(dice)
|
|
if diceMod:
|
|
total += diceMod
|
|
msg = f"<b>Rolled {', '.join(map(str, dice))}"
|
|
if diceMod:
|
|
if diceMod > 0:
|
|
msg += " +"
|
|
msg += " " + str(diceMod)
|
|
msg += " = " + str(total) + "</b>"
|
|
return msg
|
|
|
|
|
|
@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]
|
|
data["date"] = int(time.time())
|
|
post_id = db.insert_quest_post(room, post, data["date"])
|
|
data["post_id"] = post_id
|
|
|
|
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)
|
|
|
|
|
|
@socketio.on("dice_post")
|
|
def dice_(data):
|
|
"""
|
|
Called when the QM posts a new dice call.
|
|
"""
|
|
print(data)
|
|
room = data["room"]
|
|
res = db.get_quest_meta(quest_id=room)
|
|
if not res:
|
|
return
|
|
if session.get("user_id") != res[3]:
|
|
return
|
|
|
|
date = int(time.time())
|
|
data["date"] = date
|
|
data["post_id"] = 0
|
|
|
|
emit("dice_post", data, room=room)
|