artemis/core/mucha.py

from typing import Dict, Any, Optional
import logging, coloredlogs
from logging.handlers import TimedRotatingFileHandler
from twisted.web import resource
from twisted.web.http import Request
from datetime import datetime
import pytz

from core.config import CoreConfig

class MuchaServlet:
    def __init__(self, cfg: CoreConfig) -> None:        
        self.config = cfg

        self.logger = logging.getLogger('mucha')
        log_fmt_str = "[%(asctime)s] Mucha | %(levelname)s | %(message)s"
        log_fmt = logging.Formatter(log_fmt_str)
        
        fileHandler = TimedRotatingFileHandler("{0}/{1}.log".format(self.config.server.log_dir, "mucha"), when="d", backupCount=10)
        fileHandler.setFormatter(log_fmt)
        
        consoleHandler = logging.StreamHandler()
        consoleHandler.setFormatter(log_fmt)

        self.logger.addHandler(fileHandler)
        self.logger.addHandler(consoleHandler)
        
        self.logger.setLevel(logging.INFO)
        coloredlogs.install(level=logging.INFO, logger=self.logger, fmt=log_fmt_str)

    def handle_boardauth(self, request: Request, _: Dict) -> bytes:
        req_dict = self.mucha_preprocess(request.content.getvalue())
        if req_dict is None:
            self.logger.error(f"Error processing mucha request {request.content.getvalue()}")
            return b""

        req = MuchaAuthRequest(req_dict)
        self.logger.info(f"Mucha request {vars(req)}")
        resp = MuchaAuthResponse(mucha_url=f"{self.config.mucha.hostname}:{self.config.mucha.port}")
        self.logger.info(f"Mucha response {vars(resp)}")

        return self.mucha_postprocess(vars(resp))
    
    def handle_updatecheck(self, request: Request, _: Dict) -> bytes:
        req_dict = self.mucha_preprocess(request.content.getvalue())
        if req_dict is None:
            self.logger.error(f"Error processing mucha request {request.content.getvalue()}")
            return b""

        req = MuchaUpdateRequest(req_dict)
        self.logger.info(f"Mucha request {vars(req)}")
        resp = MuchaUpdateResponse(mucha_url=f"{self.config.mucha.hostname}:{self.config.mucha.port}")
        self.logger.info(f"Mucha response {vars(resp)}")

        return self.mucha_postprocess(vars(resp))

    def mucha_preprocess(self, data: bytes) -> Optional[Dict]:
        try:
            ret: Dict[str, Any] = {}
            
            for x in data.decode().split('&'):
                kvp = x.split('=')
                if len(kvp) == 2:
                    ret[kvp[0]] = kvp[1]

            return ret
        
        except:
            self.logger.error(f"Error processing mucha request {data}")
            return None

    def mucha_postprocess(self, data: dict) -> Optional[bytes]:
        try:
            urlencode = ""
            for k,v in data.items():
                urlencode += f"{k}={v}&"

            return urlencode.encode()

        except:
            self.logger.error("Error processing mucha response")
            return None

class MuchaAuthRequest():
    def __init__(self, request: Dict) -> None:
        self.gameVer = "" if "gameVer" not in request else request["gameVer"]
        self.sendDate = "" if "sendDate" not in request else request["sendDate"]
        self.serialNum = "" if "serialNum" not in request else request["serialNum"]
        self.gameCd = "" if "gameCd" not in request else request["gameCd"]
        self.boardType = "" if "boardType" not in request else request["boardType"]
        self.boardId = "" if "boardId" not in request else request["boardId"]
        self.placeId = "" if "placeId" not in request else request["placeId"]
        self.storeRouterIp = "" if "storeRouterIp" not in request else request["storeRouterIp"]
        self.countryCd = "" if "countryCd" not in request else request["countryCd"]
        self.useToken = "" if "useToken" not in request else request["useToken"]
        self.allToken = "" if "allToken" not in request else request["allToken"]

class MuchaAuthResponse():
    def __init__(self, mucha_url: str = "localhost") -> None:
        self.RESULTS = "001"        
        self.AUTH_INTERVAL = "86400"
        self.SERVER_TIME = datetime.strftime(datetime.now(), "%Y%m%d%H%M")
        self.UTC_SERVER_TIME = datetime.strftime(datetime.now(pytz.UTC), "%Y%m%d%H%M")

        self.CHARGE_URL = f"https://{mucha_url}/charge/"        
        self.FILE_URL = f"https://{mucha_url}/file/"
        self.URL_1 = f"https://{mucha_url}/url1/"
        self.URL_2 = f"https://{mucha_url}/url2/"
        self.URL_3 = f"https://{mucha_url}/url3/"
        
        self.PLACE_ID = "JPN123"        
        self.COUNTRY_CD = "JPN"        
        self.SHOP_NAME = "TestShop!"
        self.SHOP_NICKNAME = "TestShop"
        self.AREA_0 = "008"
        self.AREA_1 = "009"
        self.AREA_2 = "010"
        self.AREA_3 = "011"
        self.AREA_FULL_0 = ""
        self.AREA_FULL_1 = ""
        self.AREA_FULL_2 = ""
        self.AREA_FULL_3 = ""
        
        self.SHOP_NAME_EN = "TestShop!"
        self.SHOP_NICKNAME_EN = "TestShop"
        self.AREA_0_EN = "008"
        self.AREA_1_EN = "009"
        self.AREA_2_EN = "010"
        self.AREA_3_EN = "011"
        self.AREA_FULL_0_EN = ""
        self.AREA_FULL_1_EN = ""
        self.AREA_FULL_2_EN = ""
        self.AREA_FULL_3_EN = ""

        self.PREFECTURE_ID = "1"        
        self.EXPIRATION_DATE = "null"
        self.USE_TOKEN = "0"
        self.CONSUME_TOKEN = "0"
        self.DONGLE_FLG = "1"
        self.FORCE_BOOT = "0"

class MuchaUpdateRequest():
    def __init__(self, request: Dict) -> None:
        self.gameVer = "" if "gameVer" not in request else request["gameVer"]
        self.gameCd = "" if "gameCd" not in request else request["gameCd"]
        self.serialNum = "" if "serialNum" not in request else request["serialNum"]
        self.countryCd = "" if "countryCd" not in request else request["countryCd"]
        self.placeId = "" if "placeId" not in request else request["placeId"]
        self.storeRouterIp = "" if "storeRouterIp" not in request else request["storeRouterIp"]

class MuchaUpdateResponse():
    def __init__(self, game_ver: str = "PKFN0JPN01.01", mucha_url: str = "localhost") -> None:
        self.RESULTS = "001"
        self.UPDATE_VER_1 = game_ver
        self.UPDATE_URL_1 = f"https://{mucha_url}/updUrl1/"
        self.UPDATE_SIZE_1 = "0"
        self.UPDATE_CRC_1 = "0000000000000000"
        self.CHECK_URL_1 = f"https://{mucha_url}/checkUrl/"
        self.EXE_VER_1 = game_ver
        self.INFO_SIZE_1 = "0"
        self.COM_SIZE_1 = "0"
        self.COM_TIME_1 = "0"
        self.LAN_INFO_SIZE_1 = "0"