Before code cleanup

Author: Troplo

app/src/main/java/com/troplo/privateuploader/MainScreen.kt

@@ -44,7 +44,7 @@ fun MainScreen(user: User?) {
     val token = SessionManager(context).getAuthToken()
     if (token != null) {
         SocketHandler.initializeSocket(token, context)
-        UserStore.initializeUser(token)
+        UserStore.initializeUser(context)
     }
     val navController = rememberNavController()
     val panelState = rememberOverlappingPanelsState()
@@ -63,8 +63,9 @@ fun MainScreen(user: User?) {
         topBar = {
             if(!SocketHandler.connected.value) {
                 ConnectingBanner()
+            } else {
+                TopBarNav(navController = navController, user = user)
             }
-            TopBarNav(navController =  navController, user = user)
          },
         bottomBar = { BottomBarNav(navController = navController, panelState = panelState, closePanels = closePanelsFunc) },
     ) { paddingValues ->

app/src/main/java/com/troplo/privateuploader/api/SessionManager.kt

@@ -2,7 +2,9 @@ package com.troplo.privateuploader.api
 
 import android.content.Context
 import android.content.SharedPreferences
+import com.google.gson.Gson
 import com.troplo.privateuploader.R
+import com.troplo.privateuploader.data.model.User
 
 class SessionManager (context: Context) {
   private var prefs: SharedPreferences = context.getSharedPreferences(context.getString(R.string.app_name), Context.MODE_PRIVATE)
@@ -36,4 +38,19 @@ class SessionManager (context: Context) {
     editor.putInt("lastChatId", id)
     editor.apply()
   }
+
+  fun setUserCache(user: User?) {
+    val editor = prefs.edit()
+    editor.putString("user", Gson().toJson(user))
+    editor.apply()
+  }
+
+  fun getUserCache(): User? {
+    val user = prefs.getString("user", null)
+    return if(user != null) {
+      Gson().fromJson(user, User::class.java)
+    } else {
+      null
+    }
+  }
 }

app/src/main/java/com/troplo/privateuploader/api/stores/UserStore.kt

@@ -1,5 +1,7 @@
 package com.troplo.privateuploader.api.stores
 
+import android.content.Context
+import com.troplo.privateuploader.api.SessionManager
 import com.troplo.privateuploader.api.TpuApi
 import com.troplo.privateuploader.data.model.User
 import kotlinx.coroutines.CoroutineScope
@@ -11,12 +13,16 @@ import java.net.URISyntaxException
 object UserStore {
   private var user: User? = null
 
-  fun initializeUser(token: String) {
+  fun initializeUser(context: Context) {
     try {
       CoroutineScope(
         Dispatchers.IO
       ).launch {
+        if(SessionManager(context).getUserCache() != null) {
+          user = SessionManager(context).getUserCache()
+        }
         user = TpuApi.retrofitService.getUser().execute().body()
+        SessionManager(context).setUserCache(user)
       }
     } catch (e: URISyntaxException) {
       e.printStackTrace()