Overhaul Sessioning

src/ezgg_lan_manager/components/UserInfoBox.py

@@ -6,6 +6,7 @@ from rio import Component, TextStyle, Color, Button, Text, Rectangle, Column, Ro
 
 from src.ezgg_lan_manager.components.UserInfoBoxButton import UserInfoBoxButton
 from src.ezgg_lan_manager.services.LocalDataService import LocalData, LocalDataService
+from src.ezgg_lan_manager.services.RefreshService import RefreshService
 from src.ezgg_lan_manager.services.UserService import UserService
 from src.ezgg_lan_manager.services.AccountingService import AccountingService
 from src.ezgg_lan_manager.services.TicketingService import TicketingService
@@ -13,7 +14,7 @@ from src.ezgg_lan_manager.services.SeatingService import SeatingService
 from src.ezgg_lan_manager.types.Seat import Seat
 from src.ezgg_lan_manager.types.Ticket import Ticket
 from src.ezgg_lan_manager.types.User import User
-from src.ezgg_lan_manager.types.SessionStorage import SessionStorage
+from src.ezgg_lan_manager.types.UserSession import UserSession
 
 
 class StatusButton(Component):
@@ -41,6 +42,7 @@ class StatusButton(Component):
 
 
 class UserInfoBox(Component):
+    user_id: int
     status_change_cb: EventHandler = None
     TEXT_STYLE = TextStyle(fill=Color.from_hex("02dac5"), font_size=0.9)
     user: Optional[User] = None
@@ -53,31 +55,26 @@ class UserInfoBox(Component):
         return choice(["Guten Tacho", "Tuten Gag", "Servus", "Moinjour", "Hallöchen", "Heyho", "Moinsen"])
 
     async def logout(self) -> None:
-        await self.session[SessionStorage].clear()
+        self.session.detach(UserSession)
         self.user = None
         self.session[LocalDataService].del_session(self.session[LocalData].stored_session_token)
         self.session[LocalData].stored_session_token = None
         self.session.attach(self.session[LocalData])
-        self.status_change_cb()
-        self.session.navigate_to("/")
+        await self.status_change_cb()
+        await self.session[RefreshService].trigger_refresh()
 
     @event.on_populate
     async def async_init(self) -> None:
-        if self.session[SessionStorage].user_id:
-            self.user = await self.session[UserService].get_user(self.session[SessionStorage].user_id)
-            self.user_balance = await self.session[AccountingService].get_balance(self.user.user_id)
-            self.user_ticket = await self.session[TicketingService].get_user_ticket(self.user.user_id)
-            self.user_seat = await self.session[SeatingService].get_user_seat(self.user.user_id)
-            self.session[AccountingService].add_update_hook(self.update)
-
-    async def update(self) -> None:
-        if not self.user:
-            self.user = await self.session[UserService].get_user(self.session[SessionStorage].user_id)
-        if not self.user:
-            return
+        self.user = await self.session[UserService].get_user(self.user_id)
         self.user_balance = await self.session[AccountingService].get_balance(self.user.user_id)
         self.user_ticket = await self.session[TicketingService].get_user_ticket(self.user.user_id)
         self.user_seat = await self.session[SeatingService].get_user_seat(self.user.user_id)
+        self.session[AccountingService].add_update_hook(self.update)
+
+    async def update(self) -> None:
+        self.user_balance = await self.session[AccountingService].get_balance(self.user_id)
+        self.user_ticket = await self.session[TicketingService].get_user_ticket(self.user_id)
+        self.user_seat = await self.session[SeatingService].get_user_seat(self.user_id)
 
     def build(self) -> Component:
         if not self.user: