merge HintedMultiServer into MultiServer
This commit is contained in:
		
							parent
							
								
									5d4707dacd
								
							
						
					
					
						commit
						5d217511fe
					
				| 
						 | 
					@ -1,430 +0,0 @@
 | 
				
			||||||
import aioconsole
 | 
					 | 
				
			||||||
import argparse
 | 
					 | 
				
			||||||
import asyncio
 | 
					 | 
				
			||||||
import functools
 | 
					 | 
				
			||||||
import json
 | 
					 | 
				
			||||||
import logging
 | 
					 | 
				
			||||||
import re
 | 
					 | 
				
			||||||
import urllib.request
 | 
					 | 
				
			||||||
import websockets
 | 
					 | 
				
			||||||
import zlib
 | 
					 | 
				
			||||||
import datetime
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
import Items
 | 
					 | 
				
			||||||
import Regions
 | 
					 | 
				
			||||||
from MultiClient import ReceivedItem, get_item_name_from_id, get_location_name_from_address
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
class Client:
 | 
					 | 
				
			||||||
    def __init__(self, socket):
 | 
					 | 
				
			||||||
        self.socket = socket
 | 
					 | 
				
			||||||
        self.auth = False
 | 
					 | 
				
			||||||
        self.name = None
 | 
					 | 
				
			||||||
        self.team = None
 | 
					 | 
				
			||||||
        self.slot = None
 | 
					 | 
				
			||||||
        self.send_index = 0
 | 
					 | 
				
			||||||
        self.hints_used = 0
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
class Context:
 | 
					 | 
				
			||||||
    def __init__(self, host, port, password):
 | 
					 | 
				
			||||||
        self.data_filename = None
 | 
					 | 
				
			||||||
        self.save_filename = None
 | 
					 | 
				
			||||||
        self.disable_save = False
 | 
					 | 
				
			||||||
        self.players = 0
 | 
					 | 
				
			||||||
        self.rom_names = {}
 | 
					 | 
				
			||||||
        self.locations = {}
 | 
					 | 
				
			||||||
        self.host = host
 | 
					 | 
				
			||||||
        self.port = port
 | 
					 | 
				
			||||||
        self.password = password
 | 
					 | 
				
			||||||
        self.server = None
 | 
					 | 
				
			||||||
        self.clients = []
 | 
					 | 
				
			||||||
        self.received_items = {}
 | 
					 | 
				
			||||||
        self.starttime = datetime.datetime.now()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def get_room_info(ctx : Context):
 | 
					 | 
				
			||||||
    return {
 | 
					 | 
				
			||||||
        'password': ctx.password is not None,
 | 
					 | 
				
			||||||
        'slots': ctx.players,
 | 
					 | 
				
			||||||
        'players': [(client.name, client.team, client.slot) for client in ctx.clients if client.auth]
 | 
					 | 
				
			||||||
    }
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def same_name(lhs, rhs):
 | 
					 | 
				
			||||||
    return lhs.lower() == rhs.lower()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def same_team(lhs, rhs):
 | 
					 | 
				
			||||||
    return (type(lhs) is type(rhs)) and ((not lhs and not rhs) or (lhs.lower() == rhs.lower()))
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
async def send_msgs(websocket, msgs):
 | 
					 | 
				
			||||||
    if not websocket or not websocket.open or websocket.closed:
 | 
					 | 
				
			||||||
        return
 | 
					 | 
				
			||||||
    try:
 | 
					 | 
				
			||||||
        await websocket.send(json.dumps(msgs))
 | 
					 | 
				
			||||||
    except websockets.ConnectionClosed:
 | 
					 | 
				
			||||||
        pass
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def broadcast_all(ctx : Context, msgs):
 | 
					 | 
				
			||||||
    for client in ctx.clients:
 | 
					 | 
				
			||||||
        if client.auth:
 | 
					 | 
				
			||||||
            asyncio.create_task(send_msgs(client.socket, msgs))
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def broadcast_team(ctx : Context, team, msgs):
 | 
					 | 
				
			||||||
    for client in ctx.clients:
 | 
					 | 
				
			||||||
        if client.auth and same_team(client.team, team):
 | 
					 | 
				
			||||||
            asyncio.create_task(send_msgs(client.socket, msgs))
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def notify_all(ctx : Context, text):
 | 
					 | 
				
			||||||
    print("Notice (all): %s" % text)
 | 
					 | 
				
			||||||
    broadcast_all(ctx, [['Print', text]])
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def notify_team(ctx : Context, team : str, text : str):
 | 
					 | 
				
			||||||
    print("Team notice (%s): %s" % ("Default" if not team else team, text))
 | 
					 | 
				
			||||||
    broadcast_team(ctx, team, [['Print', text]])
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def notify_client(client : Client, text : str):
 | 
					 | 
				
			||||||
    if not client.auth:
 | 
					 | 
				
			||||||
        return
 | 
					 | 
				
			||||||
    print("Player notice (%s): %s" % (client.name, text))
 | 
					 | 
				
			||||||
    asyncio.create_task(send_msgs(client.socket,  [['Print', text]]))
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
async def server(websocket, path, ctx : Context):
 | 
					 | 
				
			||||||
    client = Client(websocket)
 | 
					 | 
				
			||||||
    ctx.clients.append(client)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    try:
 | 
					 | 
				
			||||||
        await on_client_connected(ctx, client)
 | 
					 | 
				
			||||||
        async for data in websocket:
 | 
					 | 
				
			||||||
            for msg in json.loads(data):
 | 
					 | 
				
			||||||
                if len(msg) == 1:
 | 
					 | 
				
			||||||
                    cmd = msg
 | 
					 | 
				
			||||||
                    args = None
 | 
					 | 
				
			||||||
                else:
 | 
					 | 
				
			||||||
                    cmd = msg[0]
 | 
					 | 
				
			||||||
                    args = msg[1]
 | 
					 | 
				
			||||||
                await process_client_cmd(ctx, client, cmd, args)
 | 
					 | 
				
			||||||
    except Exception as e:
 | 
					 | 
				
			||||||
        if not isinstance(e, websockets.WebSocketException):
 | 
					 | 
				
			||||||
            logging.exception(e)
 | 
					 | 
				
			||||||
    finally:
 | 
					 | 
				
			||||||
        await on_client_disconnected(ctx, client)
 | 
					 | 
				
			||||||
        ctx.clients.remove(client)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
async def on_client_connected(ctx : Context, client : Client):
 | 
					 | 
				
			||||||
    await send_msgs(client.socket, [['RoomInfo', get_room_info(ctx)]])
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
async def on_client_disconnected(ctx : Context, client : Client):
 | 
					 | 
				
			||||||
    if client.auth:
 | 
					 | 
				
			||||||
        await on_client_left(ctx, client)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
async def on_client_joined(ctx : Context, client : Client):
 | 
					 | 
				
			||||||
    notify_all(ctx, "%s has joined the game as player %d for %s" % (client.name, client.slot, "the default team" if not client.team else "team %s" % client.team))
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
async def on_client_left(ctx : Context, client : Client):
 | 
					 | 
				
			||||||
    notify_all(ctx, "%s (Player %d, %s) has left the game" % (client.name, client.slot, "Default team" if not client.team else "Team %s" % client.team))
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def get_connected_players_string(ctx : Context):
 | 
					 | 
				
			||||||
    auth_clients = [c for c in ctx.clients if c.auth]
 | 
					 | 
				
			||||||
    if not auth_clients:
 | 
					 | 
				
			||||||
        return 'No player connected'
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    auth_clients.sort(key=lambda c: ('' if not c.team else c.team.lower(), c.slot))
 | 
					 | 
				
			||||||
    current_team = 0
 | 
					 | 
				
			||||||
    text = ''
 | 
					 | 
				
			||||||
    for c in auth_clients:
 | 
					 | 
				
			||||||
        if c.team != current_team:
 | 
					 | 
				
			||||||
            text += '::' + ('default team' if not c.team else c.team) + ':: '
 | 
					 | 
				
			||||||
            current_team = c.team
 | 
					 | 
				
			||||||
        text += '%d:%s ' % (c.slot, c.name)
 | 
					 | 
				
			||||||
    return 'Connected players: ' + text[:-1]
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def get_player_name_in_team(ctx : Context, team, slot):
 | 
					 | 
				
			||||||
    for client in ctx.clients:
 | 
					 | 
				
			||||||
        if client.auth and same_team(team, client.team) and client.slot == slot:
 | 
					 | 
				
			||||||
            return client.name
 | 
					 | 
				
			||||||
    return "Player %d" % slot
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def get_client_from_name(ctx : Context, name):
 | 
					 | 
				
			||||||
    for client in ctx.clients:
 | 
					 | 
				
			||||||
        if client.auth and same_name(name, client.name):
 | 
					 | 
				
			||||||
            return client
 | 
					 | 
				
			||||||
    return None
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def get_received_items(ctx : Context, team, player):
 | 
					 | 
				
			||||||
    for (c_team, c_id), items in ctx.received_items.items():
 | 
					 | 
				
			||||||
        if c_id == player and same_team(c_team, team):
 | 
					 | 
				
			||||||
            return items
 | 
					 | 
				
			||||||
    ctx.received_items[(team, player)] = []
 | 
					 | 
				
			||||||
    return ctx.received_items[(team, player)]
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def tuplize_received_items(items):
 | 
					 | 
				
			||||||
    return [(item.item, item.location, item.player_id, item.player_name) for item in items]
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def send_new_items(ctx : Context):
 | 
					 | 
				
			||||||
    for client in ctx.clients:
 | 
					 | 
				
			||||||
        if not client.auth:
 | 
					 | 
				
			||||||
            continue
 | 
					 | 
				
			||||||
        items = get_received_items(ctx, client.team, client.slot)
 | 
					 | 
				
			||||||
        if len(items) > client.send_index:
 | 
					 | 
				
			||||||
            asyncio.create_task(send_msgs(client.socket, [['ReceivedItems', (client.send_index, tuplize_received_items(items)[client.send_index:])]]))
 | 
					 | 
				
			||||||
            client.send_index = len(items)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def forfeit_player(ctx : Context, team, slot, name):
 | 
					 | 
				
			||||||
    all_locations = [values[0] for values in Regions.location_table.values() if type(values[0]) is int]
 | 
					 | 
				
			||||||
    notify_all(ctx, "%s (Player %d) in team %s has forfeited" % (name, slot, team if team else 'default'))
 | 
					 | 
				
			||||||
    register_location_checks(ctx, name, team, slot, all_locations)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
def register_location_checks(ctx : Context, name, team, slot, locations):
 | 
					 | 
				
			||||||
    found_items = False
 | 
					 | 
				
			||||||
    for location in locations:
 | 
					 | 
				
			||||||
        if (location, slot) in ctx.locations:
 | 
					 | 
				
			||||||
            target_item, target_player = ctx.locations[(location, slot)]
 | 
					 | 
				
			||||||
            if target_player != slot:
 | 
					 | 
				
			||||||
                found = False
 | 
					 | 
				
			||||||
                recvd_items = get_received_items(ctx, team, target_player)
 | 
					 | 
				
			||||||
                for recvd_item in recvd_items:
 | 
					 | 
				
			||||||
                    if recvd_item.location == location and recvd_item.player_id == slot:
 | 
					 | 
				
			||||||
                        found = True
 | 
					 | 
				
			||||||
                        break
 | 
					 | 
				
			||||||
                if not found:
 | 
					 | 
				
			||||||
                    new_item = ReceivedItem(target_item, location, slot, name)
 | 
					 | 
				
			||||||
                    recvd_items.append(new_item)
 | 
					 | 
				
			||||||
                    target_player_name = get_player_name_in_team(ctx, team, target_player)
 | 
					 | 
				
			||||||
                    broadcast_team(ctx, team, [['ItemSent', (name, target_player_name, target_item, location)]])
 | 
					 | 
				
			||||||
                    print('(%s) %s sent %s to %s (%s)' % (team if team else 'Team', name, get_item_name_from_id(target_item), target_player_name, get_location_name_from_address(location)))
 | 
					 | 
				
			||||||
                    found_items = True
 | 
					 | 
				
			||||||
    send_new_items(ctx)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    if found_items and not ctx.disable_save:
 | 
					 | 
				
			||||||
        try:
 | 
					 | 
				
			||||||
            with open(ctx.save_filename, "wb") as f:
 | 
					 | 
				
			||||||
                jsonstr = json.dumps((ctx.players,
 | 
					 | 
				
			||||||
                                      [(k, v) for k, v in ctx.rom_names.items()],
 | 
					 | 
				
			||||||
                                      [(k, [i.__dict__ for i in v]) for k, v in ctx.received_items.items()]))
 | 
					 | 
				
			||||||
                f.write(zlib.compress(jsonstr.encode("utf-8")))
 | 
					 | 
				
			||||||
        except Exception as e:
 | 
					 | 
				
			||||||
            logging.exception(e)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
async def process_client_cmd(ctx : Context, client : Client, cmd, args):
 | 
					 | 
				
			||||||
    if type(cmd) is not str:
 | 
					 | 
				
			||||||
        await send_msgs(client.socket, [['InvalidCmd']])
 | 
					 | 
				
			||||||
        return
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    if cmd == 'Connect':
 | 
					 | 
				
			||||||
        if not args or type(args) is not dict or \
 | 
					 | 
				
			||||||
                'password' not in args or type(args['password']) not in [str, type(None)] or \
 | 
					 | 
				
			||||||
                'name' not in args or type(args['name']) is not str or \
 | 
					 | 
				
			||||||
                'team' not in args or type(args['team']) not in [str, type(None)] or \
 | 
					 | 
				
			||||||
                'slot' not in args or type(args['slot']) not in [int, type(None)]:
 | 
					 | 
				
			||||||
            await send_msgs(client.socket, [['InvalidArguments', 'Connect']])
 | 
					 | 
				
			||||||
            return
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        errors = set()
 | 
					 | 
				
			||||||
        if ctx.password is not None and ('password' not in args or args['password'] != ctx.password):
 | 
					 | 
				
			||||||
            errors.add('InvalidPassword')
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if 'name' not in args or not args['name'] or not re.match(r'\w{1,10}', args['name']):
 | 
					 | 
				
			||||||
            errors.add('InvalidName')
 | 
					 | 
				
			||||||
        elif any([same_name(c.name, args['name']) for c in ctx.clients if c.auth]):
 | 
					 | 
				
			||||||
            errors.add('NameAlreadyTaken')
 | 
					 | 
				
			||||||
        else:
 | 
					 | 
				
			||||||
            client.name = args['name']
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if 'team' in args and args['team'] is not None and not re.match(r'\w{1,15}', args['team']):
 | 
					 | 
				
			||||||
            errors.add('InvalidTeam')
 | 
					 | 
				
			||||||
        else:
 | 
					 | 
				
			||||||
            client.team = args['team'] if 'team' in args else None
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if 'slot' in args and any([c.slot == args['slot'] for c in ctx.clients if c.auth and same_team(c.team, client.team)]):
 | 
					 | 
				
			||||||
            errors.add('SlotAlreadyTaken')
 | 
					 | 
				
			||||||
        elif 'slot' not in args or not args['slot']:
 | 
					 | 
				
			||||||
            for slot in range(1, ctx.players + 1):
 | 
					 | 
				
			||||||
                if slot not in [c.slot for c in ctx.clients if c.auth and same_team(c.team, client.team)]:
 | 
					 | 
				
			||||||
                    client.slot = slot
 | 
					 | 
				
			||||||
                    break
 | 
					 | 
				
			||||||
                elif slot == ctx.players:
 | 
					 | 
				
			||||||
                    errors.add('SlotAlreadyTaken')
 | 
					 | 
				
			||||||
        elif args['slot'] not in range(1, ctx.players + 1):
 | 
					 | 
				
			||||||
            errors.add('InvalidSlot')
 | 
					 | 
				
			||||||
        else:
 | 
					 | 
				
			||||||
            client.slot = args['slot']
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if errors:
 | 
					 | 
				
			||||||
            client.name = None
 | 
					 | 
				
			||||||
            client.team = None
 | 
					 | 
				
			||||||
            client.slot = None
 | 
					 | 
				
			||||||
            await send_msgs(client.socket, [['ConnectionRefused', list(errors)]])
 | 
					 | 
				
			||||||
        else:
 | 
					 | 
				
			||||||
            client.auth = True
 | 
					 | 
				
			||||||
            reply = [['Connected', ctx.rom_names[client.slot]]]
 | 
					 | 
				
			||||||
            items = get_received_items(ctx, client.team, client.slot)
 | 
					 | 
				
			||||||
            if items:
 | 
					 | 
				
			||||||
                reply.append(['ReceivedItems', (0, tuplize_received_items(items))])
 | 
					 | 
				
			||||||
                client.send_index = len(items)
 | 
					 | 
				
			||||||
            await send_msgs(client.socket, reply)
 | 
					 | 
				
			||||||
            await on_client_joined(ctx, client)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    if not client.auth:
 | 
					 | 
				
			||||||
        return
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    if cmd == 'Sync':
 | 
					 | 
				
			||||||
        items = get_received_items(ctx, client.team, client.slot)
 | 
					 | 
				
			||||||
        if items:
 | 
					 | 
				
			||||||
            client.send_index = len(items)
 | 
					 | 
				
			||||||
            await send_msgs(client.socket, ['ReceivedItems', (0, tuplize_received_items(items))])
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    if cmd == 'LocationChecks':
 | 
					 | 
				
			||||||
        if type(args) is not list:
 | 
					 | 
				
			||||||
            await send_msgs(client.socket, [['InvalidArguments', 'LocationChecks']])
 | 
					 | 
				
			||||||
            return
 | 
					 | 
				
			||||||
        register_location_checks(ctx, client.name, client.team, client.slot, args)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    if cmd == 'Say':
 | 
					 | 
				
			||||||
        if type(args) is not str or not args.isprintable():
 | 
					 | 
				
			||||||
            await send_msgs(client.socket, [['InvalidArguments', 'Say']])
 | 
					 | 
				
			||||||
            return
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        notify_all(ctx, client.name + ': ' + args)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if args[:8] == '!players':
 | 
					 | 
				
			||||||
            notify_all(ctx, get_connected_players_string(ctx))
 | 
					 | 
				
			||||||
        elif args[:8] == '!forfeit':
 | 
					 | 
				
			||||||
            forfeit_player(ctx, client.team, client.slot, client.name)
 | 
					 | 
				
			||||||
        elif args.startswith("!hint"):
 | 
					 | 
				
			||||||
            pass
 | 
					 | 
				
			||||||
def set_password(ctx : Context, password):
 | 
					 | 
				
			||||||
    ctx.password = password
 | 
					 | 
				
			||||||
    print('Password set to ' + password if password is not None else 'Password disabled')
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
async def console(ctx : Context):
 | 
					 | 
				
			||||||
    while True:
 | 
					 | 
				
			||||||
        input = await aioconsole.ainput()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        command = input.split()
 | 
					 | 
				
			||||||
        if not command:
 | 
					 | 
				
			||||||
            continue
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        if command[0] == '/exit':
 | 
					 | 
				
			||||||
            ctx.server.ws_server.close()
 | 
					 | 
				
			||||||
            break
 | 
					 | 
				
			||||||
        try:
 | 
					 | 
				
			||||||
            if command[0] == '/players':
 | 
					 | 
				
			||||||
                print(get_connected_players_string(ctx))
 | 
					 | 
				
			||||||
            elif command[0] == '/password':
 | 
					 | 
				
			||||||
                set_password(ctx, command[1] if len(command) > 1 else None)
 | 
					 | 
				
			||||||
            elif command[0] == '/kick' and len(command) > 1:
 | 
					 | 
				
			||||||
                client = get_client_from_name(ctx, command[1])
 | 
					 | 
				
			||||||
                if client and client.socket and not client.socket.closed:
 | 
					 | 
				
			||||||
                    await client.socket.close()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
            elif command[0] == '/forfeitslot' and len(command) == 3 and command[2].isdigit():
 | 
					 | 
				
			||||||
                team = command[1] if command[1] != 'default' else None
 | 
					 | 
				
			||||||
                slot = int(command[2])
 | 
					 | 
				
			||||||
                name = get_player_name_in_team(ctx, team, slot)
 | 
					 | 
				
			||||||
                forfeit_player(ctx, team, slot, name)
 | 
					 | 
				
			||||||
            elif command[0] == '/forfeitplayer' and len(command) > 1:
 | 
					 | 
				
			||||||
                client = get_client_from_name(ctx, command[1])
 | 
					 | 
				
			||||||
                if client:
 | 
					 | 
				
			||||||
                    forfeit_player(ctx, client.team, client.slot, client.name)
 | 
					 | 
				
			||||||
            elif command[0] == '/senditem' and len(command) > 2:
 | 
					 | 
				
			||||||
                [(player, item)] = re.findall(r'\S* (\S*) (.*)', input)
 | 
					 | 
				
			||||||
                if item in Items.item_table:
 | 
					 | 
				
			||||||
                    client = get_client_from_name(ctx, player)
 | 
					 | 
				
			||||||
                    if client:
 | 
					 | 
				
			||||||
                        new_item = ReceivedItem(Items.item_table[item][3], "cheat console", 0, "server")
 | 
					 | 
				
			||||||
                        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:
 | 
					 | 
				
			||||||
                    print("Unknown item: " + item)
 | 
					 | 
				
			||||||
            elif command[0] == '/hint':
 | 
					 | 
				
			||||||
                if len(command) > 2:
 | 
					 | 
				
			||||||
                    player_number = 0
 | 
					 | 
				
			||||||
                    if command[1].isnumeric():
 | 
					 | 
				
			||||||
                        player_number = int(command[1])
 | 
					 | 
				
			||||||
                    else:
 | 
					 | 
				
			||||||
                        client = get_client_from_name(ctx, command[1])
 | 
					 | 
				
			||||||
                        if client:
 | 
					 | 
				
			||||||
                            player_number = client.slot
 | 
					 | 
				
			||||||
                        else:
 | 
					 | 
				
			||||||
                            print(f"Player with name {command[1]} not found.")
 | 
					 | 
				
			||||||
                    if player_number:
 | 
					 | 
				
			||||||
                        item = " ".join(command[2:])
 | 
					 | 
				
			||||||
                        if item in Items.item_table:
 | 
					 | 
				
			||||||
                            seeked_item_id = Items.item_table[item][3]
 | 
					 | 
				
			||||||
                            for check, result in ctx.locations.items():
 | 
					 | 
				
			||||||
                                item_id, receiving_player = result
 | 
					 | 
				
			||||||
                                if receiving_player == player_number and item_id == seeked_item_id:
 | 
					 | 
				
			||||||
                                    location_id, finding_player = check
 | 
					 | 
				
			||||||
                                    notify_all(ctx, f"[Hint]: P{player_number}'s {item} can be found in {get_location_name_from_address(location_id)} in "
 | 
					 | 
				
			||||||
                                          f"P{finding_player}'s World")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
                        else:
 | 
					 | 
				
			||||||
                            print("Unknown item: " + item)
 | 
					 | 
				
			||||||
                else:
 | 
					 | 
				
			||||||
                    print("Use /hint {Playernumber/Playername} {itemname}\nFor example /hint 1 Lamp")
 | 
					 | 
				
			||||||
            elif command[0][0] != '/':
 | 
					 | 
				
			||||||
                notify_all(ctx, '[Server]: ' + input)
 | 
					 | 
				
			||||||
        except Exception as e:
 | 
					 | 
				
			||||||
            import traceback
 | 
					 | 
				
			||||||
            traceback.print_exc()
 | 
					 | 
				
			||||||
async def main():
 | 
					 | 
				
			||||||
    parser = argparse.ArgumentParser()
 | 
					 | 
				
			||||||
    parser.add_argument('--host', default=None)
 | 
					 | 
				
			||||||
    parser.add_argument('--port', default=38281, type=int)
 | 
					 | 
				
			||||||
    parser.add_argument('--password', default=None)
 | 
					 | 
				
			||||||
    parser.add_argument('--multidata', default=None)
 | 
					 | 
				
			||||||
    parser.add_argument('--savefile', default=None)
 | 
					 | 
				
			||||||
    parser.add_argument('--disable_save', default=False, action='store_true')
 | 
					 | 
				
			||||||
    parser.add_argument('--hint_timer', default=-1)
 | 
					 | 
				
			||||||
    args = parser.parse_args()
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    ctx = Context(args.host, args.port, args.password)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    ctx.data_filename = args.multidata
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    try:
 | 
					 | 
				
			||||||
        if not ctx.data_filename:
 | 
					 | 
				
			||||||
            import tkinter
 | 
					 | 
				
			||||||
            import tkinter.filedialog
 | 
					 | 
				
			||||||
            root = tkinter.Tk()
 | 
					 | 
				
			||||||
            root.withdraw()
 | 
					 | 
				
			||||||
            ctx.data_filename = tkinter.filedialog.askopenfilename(filetypes=(("Multiworld data","*multidata"),))
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        with open(ctx.data_filename, 'rb') as f:
 | 
					 | 
				
			||||||
            jsonobj = json.loads(zlib.decompress(f.read()).decode("utf-8"))
 | 
					 | 
				
			||||||
            ctx.players = jsonobj[0]
 | 
					 | 
				
			||||||
            ctx.rom_names = {k: v for k, v in jsonobj[1]}
 | 
					 | 
				
			||||||
            ctx.locations = {tuple(k): tuple(v) for k, v in jsonobj[2]}
 | 
					 | 
				
			||||||
    except Exception as e:
 | 
					 | 
				
			||||||
        print('Failed to read multiworld data (%s)' % e)
 | 
					 | 
				
			||||||
        return
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    ip = urllib.request.urlopen('https://v4.ident.me').read().decode('utf8') if not ctx.host else ctx.host
 | 
					 | 
				
			||||||
    print('Hosting game of %d players (%s) at %s:%d' % (ctx.players, 'No password' if not ctx.password else 'Password: %s' % ctx.password, ip, ctx.port))
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    ctx.disable_save = args.disable_save
 | 
					 | 
				
			||||||
    if not ctx.disable_save:
 | 
					 | 
				
			||||||
        if not ctx.save_filename:
 | 
					 | 
				
			||||||
            ctx.save_filename = (ctx.data_filename[:-9] if ctx.data_filename[-9:] == 'multidata' else (ctx.data_filename + '_')) + 'multisave'
 | 
					 | 
				
			||||||
        try:
 | 
					 | 
				
			||||||
            with open(ctx.save_filename, 'rb') as f:
 | 
					 | 
				
			||||||
                jsonobj = json.loads(zlib.decompress(f.read()).decode("utf-8"))
 | 
					 | 
				
			||||||
                players = jsonobj[0]
 | 
					 | 
				
			||||||
                rom_names = {k: v for k, v in jsonobj[1]}
 | 
					 | 
				
			||||||
                received_items = {tuple(k): [ReceivedItem(**i) for i in v] for k, v in jsonobj[2]}
 | 
					 | 
				
			||||||
                if players != ctx.players or rom_names != ctx.rom_names:
 | 
					 | 
				
			||||||
                    raise Exception('Save file mismatch, will start a new game')
 | 
					 | 
				
			||||||
                ctx.received_items = received_items
 | 
					 | 
				
			||||||
                print('Loaded save file with %d received items for %d players' % (sum([len(p) for p in received_items.values()]), len(received_items)))
 | 
					 | 
				
			||||||
        except FileNotFoundError:
 | 
					 | 
				
			||||||
            print('No save data found, starting a new game')
 | 
					 | 
				
			||||||
        except Exception as e:
 | 
					 | 
				
			||||||
            print(e)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    ctx.server = websockets.serve(functools.partial(server,ctx=ctx), ctx.host, ctx.port, ping_timeout=None, ping_interval=None)
 | 
					 | 
				
			||||||
    await ctx.server
 | 
					 | 
				
			||||||
    await console(ctx)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
if __name__ == '__main__':
 | 
					 | 
				
			||||||
    loop = asyncio.get_event_loop()
 | 
					 | 
				
			||||||
    loop.run_until_complete(main())
 | 
					 | 
				
			||||||
    loop.run_until_complete(asyncio.gather(*asyncio.Task.all_tasks()))
 | 
					 | 
				
			||||||
    loop.close()
 | 
					 | 
				
			||||||
							
								
								
									
										110
									
								
								MultiServer.py
								
								
								
								
							
							
						
						
									
										110
									
								
								MultiServer.py
								
								
								
								
							| 
						 | 
					@ -268,54 +268,76 @@ 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()
 | 
				
			||||||
 | 
					        try:
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        command = shlex.split(input)
 | 
					            command = shlex.split(input)
 | 
				
			||||||
        if not command:
 | 
					            if not command:
 | 
				
			||||||
            continue
 | 
					                continue
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        if command[0] == '/exit':
 | 
					            if command[0] == '/exit':
 | 
				
			||||||
            ctx.server.ws_server.close()
 | 
					                ctx.server.ws_server.close()
 | 
				
			||||||
            break
 | 
					                break
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        if command[0] == '/players':
 | 
					            if command[0] == '/players':
 | 
				
			||||||
            print(get_connected_players_string(ctx))
 | 
					                print(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() == player.lower():
 | 
					                    if client.auth and client.name.lower() == command[1].lower() and (team is None or team == client.team):
 | 
				
			||||||
                        new_item = ReceivedItem(Items.item_table[item][3], "cheat console", client.slot)
 | 
					                        if client.socket and not client.socket.closed:
 | 
				
			||||||
                        get_received_items(ctx, client.team, client.slot).append(new_item)
 | 
					                            await client.socket.close()
 | 
				
			||||||
                        notify_all(ctx, 'Cheat console: sending "' + item + '" to ' + client.name)
 | 
					 | 
				
			||||||
                send_new_items(ctx)
 | 
					 | 
				
			||||||
            else:
 | 
					 | 
				
			||||||
                print("Unknown item: " + item)
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
        if command[0][0] != '/':
 | 
					            if command[0] == '/forfeitslot' and len(command) > 1 and command[1].isdigit():
 | 
				
			||||||
            notify_all(ctx, '[Server]: ' + input)
 | 
					                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:
 | 
				
			||||||
 | 
					                        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:
 | 
				
			||||||
 | 
					                    print("Unknown item: " + item)
 | 
				
			||||||
 | 
					            if command[0] == '/hint':
 | 
				
			||||||
 | 
					                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):
 | 
				
			||||||
 | 
					                        item = " ".join(command[2:])
 | 
				
			||||||
 | 
					                        if item in Items.item_table:
 | 
				
			||||||
 | 
					                            seeked_item_id = Items.item_table[item][3]
 | 
				
			||||||
 | 
					                            for check, result in ctx.locations.items():
 | 
				
			||||||
 | 
					                                item_id, receiving_player = result
 | 
				
			||||||
 | 
					                                print(receiving_player, client.slot)
 | 
				
			||||||
 | 
					                                if receiving_player == client.slot and item_id == seeked_item_id:
 | 
				
			||||||
 | 
					                                    location_id, finding_player = check
 | 
				
			||||||
 | 
					                                    notify_all(ctx, f"[Hint]: P{client.name}'s {item} can be found in {get_location_name_from_address(location_id)} in "
 | 
				
			||||||
 | 
					                                          f"P{finding_player}'s World")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                        else:
 | 
				
			||||||
 | 
					                            print("Unknown item: " + item)
 | 
				
			||||||
 | 
					                else:
 | 
				
			||||||
 | 
					                    print("Use /hint {Playername} {itemname}\nFor example /hint Berserker Lamp")
 | 
				
			||||||
 | 
					            if command[0][0] != '/':
 | 
				
			||||||
 | 
					                notify_all(ctx, '[Server]: ' + input)
 | 
				
			||||||
 | 
					        except:
 | 
				
			||||||
 | 
					            import traceback
 | 
				
			||||||
 | 
					            traceback.print_exc()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
async def main():
 | 
					async def main():
 | 
				
			||||||
    parser = argparse.ArgumentParser()
 | 
					    parser = argparse.ArgumentParser()
 | 
				
			||||||
| 
						 | 
					@ -379,4 +401,4 @@ if __name__ == '__main__':
 | 
				
			||||||
    loop = asyncio.get_event_loop()
 | 
					    loop = asyncio.get_event_loop()
 | 
				
			||||||
    loop.run_until_complete(main())
 | 
					    loop.run_until_complete(main())
 | 
				
			||||||
    loop.run_until_complete(asyncio.gather(*asyncio.Task.all_tasks()))
 | 
					    loop.run_until_complete(asyncio.gather(*asyncio.Task.all_tasks()))
 | 
				
			||||||
    loop.close()
 | 
					    loop.close()
 | 
				
			||||||
		Loading…
	
		Reference in New Issue