1
0
forked from Hay1tsme/artemis
artemis/titles/pokken/index.py

127 lines
4.1 KiB
Python
Raw Normal View History

from typing import Tuple
2023-02-17 07:09:26 +00:00
from twisted.web.http import Request
from twisted.web import resource, server
from twisted.internet import reactor, endpoints
import yaml
import logging, coloredlogs
from logging.handlers import TimedRotatingFileHandler
from titles.pokken.proto import jackal_pb2
2023-02-27 21:51:17 +00:00
from os import path
2023-02-28 05:41:32 +00:00
from google.protobuf.message import DecodeError
2023-02-17 07:09:26 +00:00
from core.config import CoreConfig
from titles.pokken.config import PokkenConfig
from titles.pokken.base import PokkenBase
from titles.pokken.const import PokkenConstants
2023-02-17 07:09:26 +00:00
2023-03-09 16:38:58 +00:00
2023-02-17 07:09:26 +00:00
class PokkenServlet(resource.Resource):
def __init__(self, core_cfg: CoreConfig, cfg_dir: str) -> None:
self.isLeaf = True
self.core_cfg = core_cfg
self.config_dir = cfg_dir
self.game_cfg = PokkenConfig()
if path.exists(f"{cfg_dir}/pokken.yaml"):
self.game_cfg.update(yaml.safe_load(open(f"{cfg_dir}/pokken.yaml")))
2023-02-17 07:09:26 +00:00
self.logger = logging.getLogger("pokken")
if not hasattr(self.logger, "inited"):
log_fmt_str = "[%(asctime)s] Pokken | %(levelname)s | %(message)s"
log_fmt = logging.Formatter(log_fmt_str)
2023-03-09 16:38:58 +00:00
fileHandler = TimedRotatingFileHandler(
"{0}/{1}.log".format(self.core_cfg.server.log_dir, "pokken"),
encoding="utf8",
when="d",
backupCount=10,
)
2023-02-17 07:09:26 +00:00
fileHandler.setFormatter(log_fmt)
2023-03-09 16:38:58 +00:00
2023-02-17 07:09:26 +00:00
consoleHandler = logging.StreamHandler()
consoleHandler.setFormatter(log_fmt)
self.logger.addHandler(fileHandler)
self.logger.addHandler(consoleHandler)
2023-03-09 16:38:58 +00:00
2023-02-17 07:09:26 +00:00
self.logger.setLevel(self.game_cfg.server.loglevel)
2023-03-09 16:38:58 +00:00
coloredlogs.install(
level=self.game_cfg.server.loglevel, logger=self.logger, fmt=log_fmt_str
)
2023-02-17 07:09:26 +00:00
self.logger.inited = True
self.base = PokkenBase(core_cfg, self.game_cfg)
@classmethod
2023-03-09 16:38:58 +00:00
def get_allnet_info(
cls, game_code: str, core_cfg: CoreConfig, cfg_dir: str
) -> Tuple[bool, str, str]:
game_cfg = PokkenConfig()
2023-03-09 16:38:58 +00:00
if path.exists(f"{cfg_dir}/{PokkenConstants.CONFIG_NAME}"):
2023-03-09 16:38:58 +00:00
game_cfg.update(
yaml.safe_load(open(f"{cfg_dir}/{PokkenConstants.CONFIG_NAME}"))
)
if not game_cfg.server.enable:
return (False, "", "")
2023-03-09 16:38:58 +00:00
return (
True,
f"https://{game_cfg.server.hostname}:{game_cfg.server.port}/{game_code}/$v/",
f"{game_cfg.server.hostname}/SDAK/$v/",
2023-03-09 16:38:58 +00:00
)
2023-03-05 03:46:26 +00:00
@classmethod
2023-03-09 16:38:58 +00:00
def get_mucha_info(
cls, core_cfg: CoreConfig, cfg_dir: str
) -> Tuple[bool, str, str]:
2023-03-05 03:46:26 +00:00
game_cfg = PokkenConfig()
2023-03-09 16:38:58 +00:00
2023-03-05 03:46:26 +00:00
if path.exists(f"{cfg_dir}/{PokkenConstants.CONFIG_NAME}"):
2023-03-09 16:38:58 +00:00
game_cfg.update(
yaml.safe_load(open(f"{cfg_dir}/{PokkenConstants.CONFIG_NAME}"))
)
2023-03-05 03:46:26 +00:00
if not game_cfg.server.enable:
return (False, "")
2023-03-09 16:38:58 +00:00
return (True, "PKF2")
2023-03-09 16:38:58 +00:00
def setup(self) -> None:
# TODO: Setup matching, stun, turn and admission servers
pass
2023-02-17 07:09:26 +00:00
2023-03-09 16:38:58 +00:00
def render_POST(
self, request: Request, version: int = 0, endpoints: str = ""
) -> bytes:
2023-02-28 05:41:32 +00:00
if endpoints.startswith("/matching"):
2023-02-17 07:09:26 +00:00
self.logger.info("Matching request")
self.logger.debug(request.content)
2023-03-09 16:38:58 +00:00
2023-02-28 05:41:32 +00:00
content = request.content.getvalue()
if content == b"":
self.logger.info("Empty request")
return b""
2023-02-17 07:09:26 +00:00
pokken_request = jackal_pb2.Request()
2023-02-28 05:41:32 +00:00
try:
pokken_request.ParseFromString(content)
except DecodeError as e:
self.logger.warn(f"{e} {content}")
return b""
endpoint = jackal_pb2.MessageType.DESCRIPTOR.values_by_number[
pokken_request.type
].name.lower()
2023-03-09 16:38:58 +00:00
2023-02-17 07:09:26 +00:00
self.logger.info(f"{endpoint} request")
handler = getattr(self.base, f"handle_{endpoint}", None)
if handler is None:
self.logger.warn(f"No handler found for message type {endpoint}")
return self.base.handle_noop(pokken_request)
2023-03-12 01:02:58 +00:00
ret = handler(pokken_request)
self.logger.debug(f"Response: {ret}")
return ret