TF-4045 Only show upgrade storage when premium is available and quota is available

lib/features/quotas/presentation/quotas_controller.dart

@@ -1,13 +1,15 @@
 import 'package:core/presentation/state/failure.dart';
 import 'package:core/presentation/state/success.dart';
+import 'package:core/utils/app_logger.dart';
+import 'package:dartz/dartz.dart';
 import 'package:flutter/material.dart';
 import 'package:get/get.dart';
 import 'package:jmap_dart_client/jmap/account_id.dart';
 import 'package:jmap_dart_client/jmap/core/capability/capability_identifier.dart';
-import 'package:jmap_dart_client/jmap/quotas/quota.dart';
 import 'package:tmail_ui_user/features/base/base_controller.dart';
+import 'package:tmail_ui_user/features/home/data/exceptions/session_exceptions.dart';
 import 'package:tmail_ui_user/features/mailbox_dashboard/presentation/controller/mailbox_dashboard_controller.dart';
-import 'package:tmail_ui_user/features/mailbox_dashboard/presentation/extensions/handle_paywall_extension.dart';
+import 'package:tmail_ui_user/features/mailbox_dashboard/presentation/extensions/validate_premium_storage_extension.dart';
 import 'package:tmail_ui_user/features/quotas/domain/extensions/list_quotas_extensions.dart';
 import 'package:tmail_ui_user/features/quotas/domain/state/get_quotas_state.dart';
 import 'package:tmail_ui_user/features/quotas/domain/use_case/get_quotas_interactor.dart';
@@ -19,7 +21,6 @@ class QuotasController extends BaseController {
 
   final GetQuotasInteractor _getQuotasInteractor;
 
-  final octetsQuota = Rxn<Quota>();
   final isBannerEnabled = RxBool(false);
 
   late Worker accountIdListener;
@@ -31,26 +32,39 @@ class QuotasController extends BaseController {
   }
 
   void _handleGetQuotasSuccess(GetQuotasSuccess success) {
-    octetsQuota.value = success.quotas.octetsQuota;
+    log('$runtimeType::_handleGetQuotasSuccess: Quotas is ${success.quotas}');
+    mailboxDashBoardController.octetsQuota.value = success.quotas.octetsQuota;
     isBannerEnabled.value = true;
   }
 
   void _initWorker() {
-    accountIdListener = ever(mailboxDashBoardController.accountId, (accountId) {
-      final session = mailboxDashBoardController.sessionCurrent;
-      if (accountId is AccountId &&
-          session != null &&
-          CapabilityIdentifier.jmapQuota.isSupported(session, accountId)
-      ) {
-        _getQuotasAction(accountId);
-      }
-    });
+    accountIdListener = ever(
+      mailboxDashBoardController.accountId,
+      (_) => reloadQuota(),
+    );
+  }
+
+  bool get isStorageCapabilitySupported {
+    final accountId = mailboxDashBoardController.accountId.value;
+    final session = mailboxDashBoardController.sessionCurrent;
+
+    if (accountId != null && session != null) {
+      return CapabilityIdentifier.jmapQuota.isSupported(session, accountId);
+    } else {
+      return false;
+    }
   }
 
   void reloadQuota() {
-    if (mailboxDashBoardController.accountId.value == null) return;
-    
-    _getQuotasAction(mailboxDashBoardController.accountId.value!);
+    if (isStorageCapabilitySupported) {
+      _getQuotasAction(mailboxDashBoardController.accountId.value!);
+    } else {
+      consumeState(
+        Stream.value(
+          Left(GetQuotasFailure(NotFoundAccountIdException())),
+        ),
+      );
+    }
   }
 
   @override
@@ -77,6 +91,8 @@ class QuotasController extends BaseController {
   @override
   void handleFailureViewState(Failure failure) {
     if (failure is GetQuotasFailure) {
+      logError('$runtimeType::handleFailureViewState');
+      mailboxDashBoardController.octetsQuota.value = null;
       isBannerEnabled.value = true;
     } else {
       super.handleFailureViewState(failure);
@@ -89,7 +105,7 @@ class QuotasController extends BaseController {
   }
 
   void handleManageMyStorage(BuildContext context) {
-    mailboxDashBoardController.navigateToPaywall(context);
+    mailboxDashBoardController.paywallController?.navigateToPaywall(context);
   }
 
   void closeBanner() {

lib/features/quotas/presentation/quotas_view.dart

@@ -17,7 +17,7 @@ class QuotasView extends GetWidget<QuotasController> {
   @override
   Widget build(BuildContext context) {
     return Obx(() {
-      final octetQuota = controller.octetsQuota.value;
+      final octetQuota = controller.mailboxDashBoardController.octetsQuota.value;
       bool isDesktop = controller.responsiveUtils.isDesktop(context);
 
       if (octetQuota != null && octetQuota.isStorageUsageIndicatorAppear) {

lib/features/quotas/presentation/widget/quotas_banner_widget.dart

@@ -18,7 +18,10 @@ class QuotasBannerWidget extends StatelessWidget {
   @override
   Widget build(BuildContext context) {
     return Obx(() {
-      final octetQuota = _quotasController.octetsQuota.value;
+      final octetQuota = _quotasController
+          .mailboxDashBoardController
+          .octetsQuota
+          .value;
 
       if (octetQuota != null &&
           octetQuota.allowedDisplayToQuotaBanner &&