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

188 lines
6.2 KiB
Python
Raw Normal View History

from typing import Tuple
2023-02-17 07:09:26 +00:00
from twisted.web.http import Request
2023-03-12 04:42:12 +00:00
from twisted.web import resource
from twisted.internet import reactor, endpoints
from autobahn.twisted.websocket import WebSocketServerFactory
2023-03-12 04:42:12 +00:00
import json, ast
from datetime import datetime
2023-02-17 07:09:26 +00:00
import yaml
import logging, coloredlogs
from logging.handlers import TimedRotatingFileHandler
2023-03-12 04:42:12 +00:00
import inflection
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 import CoreConfig, Utils
from .config import PokkenConfig
from .base import PokkenBase
from .const import PokkenConstants
from .proto import jackal_pb2
from .services import PokkenStunProtocol, PokkenAdmissionFactory, PokkenAdmissionProtocol
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
2023-04-12 06:34:29 +00:00
return (True, "PKF1")
2023-03-09 16:38:58 +00:00
def setup(self) -> None:
2023-03-12 04:42:12 +00:00
# TODO: Setup stun, turn (UDP) and admission (WSS) servers
reactor.listenUDP(
self.game_cfg.server.port_stun, PokkenStunProtocol(self.core_cfg, self.game_cfg, "Stun")
)
reactor.listenUDP(
self.game_cfg.server.port_turn, PokkenStunProtocol(self.core_cfg, self.game_cfg, "Turn")
)
factory = WebSocketServerFactory(f"ws://{self.game_cfg.server.hostname}:{self.game_cfg.server.port_admission}")
factory.protocol = PokkenAdmissionProtocol
reactor.listenTCP(
self.game_cfg.server.port_admission, PokkenAdmissionFactory(self.core_cfg, self.game_cfg)
)
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-03-12 04:42:12 +00:00
if endpoints == "matching":
return self.handle_matching(request)
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-04-12 06:34:29 +00:00
self.logger.debug(pokken_request)
2023-03-09 16:38:58 +00:00
2023-02-17 07:09:26 +00:00
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 20:30:57 +00:00
self.logger.info(f"{endpoint} request from {Utils.get_ip_addr(request)}")
2023-03-12 01:02:58 +00:00
ret = handler(pokken_request)
return ret
2023-03-12 04:42:12 +00:00
def handle_matching(self, request: Request) -> bytes:
content = request.content.getvalue()
client_ip = Utils.get_ip_addr(request)
2023-03-12 04:42:12 +00:00
if content is None or content == b"":
self.logger.info("Empty matching request")
return json.dumps(self.base.handle_matching_noop()).encode()
json_content = ast.literal_eval(
content.decode()
.replace("null", "None")
.replace("true", "True")
.replace("false", "False")
)
2023-03-12 04:42:12 +00:00
self.logger.info(f"Matching {json_content['call']} request")
self.logger.debug(json_content)
handler = getattr(
self.base,
f"handle_matching_{inflection.underscore(json_content['call'])}",
None,
)
2023-03-12 04:42:12 +00:00
if handler is None:
self.logger.warn(
f"No handler found for message type {json_content['call']}"
)
2023-03-12 04:42:12 +00:00
return json.dumps(self.base.handle_matching_noop()).encode()
2023-03-12 04:42:12 +00:00
ret = handler(json_content, client_ip)
2023-03-12 04:42:12 +00:00
if ret is None:
ret = {}
2023-03-12 04:42:12 +00:00
if "result" not in ret:
ret["result"] = "true"
if "data" not in ret:
ret["data"] = {}
if "timestamp" not in ret:
ret["timestamp"] = int(datetime.now().timestamp() * 1000)
2023-03-12 04:42:12 +00:00
self.logger.debug(f"Response {ret}")
return json.dumps(ret).encode()