Added some improvements for states onResume/onPause.

Author: 23rd

TMessagesProj/src/main/java/org/telegram/messenger/LifecycleHelper.java

@@ -0,0 +1,67 @@
+package org.telegram.messenger;
+
+import android.app.Activity;
+import android.os.Handler;
+import android.os.Looper;
+import android.view.View;
+
+/**
+ * Helper class to manage app lifecycle and prevent white screen issues
+ */
+public class LifecycleHelper {
+    private static final String TAG = "LifecycleHelper";
+    private static final long UI_RESTORE_DELAY = 100;
+    
+    private static Handler mainHandler = new Handler(Looper.getMainLooper());
+    
+    /**
+     * Ensures UI components are visible and properly restored
+     */
+    public static void ensureUIVisible(Activity activity, View... views) {
+        if (activity == null || activity.isFinishing()) {
+            return;
+        }
+        
+        mainHandler.post(() -> {
+            try {
+                for (View view : views) {
+                    if (view != null && view.getVisibility() != View.VISIBLE) {
+                        view.setVisibility(View.VISIBLE);
+                    }
+                }
+            } catch (Exception e) {
+                FileLog.e(TAG, e);
+            }
+        });
+    }
+    
+    /**
+     * Delays UI restoration to ensure proper initialization
+     */
+    public static void delayedUIRestore(Runnable restoreAction) {
+        if (restoreAction == null) {
+            return;
+        }
+        
+        mainHandler.postDelayed(() -> {
+            try {
+                restoreAction.run();
+            } catch (Exception e) {
+                FileLog.e(TAG, e);
+            }
+        }, UI_RESTORE_DELAY);
+    }
+    
+    /**
+     * Checks if the app needs to restore its state
+     */
+    public static boolean needsStateRestore(Activity activity) {
+        if (activity == null) {
+            return false;
+        }
+        
+        // Check if main UI components are missing or invisible
+        View contentView = activity.findViewById(android.R.id.content);
+        return contentView == null || contentView.getVisibility() != View.VISIBLE;
+    }
+}

TMessagesProj/src/main/java/org/telegram/ui/LaunchActivity.java

@@ -128,6 +128,7 @@
 import org.telegram.messenger.UserNameResolver;
 import org.telegram.messenger.UserObject;
 import org.telegram.messenger.Utilities;
+import org.telegram.messenger.LifecycleHelper;
 import org.telegram.messenger.browser.Browser;
 import org.telegram.messenger.pip.PipActivityController;
 import org.telegram.messenger.pip.activity.IPipActivity;
@@ -965,6 +966,20 @@ public void onPreviewOpenAnimationEnd() {
         }
         checkLayout();
         checkSystemBarColors();
+        
+        // Handle app restoration after process kill
+        if (savedInstanceState != null) {
+            // Ensure fragments are properly restored
+            if (actionBarLayout.getFragmentStack().isEmpty()) {
+                if (UserConfig.getInstance(currentAccount).isClientActivated()) {
+                    DialogsActivity dialogsActivity = new DialogsActivity(null);
+                    dialogsActivity.setSideMenu(sideMenu);
+                    actionBarLayout.addFragmentToStack(dialogsActivity);
+                    drawerLayoutContainer.setAllowOpenDrawer(true, false);
+                }
+            }
+        }
+        
         handleIntent(getIntent(), false, savedInstanceState != null, false, null, true, true);
         try {
             String os1 = Build.DISPLAY;
@@ -6779,6 +6794,11 @@ protected void onPause() {
         pipActivityHandler.onPause();
         NotificationCenter.getGlobalInstance().postNotificationName(NotificationCenter.stopAllHeavyOperations, 4096);
         ApplicationLoader.mainInterfacePaused = true;
+        
+        // Ensure UI state is properly saved before going to background
+        if (actionBarLayout != null) {
+            actionBarLayout.saveLastFragment();
+        }
         int account = currentAccount;
         Utilities.stageQueue.postRunnable(() -> {
             ApplicationLoader.mainInterfacePausedStageQueue = true;
@@ -6832,6 +6852,21 @@ protected void onStart() {
         if (GroupCallActivity.groupCallInstance != null) {
             GroupCallActivity.groupCallInstance.onResume();
         }
+        
+        // Ensure fragments are properly restored after app comes back from background
+        if (actionBarLayout != null && actionBarLayout.getFragmentStack().isEmpty()) {
+            // If fragment stack is empty, restore default fragment
+            if (UserConfig.getInstance(currentAccount).isClientActivated()) {
+                DialogsActivity dialogsActivity = new DialogsActivity(null);
+                if (sideMenu != null) {
+                    dialogsActivity.setSideMenu(sideMenu);
+                }
+                actionBarLayout.addFragmentToStack(dialogsActivity);
+                if (drawerLayoutContainer != null) {
+                    drawerLayoutContainer.setAllowOpenDrawer(true, false);
+                }
+            }
+        }
     }
 
     @Override
@@ -6972,6 +7007,27 @@ protected void onResume() {
             onResumeStaticCallback.run();
             onResumeStaticCallback = null;
         }
+        
+        // Fix for white screen issue after device sleep
+        LifecycleHelper.ensureUIVisible(this, frameLayout, drawerLayoutContainer);
+        
+        // Delayed UI restore if needed
+        if (LifecycleHelper.needsStateRestore(this)) {
+            LifecycleHelper.delayedUIRestore(() -> {
+                if (actionBarLayout != null && actionBarLayout.getFragmentStack().isEmpty()) {
+                    if (UserConfig.getInstance(currentAccount).isClientActivated()) {
+                        DialogsActivity dialogsActivity = new DialogsActivity(null);
+                        if (sideMenu != null) {
+                            dialogsActivity.setSideMenu(sideMenu);
+                        }
+                        actionBarLayout.addFragmentToStack(dialogsActivity);
+                        if (drawerLayoutContainer != null) {
+                            drawerLayoutContainer.setAllowOpenDrawer(true, false);
+                        }
+                    }
+                }
+            });
+        }
         if (Theme.selectedAutoNightType == Theme.AUTO_NIGHT_TYPE_SYSTEM) {
             Theme.checkAutoNightThemeConditions();
         }
@@ -8193,6 +8249,12 @@ public void hideVisibleActionMode() {
     protected void onSaveInstanceState(Bundle outState) {
         try {
             super.onSaveInstanceState(outState);
+            
+            // Save current account and app state
+            outState.putInt("currentAccount", currentAccount);
+            outState.putBoolean("isActive", isActive);
+            outState.putBoolean("isResumed", isResumed);
+            outState.putBoolean("isStarted", isStarted);
             BaseFragment lastFragment = null;
             if (AndroidUtilities.isTablet()) {
                 if (layersActionBarLayout != null && !layersActionBarLayout.getFragmentStack().isEmpty()) {
@@ -8236,6 +8298,35 @@ protected void onSaveInstanceState(Bundle outState) {
             FileLog.e(e);
         }
     }
+    
+    @Override
+    protected void onRestoreInstanceState(Bundle savedInstanceState) {
+        try {
+            super.onRestoreInstanceState(savedInstanceState);
+            
+            // Restore app state
+            if (savedInstanceState != null) {
+                int savedAccount = savedInstanceState.getInt("currentAccount", currentAccount);
+                if (savedAccount != currentAccount && UserConfig.isValidAccount(savedAccount)) {
+                    switchToAccount(savedAccount, false);
+                }
+                
+                isActive = savedInstanceState.getBoolean("isActive", true);
+                isResumed = savedInstanceState.getBoolean("isResumed", false);
+                isStarted = savedInstanceState.getBoolean("isStarted", false);
+                
+                // Ensure UI is visible after restore
+                if (frameLayout != null) {
+                    frameLayout.setVisibility(View.VISIBLE);
+                }
+                if (drawerLayoutContainer != null) {
+                    drawerLayoutContainer.setVisibility(View.VISIBLE);
+                }
+            }
+        } catch (Exception e) {
+            FileLog.e(e);
+        }
+    }
 
     @Override
     public void onBackPressed() {