Overhaul Sessioning

src/ezgg_lan_manager/components/DesktopNavigation.py

@@ -1,47 +1,29 @@
 from typing import Optional, Callable
 
-from rio import *
+from rio import Component, event, Spacer, Card, Column, Text, TextStyle
 
-from src.ezgg_lan_manager import ConfigurationService, UserService, LocalDataService
+from src.ezgg_lan_manager.services.ConfigurationService import ConfigurationService
+from src.ezgg_lan_manager.services.UserService import UserService
 from src.ezgg_lan_manager.components.DesktopNavigationButton import DesktopNavigationButton
 from src.ezgg_lan_manager.components.NavigationSponsorBox import NavigationSponsorBox
 from src.ezgg_lan_manager.components.UserInfoAndLoginBox import UserInfoAndLoginBox
-from src.ezgg_lan_manager.services.LocalDataService import LocalData
-from src.ezgg_lan_manager.types.SessionStorage import SessionStorage
 from src.ezgg_lan_manager.types.User import User
+from src.ezgg_lan_manager.types.UserSession import UserSession
 
 
 class DesktopNavigation(Component):
     user: Optional[User] = None
-    force_login_box_refresh: list[Callable] = []
 
     @event.on_populate
-    async def async_init(self) -> None:
-        self.session[SessionStorage].subscribe_to_logged_in_or_out_event(str(self.__class__), self.async_init)
-        local_data = self.session[LocalData]
-        if local_data.stored_session_token:
-            session_ = self.session[LocalDataService].verify_token(local_data.stored_session_token)
-            if session_:
-                self.session.detach(SessionStorage)
-                self.session.attach(session_)
-                self.user = await self.session[UserService].get_user(session_.user_id)
-                try:
-                    # Hack-around, maybe fix in the future
-                    self.force_login_box_refresh[-1]()
-                except IndexError:
-                    pass
-
-                return
-
-        if self.session[SessionStorage].user_id:
-            self.user = await self.session[UserService].get_user(self.session[SessionStorage].user_id)
-        else:
+    async def on_populate(self) -> None:
+        try:
+            self.user = await self.session[UserService].get_user(self.session[UserSession].user_id)
+        except KeyError:
             self.user = None
 
     def build(self) -> Component:
         lan_info = self.session[ConfigurationService].get_lan_info()
-        user_info_and_login_box = UserInfoAndLoginBox()
-        self.force_login_box_refresh.append(user_info_and_login_box.force_refresh)
+        user_info_and_login_box = UserInfoAndLoginBox(state_changed_cb=self.on_populate)
         navigation = [
             DesktopNavigationButton("News", "./news"),
             Spacer(min_height=0.7),