alttpr-gwaa-kiwi/alttpr-python
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Add function to process commands to MultiServer.py

Browse Source
This commit is contained in:
karafruit
2023-03-02 18:35:34 -06:00
parent e53a9a3e2a
commit c191a224c0
1 changed files with 48 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

91
MultiServer.py
View File

@@ -46,6 +46,9 @@ class Context:
self.lookup_name_to_id = {} self.lookup_name_to_id = {}
self.lookup_id_to_name = {} self.lookup_id_to_name = {}
async def process_command(self, input):
await process_command(ctx, input)
async def send_msgs(websocket, msgs): async def send_msgs(websocket, msgs):
if not websocket or not websocket.open or websocket.closed: if not websocket or not websocket.open or websocket.closed:
return return
@@ -296,54 +299,56 @@ def set_password(ctx : Context, password):
async def console(ctx : Context): async def console(ctx : Context):
while True: while True:
input = await aioconsole.ainput() input = await aioconsole.ainput()
await process_command(ctx, input)
command = shlex.split(input) async def process_command(ctx : Context, input):
if not command: command = shlex.split(input)
continue if not command:
return
if command[0] == '/exit': if command[0] == '/exit':
ctx.server.ws_server.close() ctx.server.ws_server.close()
break return
if command[0] == '/players': if command[0] == '/players':
logging.info(get_connected_players_string(ctx)) logging.info(get_connected_players_string(ctx))
if command[0] == '/password': if command[0] == '/password':
set_password(ctx, command[1] if len(command) > 1 else None) set_password(ctx, command[1] if len(command) > 1 else None)
if command[0] == '/kick' and len(command) > 1: if command[0] == '/kick' and len(command) > 1:
team = int(command[2]) - 1 if len(command) > 2 and command[2].isdigit() else None team = int(command[2]) - 1 if len(command) > 2 and command[2].isdigit() else None
for client in ctx.clients:
if client.auth and client.name.lower() == command[1].lower() and (team is None or team == client.team):
if client.socket and not client.socket.closed:
await client.socket.close()
if command[0] == '/forfeitslot' and len(command) > 1 and command[1].isdigit():
if len(command) > 2 and command[2].isdigit():
team = int(command[1]) - 1
slot = int(command[2])
else:
team = 0
slot = int(command[1])
forfeit_player(ctx, team, slot)
if command[0] == '/forfeitplayer' and len(command) > 1:
team = int(command[2]) - 1 if len(command) > 2 and command[2].isdigit() else None
for client in ctx.clients:
if client.auth and client.name.lower() == command[1].lower() and (team is None or team == client.team):
if client.socket and not client.socket.closed:
forfeit_player(ctx, client.team, client.slot)
if command[0] == '/senditem' and len(command) > 2:
[(player, item)] = re.findall(r'\S* (\S*) (.*)', input)
if item in Items.item_table:
for client in ctx.clients: for client in ctx.clients:
if client.auth and client.name.lower() == command[1].lower() and (team is None or team == client.team): if client.auth and client.name.lower() == player.lower():
if client.socket and not client.socket.closed: new_item = ReceivedItem(Items.item_table[item][3], "cheat console", client.slot)
await client.socket.close() get_received_items(ctx, client.team, client.slot).append(new_item)
notify_all(ctx, 'Cheat console: sending "' + item + '" to ' + client.name)
send_new_items(ctx)
else:
logging.warning("Unknown item: " + item)
if command[0] == '/forfeitslot' and len(command) > 1 and command[1].isdigit(): if command[0][0] != '/':
if len(command) > 2 and command[2].isdigit(): notify_all(ctx, '[Server]: ' + input)
team = int(command[1]) - 1
slot = int(command[2])
else:
team = 0
slot = int(command[1])
forfeit_player(ctx, team, slot)
if command[0] == '/forfeitplayer' and len(command) > 1:
team = int(command[2]) - 1 if len(command) > 2 and command[2].isdigit() else None
for client in ctx.clients:
if client.auth and client.name.lower() == command[1].lower() and (team is None or team == client.team):
if client.socket and not client.socket.closed:
forfeit_player(ctx, client.team, client.slot)
if command[0] == '/senditem' and len(command) > 2:
[(player, item)] = re.findall(r'\S* (\S*) (.*)', input)
if item in Items.item_table:
for client in ctx.clients:
if client.auth and client.name.lower() == player.lower():
new_item = ReceivedItem(Items.item_table[item][3], "cheat console", client.slot)
get_received_items(ctx, client.team, client.slot).append(new_item)
notify_all(ctx, 'Cheat console: sending "' + item + '" to ' + client.name)
send_new_items(ctx)
else:
logging.warning("Unknown item: " + item)
if command[0][0] != '/':
notify_all(ctx, '[Server]: ' + input)
def init_lookups(ctx): def init_lookups(ctx):