port frontend changes from different project

titles/pokken/frontend.py

@@ -1,10 +1,12 @@
 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 twisted.web.server import Session
 
-from core.frontend import FE_Base, IUserSession
+from core.frontend import FE_Base, UserSession
 from core.config import CoreConfig
 from .database import PokkenData
 from .config import PokkenConfig
@@ -12,6 +14,8 @@ 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:
@@ -23,17 +27,75 @@ class PokkenFrontend(FE_Base):
                 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']),
+        ]
 
-    def render_GET(self, request: Request) -> bytes:
+    async def render_GET(self, request: Request) -> Response:
         template = self.environment.get_template(
-            "titles/pokken/frontend/pokken_index.jinja"
+            "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
 
-        sesh: Session = request.getSession()
-        usr_sesh = IUserSession(sesh)
+        else:
+            err = 0
+        
+        if "s" in request.query_params:
+            try:
+                succ = int(request.query_params.get("s", 0))
+            except Exception:
+                succ = 0
 
-        return template.render(
+        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)
-        ).encode("utf-16")
+            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)