skorotkiewicz/rDrama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add function to nuke the chat state.

Browse source
This commit is contained in:
Ben Rog-Wilhelm 2023-08-27 11:41:05 -05:00
parent 934b6e2b68
commit 7faba478b9
1 changed files with 23 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

34
files/routes/chat.py
View file

@ -3,7 +3,7 @@ import time
import uuid import uuid
from typing import Any, Final from typing import Any, Final
from flask_socketio import SocketIO, emit from flask_socketio import SocketIO, emit, disconnect
from files.__main__ import app, cache, limiter from files.__main__ import app, cache, limiter
from files.helpers.alerts import * from files.helpers.alerts import *
@ -64,8 +64,7 @@ CHAT_SCROLLBACK_ITEMS: Final[int] = 500
typing: list[str] = [] typing: list[str] = []
online: list[str] = [] # right now we maintain this but don't actually use it anywhere online: list[str] = [] # right now we maintain this but don't actually use it anywhere
socket_ids_to_user_ids = {} connected_users = set()
user_ids_to_socket_ids = {}
def send_system_reply(text): def send_system_reply(text):
data = { data = {
@ -151,13 +150,11 @@ def speak(data, v):
@socketio.on('connect') @socketio.on('connect')
@chat_is_allowed() @chat_is_allowed()
def connect(v): def onConnect(v):
if v.username not in online: if v.username not in online:
online.append(v.username) online.append(v.username)
if not socket_ids_to_user_ids.get(request.sid): connected_users.add(request.sid)
socket_ids_to_user_ids[request.sid] = v.id
user_ids_to_socket_ids[v.id] = request.sid
emit('online', get_chat_userlist()) emit('online', get_chat_userlist())
emit('catchup', get_chat_messages()) emit('catchup', get_chat_messages())
@ -166,15 +163,13 @@ def connect(v):
@socketio.on('disconnect') @socketio.on('disconnect')
@chat_is_allowed() @chat_is_allowed()
def disconnect(v): def onDisconnect(v):
if v.username in online: if v.username in online:
online.remove(v.username) online.remove(v.username)
if v.username in typing: typing.remove(v.username) if v.username in typing: typing.remove(v.username)
if socket_ids_to_user_ids.get(request.sid): connected_users.remove(request.sid)
del socket_ids_to_user_ids[request.sid]
del user_ids_to_socket_ids[v.id]
emit('typing', typing, broadcast=True) emit('typing', typing, broadcast=True)
@ -257,3 +252,20 @@ def remove(user):
send_system_reply(f"Removed {user} from chat.") send_system_reply(f"Removed {user} from chat.")
else: else:
send_system_reply(f"Could not find user {user}.") send_system_reply(f"Could not find user {user}.")
@register_command('reset_everything_seriously', PERMS['CHAT_FULL_CONTROL'])
def reset_everything_seriously(_):
# Boot everyone
for user_sid in list(connected_users): # Loop through a shallow copy to avoid modification issues
disconnect(sid=user_sid)
# Set chat_authorized to False for all users
g.db.query(User).update({User.chat_authorized: False})
# Delete all ChatMessage entries
g.db.query(ChatMessage).delete()
# Commit the changes to the database
g.db.commit()