import yaml import jinja2 from typing import List from starlette.requests import Request from starlette.responses import Response, RedirectResponse from starlette.routing import Route from os import path from core.frontend import FE_Base, UserSession from core.config import CoreConfig from .database import PokkenData from .config import PokkenConfig from .const import PokkenConstants class PokkenFrontend(FE_Base): SN_PREFIX = PokkenConstants.SERIAL_IDENT NETID_PREFIX = PokkenConstants.NETID_PREFIX def __init__( self, cfg: CoreConfig, environment: jinja2.Environment, cfg_dir: str ) -> None: super().__init__(cfg, environment) self.data = PokkenData(cfg) self.game_cfg = PokkenConfig() if path.exists(f"{cfg_dir}/{PokkenConstants.CONFIG_NAME}"): self.game_cfg.update( yaml.safe_load(open(f"{cfg_dir}/{PokkenConstants.CONFIG_NAME}")) ) self.nav_name = "Pokken" def get_routes(self) -> List[Route]: return [ Route("/", self.render_GET, methods=['GET']), Route("/update.name", self.change_name, methods=['POST']), ] async def render_GET(self, request: Request) -> Response: template = self.environment.get_template( "titles/pokken/templates/pokken_index.jinja" ) pf = None usr_sesh = self.validate_session(request) if not usr_sesh: usr_sesh = UserSession() else: profile = await self.data.profile.get_profile(usr_sesh.user_id) if profile is not None and profile['trainer_name']: pf = profile._asdict() if "e" in request.query_params: try: err = int(request.query_params.get("e", 0)) except Exception: err = 0 else: err = 0 if "s" in request.query_params: try: succ = int(request.query_params.get("s", 0)) except Exception: succ = 0 else: succ = 0 return Response(template.render( title=f"{self.core_config.server.name} | {self.nav_name}", game_list=self.environment.globals["game_list"], sesh=vars(usr_sesh), profile=pf, error=err, success=succ )) async def change_name(self, request: Request) -> RedirectResponse: usr_sesh = self.validate_session(request) if not usr_sesh: return RedirectResponse("/game/pokken/?e=9", 303) frm = await request.form() new_name = frm.get("new_name") gender = frm.get("new_gender", 1) if len(new_name) > 14: return RedirectResponse("/game/pokken/?e=8", 303) if not gender.isdigit(): return RedirectResponse("/game/pokken/?e=4", 303) gender = int(gender) if gender != 1 and gender != 2: return RedirectResponse("/game/pokken/?e=4", 303) # no code for this yet, whatever await self.data.profile.set_profile_name(usr_sesh.user_id, new_name, gender) return RedirectResponse("/game/pokken/?s=1", 303)