TF-3157 Remove unnecessary background service worker

lib/features/push_notification/presentation/controller/push_base_controller.dart

@@ -19,10 +19,8 @@ abstract class PushBaseController {
   Session? session;
   AccountId? accountId;
 
-  StreamSubscription<Either<Failure, Success>>? _stateStreamSubscription;
-
   void consumeState(Stream<Either<Failure, Success>> newStateStream) {
-    _stateStreamSubscription = newStateStream.listen(
+    newStateStream.listen(
       _handleStateStream,
       onError: handleErrorViewState,
     );
@@ -40,11 +38,6 @@ abstract class PushBaseController {
     logError('PushBaseController::handleErrorViewState():error: $error | stackTrace: $stackTrace');
   }
 
-  void cancelStateStreamSubscription() {
-    _stateStreamSubscription?.cancel();
-    _stateStreamSubscription = null;
-  }
-
   void initialize({AccountId? accountId, Session? session}) {
     this.accountId = accountId;
     this.session = session;

lib/features/push_notification/presentation/controller/web_socket_controller.dart

@@ -1,9 +1,15 @@
+import 'dart:async';
+import 'dart:convert';
+
 import 'package:core/presentation/state/failure.dart';
 import 'package:core/presentation/state/success.dart';
 import 'package:core/utils/app_logger.dart';
+import 'package:fcm/model/type_name.dart';
 import 'package:jmap_dart_client/jmap/account_id.dart';
 import 'package:jmap_dart_client/jmap/core/session/session.dart';
-import 'package:tmail_ui_user/features/push_notification/domain/exceptions/web_socket_exceptions.dart';
+import 'package:jmap_dart_client/jmap/push/state_change.dart';
+import 'package:tmail_ui_user/features/push_notification/data/model/web_socket_echo_request.dart';
+import 'package:tmail_ui_user/features/push_notification/data/model/web_socket_push_enable_request.dart';
 import 'package:tmail_ui_user/features/push_notification/domain/state/web_socket_push_state.dart';
 import 'package:tmail_ui_user/features/push_notification/domain/usecases/connect_web_socket_interactor.dart';
 import 'package:tmail_ui_user/features/push_notification/presentation/controller/push_base_controller.dart';
@@ -11,6 +17,7 @@ import 'package:tmail_ui_user/features/push_notification/presentation/extensions
 import 'package:tmail_ui_user/features/push_notification/presentation/listener/email_change_listener.dart';
 import 'package:tmail_ui_user/features/push_notification/presentation/listener/mailbox_change_listener.dart';
 import 'package:tmail_ui_user/main/routes/route_navigation.dart';
+import 'package:web_socket_channel/web_socket_channel.dart';
 
 class WebSocketController extends PushBaseController {
   WebSocketController._internal();
@@ -22,11 +29,16 @@ class WebSocketController extends PushBaseController {
   ConnectWebSocketInteractor? _connectWebSocketInteractor;
 
   int _retryRemained = 3;
+  WebSocketChannel? _webSocketChannel;
+  Timer? _webSocketPingTimer;
+  StreamSubscription? _webSocketSubscription;
 
   @override
   void handleFailureViewState(Failure failure) {
     logError('WebSocketController::handleFailureViewState():Failure $failure');
-    cancelStateStreamSubscription();
+    _webSocketSubscription?.cancel();
+    _webSocketChannel = null;
+    _webSocketPingTimer?.cancel();
     if (failure is WebSocketConnectionFailed) {
       _handleWebSocketConnectionRetry();
     }
@@ -35,18 +47,15 @@ class WebSocketController extends PushBaseController {
   @override
   void handleSuccessViewState(Success success) {
     log('WebSocketController::handleSuccessViewState():Success $success');
-    if (success is WebSocketPushStateReceived) {
-      _handleWebSocketPushStateReceived(success);
+    if (success is WebSocketConnectionSuccess) {
+      _handleWebSocketConnectionSuccess(success);
     }
   }
 
   @override
   void handleErrorViewState(Object error, StackTrace stackTrace) {
     super.handleErrorViewState(error, stackTrace);
-    cancelStateStreamSubscription();
-    if (error is WebSocketClosedException) {
-      _handleWebSocketConnectionRetry();
-    }
+    handleFailureViewState(WebSocketConnectionFailed());
   }
 
   @override
@@ -65,26 +74,69 @@ class WebSocketController extends PushBaseController {
     consumeState(_connectWebSocketInteractor!.execute(session, accountId));
   }
   
-  void _handleWebSocketPushStateReceived(WebSocketPushStateReceived success) {
-    log('WebSocketController::_handleWebSocketPushStateReceived(): $success');
+  void _handleWebSocketConnectionSuccess(WebSocketConnectionSuccess success) {
+    log('WebSocketController::_handleWebSocketConnectionSuccess(): $success');
     _retryRemained = 3;
-    if (accountId == null || session == null) return;
-    final stateChange = success.stateChange;
-    if (stateChange == null) return;
-    final mapTypeState = stateChange.getMapTypeState(accountId!);
-    mappingTypeStateToAction(
-      mapTypeState,
-      accountId!,
-      emailChangeListener: EmailChangeListener.instance,
-      mailboxChangeListener: MailboxChangeListener.instance,
-      session!.username,
-      session: session);
+    _webSocketChannel = success.webSocketChannel;
+    _enableWebSocketPush();
+    _pingWebSocket();
+    _listenToWebSocket();
   }
 
   void _handleWebSocketConnectionRetry() {
+    _webSocketSubscription?.cancel();
+    _webSocketChannel = null;
+    _webSocketPingTimer?.cancel();
     if (_retryRemained > 0) {
       _retryRemained--;
       _connectWebSocket(accountId, session);
     }
   }
+
+  void _enableWebSocketPush() {
+    _webSocketChannel?.sink.add(jsonEncode(WebSocketPushEnableRequest.toJson(
+      dataTypes: [TypeName.emailType, TypeName.mailboxType]
+    )));
+  }
+
+  void _pingWebSocket() {
+    _webSocketPingTimer = Timer.periodic(const Duration(seconds: 10), (_) {
+      _webSocketChannel?.sink.add(jsonEncode(WebSocketEchoRequest().toJson()));
+    });
+  }
+
+  void _listenToWebSocket() {
+    _webSocketSubscription = _webSocketChannel?.stream.listen(
+      (data) {
+        log('WebSocketController::_listenToWebSocket(): $data');
+        if (session == null || accountId == null) return;
+        if (data is String) {
+          data = jsonDecode(data);
+        }
+
+        try {
+          final stateChange = StateChange.fromJson(data);
+          final mapTypeState = stateChange.getMapTypeState(accountId!);
+          mappingTypeStateToAction(
+            mapTypeState,
+            accountId!,
+            emailChangeListener: EmailChangeListener.instance,
+            mailboxChangeListener: MailboxChangeListener.instance,
+            session!.username,
+            session: session);
+        } catch (e) {
+          logError('WebSocketController::_listenToWebSocket(): Data is not StateChange');
+        }
+      },
+      cancelOnError: true,
+      onError: (error) {
+        logError('WebSocketController::_listenToWebSocket():Error: $error');
+        handleFailureViewState(WebSocketConnectionFailed(exception: error));
+      },
+      onDone: () {
+        log('WebSocketController::_listenToWebSocket():onDone');
+        _handleWebSocketConnectionRetry();
+      },
+    );
+  }
 }

lib/features/push_notification/presentation/services/fcm_receiver.dart

@@ -1,11 +1,7 @@
 import 'package:core/utils/app_logger.dart';
-import 'package:core/utils/broadcast_channel/broadcast_channel.dart';
-import 'package:core/utils/platform_info.dart';
 import 'package:firebase_messaging/firebase_messaging.dart';
 import 'package:tmail_ui_user/features/push_notification/presentation/controller/fcm_message_controller.dart';
 import 'package:tmail_ui_user/features/push_notification/presentation/services/fcm_service.dart';
-import 'package:tmail_ui_user/main/utils/app_config.dart';
-import 'package:universal_html/html.dart' as html show MessageEvent, DomException;
 
 @pragma('vm:entry-point')
 Future<void> handleFirebaseBackgroundMessage(RemoteMessage message) async {
@@ -23,32 +19,11 @@ class FcmReceiver {
 
   static const int MAX_COUNT_RETRY_TO_GET_FCM_TOKEN = 3;
 
-  int _countRetryToGetFcmToken = 0;
-
   Future onInitialFcmListener() async {
-    _countRetryToGetFcmToken = 0;
     _onForegroundMessage();
     _onBackgroundMessage();
 
-    if (PlatformInfo.isWeb) {
-      _onMessageBroadcastChannel();
-      await _requestNotificationPermissionOnWeb();
-    } else {
-      await _onHandleFcmToken();
-    }
-  }
-
-  Future<void> _requestNotificationPermissionOnWeb() async {
-    NotificationSettings notificationSetting = await FirebaseMessaging.instance.getNotificationSettings();
-    log('FcmReceiver::_requestNotificationPermissionOnWeb: authorizationStatus = ${notificationSetting.authorizationStatus}');
-    if (notificationSetting.authorizationStatus != AuthorizationStatus.authorized) {
-      notificationSetting = await FirebaseMessaging.instance.requestPermission();
-      if (notificationSetting.authorizationStatus == AuthorizationStatus.authorized) {
-        await _onHandleFcmToken();
-      }
-    } else {
-      await _onHandleFcmToken();
-    }
+    await _onHandleFcmToken();
   }
 
   void _onForegroundMessage() {
@@ -59,38 +34,17 @@ class FcmReceiver {
     FirebaseMessaging.onBackgroundMessage(handleFirebaseBackgroundMessage);
   }
 
-  void _onMessageBroadcastChannel() {
-    final broadcast = BroadcastChannel('background-message');
-    broadcast.onMessage.listen((event) {
-      if (event is html.MessageEvent) {
-        FcmService.instance.handleMessageEventBroadcastChannel(event);
-      }
-    });
-  }
-
   Future<String?> _getInitialToken() async {
     try {
-      final vapidKey = PlatformInfo.isWeb ? AppConfig.fcmVapidPublicKeyWeb : null;
-      final token = await FirebaseMessaging.instance.getToken(vapidKey: vapidKey);
+      final token = await FirebaseMessaging.instance.getToken();
       log('FcmReceiver::_getInitialToken:token: $token');
       return token;
     } catch (e) {
       logError('FcmReceiver::_getInitialToken: TYPE = ${e.runtimeType} | Exception = $e');
-      if (PlatformInfo.isWeb
-          && e is html.DomException
-          && _countRetryToGetFcmToken < MAX_COUNT_RETRY_TO_GET_FCM_TOKEN) {
-        return await _retryGetToken();
-      }
       return null;
     }
   }
 
-  Future<String?> _retryGetToken() async {
-    _countRetryToGetFcmToken++;
-    log('FcmReceiver::_retryGetToken: CountRetry = $_countRetryToGetFcmToken');
-    return await _getInitialToken();
-  }
-
   Future _onHandleFcmToken() async {
     final token = await _getInitialToken();
     FcmService.instance.handleToken(token);