A

frontend/pweb/lib/pages/loaders/account.dart

@@ -1,9 +1,8 @@
 import 'package:flutter/material.dart';
 
-import 'package:logging/logging.dart';
-
 import 'package:provider/provider.dart';
 
+import 'package:pshared/models/auth/state.dart';
 import 'package:pshared/provider/account.dart';
 
 import 'package:pweb/app/router/pages.dart';
@@ -12,57 +11,63 @@ import 'package:pweb/widgets/error/snackbar.dart';
 import 'package:pweb/generated/i18n/app_localizations.dart';
 
 
-class AccountLoader extends StatelessWidget {
+class AccountLoader extends StatefulWidget {
   final Widget child;
   const AccountLoader({super.key, required this.child});
 
-  static final _logger = Logger('loader.account');
+  @override
+  State<AccountLoader> createState() => _AccountLoaderState();
+}
 
-  void _notifyErrorAndRedirect(BuildContext context, Object error) {
+class _AccountLoaderState extends State<AccountLoader> {
+  AuthState? _handledState;
+
+  @override
+  void initState() {
+    super.initState();
     WidgetsBinding.instance.addPostFrameCallback((_) {
-      if (!context.mounted) return;
-      postNotifyUserOfErrorX(
-        context: context,
-        errorSituation: AppLocalizations.of(context)!.errorLogin,
-        exception: error,
-      );
-      navigateAndReplace(context, Pages.login);
+      if (!mounted) return;
+      Provider.of<AccountProvider>(context, listen: false).restoreIfPossible();
     });
   }
 
-  void _restoreAndHandleResult(BuildContext context, AccountProvider provider) {
-    WidgetsBinding.instance.addPostFrameCallback((_) async {
-      try {
-        await provider.restoreIfPossible();
-      } catch (error, stack) {
-        _logger.warning('Account restore failed: $error', error, stack);
-      }
-      if (!context.mounted) return;
-      final latest = Provider.of<AccountProvider>(context, listen: false);
-      if (latest.account != null) return;
-      if (latest.error != null) {
-        _notifyErrorAndRedirect(context, latest.error!);
-        return;
-      }
-      if (!latest.isLoading) {
-        navigateAndReplace(context, Pages.login);
-      }
-    });
+  void _handleSideEffects(AccountProvider provider) {
+    if (_handledState == provider.authState) return;
+    _handledState = provider.authState;
+
+    void goToLogin() {
+      if (!mounted) return;
+      navigateAndReplace(context, Pages.login);
+    }
+
+    switch (provider.authState) {
+      case AuthState.error:
+        final error = provider.error ?? Exception('Authorization failed');
+        WidgetsBinding.instance.addPostFrameCallback((_) {
+          if (!mounted) return;
+          postNotifyUserOfErrorX(
+            context: context,
+            errorSituation: AppLocalizations.of(context)!.errorLogin,
+            exception: error,
+          );
+          goToLogin();
+        });
+        break;
+      case AuthState.empty:
+        WidgetsBinding.instance.addPostFrameCallback((_) => goToLogin());
+        break;
+      default:
+        break;
+    }
   }
 
   @override
   Widget build(BuildContext context) {
     return Consumer<AccountProvider>(builder: (context, provider, _) {
-      if (provider.account != null) return child;
-      if (provider.error != null) {
-        _notifyErrorAndRedirect(context, provider.error!);
-        return const Center(child: CircularProgressIndicator());
+      _handleSideEffects(provider);
+      if (provider.authState == AuthState.ready && provider.account != null) {
+        return widget.child;
       }
-      if (!provider.isLoading) {
-        _restoreAndHandleResult(context, provider);
-      }
-      if (provider.isLoading) return const Center(child: CircularProgressIndicator());
-      // In case of error or missing account we show loader while side effects handle navigation.
       return const Center(child: CircularProgressIndicator());
     });
   }

frontend/pweb/lib/pages/loaders/organization.dart

@@ -4,9 +4,6 @@ import 'package:provider/provider.dart';
 
 import 'package:pshared/provider/organizations.dart';
 
-import 'package:pweb/app/router/pages.dart';
-import 'package:pweb/widgets/error/snackbar.dart';
-
 import 'package:pweb/generated/i18n/app_localizations.dart';
 
 
@@ -19,12 +16,20 @@ class OrganizationLoader extends StatelessWidget {
   Widget build(BuildContext context) => Consumer<OrganizationsProvider>(builder: (context, provider, _) {
     if (provider.isLoading) return const Center(child: CircularProgressIndicator());
     if (provider.error != null) {
-      postNotifyUserOfErrorX(
-        context: context, 
-        errorSituation: AppLocalizations.of(context)!.errorLogin, 
-        exception: provider.error!,
+      final loc = AppLocalizations.of(context)!;
+      return Center(
+        child: Column(
+          mainAxisSize: MainAxisSize.min,
+          children: [
+            Text(loc.errorLogin),
+            const SizedBox(height: 12),
+            ElevatedButton(
+              onPressed: provider.load,
+              child: Text(loc.retry),
+            ),
+          ],
+        ),
       );
-      navigateAndReplace(context, Pages.login);
     }
     if ((provider.error == null) && (!provider.isOrganizationSet)) {
       WidgetsBinding.instance.addPostFrameCallback((_) {

frontend/pweb/lib/pages/loaders/permissions.dart

@@ -5,9 +5,6 @@ import 'package:provider/provider.dart';
 import 'package:pshared/provider/account.dart';
 import 'package:pshared/provider/permissions.dart';
 
-import 'package:pweb/app/router/pages.dart';
-import 'package:pweb/widgets/error/snackbar.dart';
-
 import 'package:pweb/generated/i18n/app_localizations.dart';
 
 
@@ -15,26 +12,38 @@ class PermissionsLoader extends StatelessWidget {
   final Widget child;
   const PermissionsLoader({super.key, required this.child});
 
-  void _notifyError(BuildContext context, Exception exception) {
-    WidgetsBinding.instance.addPostFrameCallback((_) {
-      postNotifyUserOfErrorX(
-        context: context,
-        errorSituation: AppLocalizations.of(context)!.errorLogin,
-        exception: exception,
-      );
-      navigateAndReplace(context, Pages.login);
-    });
+  void _triggerLoadIfNeeded(PermissionsProvider provider) {
+    if (!provider.isLoading && !provider.isReady) {
+      WidgetsBinding.instance.addPostFrameCallback((_) {
+        if (!provider.isLoading && !provider.isReady) {
+          provider.load();
+        }
+      });
+    }
   }
 
   @override
   Widget build(BuildContext context) {
     return Consumer2<PermissionsProvider, AccountProvider>(
-      builder: (context, provider, accountProvider, _) {
-        if (provider.isLoading) {
-          return const Center(child: CircularProgressIndicator());
-        }
+      builder: (context, provider, _accountProvider, _) {
         if (provider.error != null) {
-          _notifyError(context, provider.error!);
+          final loc = AppLocalizations.of(context)!;
+          return Center(
+            child: Column(
+              mainAxisSize: MainAxisSize.min,
+              children: [
+                Text(loc.errorLogin),
+                const SizedBox(height: 12),
+                ElevatedButton(
+                  onPressed: provider.load,
+                  child: Text(loc.retry),
+                ),
+              ],
+            ),
+          );
+        }
+        _triggerLoadIfNeeded(provider);
+        if (provider.isLoading || !provider.isReady) {
           return const Center(child: CircularProgressIndicator());
         }
         return child;

frontend/pweb/lib/providers/two_factor.dart

@@ -17,6 +17,7 @@ class TwoFactorProvider extends ChangeNotifier {
   bool _hasError = false;
   bool _verificationSuccess = false;
   String? _errorMessage;
+  String? _currentPendingToken;
 
   bool get isSubmitting => _isSubmitting;
   bool get hasError => _hasError;
@@ -26,6 +27,12 @@ class TwoFactorProvider extends ChangeNotifier {
 
   void update(AccountProvider accountProvider) {
     _accountProvider = accountProvider;
+    final pending = accountProvider.pendingLogin;
+    final token = pending?.pendingToken.token;
+    if (token != _currentPendingToken || accountProvider.account == null) {
+      _resetState();
+      _currentPendingToken = token;
+    }
   }
 
   Future<void> submitCode(String code) async {
@@ -46,6 +53,7 @@ class TwoFactorProvider extends ChangeNotifier {
       );
       _accountProvider.completePendingLogin(account);
       _verificationSuccess = true;
+      _currentPendingToken = null;
     } catch (e) {
       _hasError = true;
       _errorMessage = e.toString();
@@ -71,4 +79,17 @@ class TwoFactorProvider extends ChangeNotifier {
       notifyListeners();
     }
   }
-}
+
+  void reset() {
+    _resetState();
+    _currentPendingToken = null;
+  }
+
+  void _resetState() {
+    _isSubmitting = false;
+    _hasError = false;
+    _errorMessage = null;
+    _verificationSuccess = false;
+    notifyListeners();
+  }
+}

frontend/pweb/lib/utils/logout.dart

@@ -1,15 +1,31 @@
 import 'package:flutter/material.dart';
 
+import 'package:go_router/go_router.dart';
 import 'package:provider/provider.dart';
 
 import 'package:pshared/provider/account.dart';
+import 'package:pshared/provider/organizations.dart';
 import 'package:pshared/provider/permissions.dart';
 
 import 'package:pweb/app/router/pages.dart';
+import 'package:pweb/providers/two_factor.dart';
 
 
-void logoutUtil(BuildContext context) {
-  context.read<AccountProvider>().logout();
-  context.read<PermissionsProvider>().reset();
-  navigateAndReplace(context, Pages.login);
+Future<void> logoutUtil(BuildContext context) async {
+  final accountProvider = context.read<AccountProvider>();
+  final permissionsProvider = context.read<PermissionsProvider>();
+  final organizationsProvider = context.read<OrganizationsProvider>();
+  final twoFactorProvider = context.read<TwoFactorProvider>();
+  await accountProvider.logout();
+  permissionsProvider.reset();
+  await organizationsProvider.reset();
+  twoFactorProvider.reset();
+
+  final router = GoRouter.of(context);
+  final loginPath = routerPage(Pages.login);
+
+  WidgetsBinding.instance.addPostFrameCallback((_) {
+    if (!context.mounted) return;
+    router.go(loginPath);
+  });
 }

frontend/pweb/lib/widgets/appbar/app_bar.dart

@@ -17,7 +17,7 @@ class PayoutAppBar extends StatelessWidget implements PreferredSizeWidget {
   final Widget title;
   final VoidCallback onAddFundsPressed;
   final List<Widget>? actions;
-  final VoidCallback? onLogout;
+  final Future<void> Function()? onLogout;
   final String? avatarUrl;
 
   @override

frontend/pweb/lib/widgets/appbar/profile.dart

@@ -7,7 +7,7 @@ class ProfileAvatar extends StatelessWidget {
   const ProfileAvatar({super.key, this.avatarUrl, this.onLogout});
 
   final String? avatarUrl;
-  final VoidCallback? onLogout;
+  final Future<void> Function()? onLogout;
 
   @override
   Widget build(BuildContext context) => PopupMenuButton<int>(
@@ -37,4 +37,4 @@ class ProfileAvatar extends StatelessWidget {
       child: avatarUrl == null ? const Icon(Icons.person, size: 24) : null,
     ),
   );
-}
+}

frontend/pweb/lib/widgets/drawer/tiles/logout.dart

@@ -1,10 +1,6 @@
 import 'package:flutter/material.dart';
 
-import 'package:provider/provider.dart';
-
-import 'package:pshared/provider/account.dart';
-
-import 'package:pweb/app/router/pages.dart';
+import 'package:pweb/utils/logout.dart';
 
 import 'package:pweb/generated/i18n/app_localizations.dart';
 
@@ -23,10 +19,8 @@ class LogoutTile extends StatelessWidget {
     );
   }
 
-  void _logout(BuildContext context) {
+  Future<void> _logout(BuildContext context) async {
     Navigator.pop(context); 
-    final accountProvider = Provider.of<AccountProvider>(context, listen: false);
-    accountProvider.logout();
-    navigateAndReplace(context, Pages.login);
+    await logoutUtil(context);
   }
 }