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)

titles/pokken/frontend/pokken_index.jinja

@@ -1,4 +0,0 @@
-{% extends "core/frontend/index.jinja" %}
-{% block content %}
-<h1>Pokken</h1>
-{% endblock content %}

titles/pokken/templates/pokken_index.jinja

@@ -0,0 +1,48 @@
+{% extends "core/templates/index.jinja" %}
+{% block content %}
+<h1>Pokken</h1>
+{% if profile is defined and profile is not none and profile.id > 0 %}
+<script type="text/javascript">
+function toggle_new_name_form() {
+  let frm = document.getElementById("new_name_form");
+  let btn = document.getElementById("btn_toggle_form");
+
+  if (frm.style['display'] != "") {
+    frm.style['display'] = "";
+    frm.style['max-height'] = "";
+    btn.innerText = "Cancel";
+  } else {
+    frm.style['display'] = "none";
+    frm.style['max-height'] = "0px";
+    btn.innerText = "Edit";
+  }
+}
+</script>
+<h3>Profile for {{ profile.trainer_name }}&nbsp;<button onclick="toggle_new_name_form()" class="btn btn-secondary" id="btn_toggle_form">Edit</button></h3>
+{% if error is defined %}
+{% include "core/templates/widgets/err_banner.jinja" %}
+{% endif %}
+{% if success is defined and success == 1 %}
+<div style="background-color: #00AA00; padding: 20px; margin-bottom: 10px; width: 15%;">
+Update successful
+</div>
+{% endif %}
+<form style="max-width: 33%; display: none; max-height: 0px;" action="/game/pokken/update.name" method="post" id="new_name_form">
+  <div class="mb-3">
+    <label for="new_name" class="form-label">New Trainer Name</label>
+    <input type="text" class="form-control" id="new_name" name="new_name" aria-describedby="new_name_help" maxlength="14">
+    <div id="new_name_help" class="form-text">Must be 14 characters or less</div>
+    <br>
+    <input type="radio" id="new_gender_male" name="new_gender" value="1" {% if profile.avatar_gender is none or profile.avatar_gender == 1 %}checked{% endif %}>
+    <label for="new_gender_male">Male</label>
+    <input type="radio" id="new_gender_female" name="new_gender" value="2" {% if profile.avatar_gender == 2 %}checked{% endif %}>
+    <label for="new_gender_male">Female</label>
+  </div>
+  <button type="submit" class="btn btn-primary">Submit</button>
+</form>
+{% elif sesh is defined and sesh is not none and sesh.user_id > 0 %}
+No profile information found for this account.
+{% else %}
+Login to view profile information.
+{% endif %}
+{% endblock content %}