151 lines
5.3 KiB
Python
151 lines
5.3 KiB
Python
from __future__ import annotations
|
|
|
|
import functools
|
|
import logging
|
|
import os
|
|
import websockets
|
|
import asyncio
|
|
import socket
|
|
import threading
|
|
import time
|
|
import random
|
|
import zlib
|
|
|
|
|
|
from .models import *
|
|
|
|
from MultiServer import Context, server, auto_shutdown, ServerCommandProcessor, ClientMessageProcessor
|
|
from Utils import get_public_ipv4, get_public_ipv6, parse_yaml
|
|
|
|
|
|
class CustomClientMessageProcessor(ClientMessageProcessor):
|
|
ctx: WebHostContext
|
|
def _cmd_video(self, platform, user):
|
|
"""Set a link for your name in the WebHostLib tracker pointing to a video stream"""
|
|
if platform.lower().startswith("t"): # twitch
|
|
self.ctx.video[self.client.team, self.client.slot] = "Twitch", user
|
|
self.ctx.save()
|
|
self.output(f"Registered Twitch Stream https://www.twitch.tv/{user}")
|
|
return True
|
|
elif platform.lower().startswith("y"): # youtube
|
|
self.ctx.video[self.client.team, self.client.slot] = "Youtube", user
|
|
self.ctx.save()
|
|
self.output(f"Registered Youtube Stream for {user}")
|
|
return True
|
|
return False
|
|
|
|
|
|
# inject
|
|
import MultiServer
|
|
MultiServer.client_message_processor = CustomClientMessageProcessor
|
|
del (MultiServer)
|
|
|
|
|
|
class DBCommandProcessor(ServerCommandProcessor):
|
|
def output(self, text: str):
|
|
logging.info(text)
|
|
|
|
|
|
class WebHostContext(Context):
|
|
def __init__(self):
|
|
super(WebHostContext, self).__init__("", 0, "", "", 1, 40, True, "enabled", "enabled", 0, 2)
|
|
self.main_loop = asyncio.get_running_loop()
|
|
self.video = {}
|
|
self.tags = ["AP", "WebHost"]
|
|
|
|
def listen_to_db_commands(self):
|
|
cmdprocessor = DBCommandProcessor(self)
|
|
|
|
while self.running:
|
|
with db_session:
|
|
commands = select(command for command in Command if command.room.id == self.room_id)
|
|
if commands:
|
|
for command in commands:
|
|
self.main_loop.call_soon_threadsafe(cmdprocessor, command.commandtext)
|
|
command.delete()
|
|
commit()
|
|
time.sleep(5)
|
|
|
|
@db_session
|
|
def load(self, room_id: int):
|
|
self.room_id = room_id
|
|
room = Room.get(id=room_id)
|
|
if room.last_port:
|
|
self.port = room.last_port
|
|
else:
|
|
self.port = get_random_port()
|
|
|
|
return self._load(self._decompress(room.seed.multidata), True)
|
|
|
|
@db_session
|
|
def init_save(self, enabled: bool = True):
|
|
self.saving = enabled
|
|
if self.saving:
|
|
existing_savegame = Room.get(id=self.room_id).multisave
|
|
if existing_savegame:
|
|
self.set_save(existing_savegame)
|
|
self._start_async_saving()
|
|
threading.Thread(target=self.listen_to_db_commands, daemon=True).start()
|
|
|
|
@db_session
|
|
def _save(self, exit_save:bool = False) -> bool:
|
|
room = Room.get(id=self.room_id)
|
|
room.multisave = self.get_save()
|
|
# saving only occurs on activity, so we can "abuse" this information to mark this as last_activity
|
|
if not exit_save: # we don't want to count a shutdown as activity, which would restart the server again
|
|
room.last_activity = datetime.utcnow()
|
|
return True
|
|
|
|
def get_save(self) -> dict:
|
|
d = super(WebHostContext, self).get_save()
|
|
d["video"] = [(tuple(playerslot), videodata) for playerslot, videodata in self.video.items()]
|
|
return d
|
|
|
|
def get_random_port():
|
|
return random.randint(49152, 65535)
|
|
|
|
def run_server_process(room_id, ponyconfig: dict):
|
|
# establish DB connection for multidata and multisave
|
|
db.bind(**ponyconfig)
|
|
db.generate_mapping(check_tables=False)
|
|
|
|
async def main():
|
|
|
|
logging.basicConfig(format='[%(asctime)s] %(message)s',
|
|
level=logging.INFO,
|
|
handlers=[
|
|
logging.FileHandler(os.path.join(LOGS_FOLDER, f"{room_id}.txt"), 'a', 'utf-8-sig')])
|
|
ctx = WebHostContext()
|
|
ctx.load(room_id)
|
|
ctx.init_save()
|
|
|
|
try:
|
|
ctx.server = websockets.serve(functools.partial(server, ctx=ctx), ctx.host, ctx.port, ping_timeout=None,
|
|
ping_interval=None)
|
|
|
|
await ctx.server
|
|
except Exception: # likely port in use - in windows this is OSError, but I didn't check the others
|
|
ctx.server = websockets.serve(functools.partial(server, ctx=ctx), ctx.host, 0, ping_timeout=None,
|
|
ping_interval=None)
|
|
|
|
await ctx.server
|
|
for wssocket in ctx.server.ws_server.sockets:
|
|
socketname = wssocket.getsockname()
|
|
if wssocket.family == socket.AF_INET6:
|
|
logging.info(f'Hosting game at [{get_public_ipv6()}]:{socketname[1]}')
|
|
with db_session:
|
|
room = Room.get(id=ctx.room_id)
|
|
room.last_port = socketname[1]
|
|
elif wssocket.family == socket.AF_INET:
|
|
logging.info(f'Hosting game at {get_public_ipv4()}:{socketname[1]}')
|
|
with db_session:
|
|
ctx.auto_shutdown = Room.get(id=room_id).timeout
|
|
ctx.shutdown_task = asyncio.create_task(auto_shutdown(ctx, []))
|
|
await ctx.shutdown_task
|
|
logging.info("Shutting down")
|
|
|
|
asyncio.run(main())
|
|
|
|
|
|
from WebHostLib import LOGS_FOLDER
|