TF-3157 Update web socket with background service worker

TF-3157 Stub BroadcastChannel for mobile build

lib/features/push_notification/data/datasource_impl/web_socket_datasource_impl.dart

@@ -1,6 +1,7 @@
-import 'dart:convert';
 
+import 'package:core/data/constants/constant.dart';
 import 'package:core/utils/app_logger.dart';
+import 'package:core/utils/broadcast_channel/broadcast_channel.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/core/capability/websocket_capability.dart';
@@ -8,17 +9,21 @@ import 'package:jmap_dart_client/jmap/core/session/session.dart';
 import 'package:model/extensions/session_extension.dart';
 import 'package:rxdart/transformers.dart';
 import 'package:tmail_ui_user/features/push_notification/data/datasource/web_socket_datasource.dart';
+import 'package:tmail_ui_user/features/push_notification/data/model/connect_web_socket_message.dart';
 import 'package:tmail_ui_user/features/push_notification/data/network/web_socket_api.dart';
 import 'package:tmail_ui_user/features/push_notification/domain/exceptions/web_socket_exceptions.dart';
+import 'package:tmail_ui_user/features/push_notification/domain/model/web_socket_action.dart';
 import 'package:tmail_ui_user/main/error/capability_validator.dart';
 import 'package:tmail_ui_user/main/exceptions/exception_thrower.dart';
-import 'package:web_socket_channel/web_socket_channel.dart';
+import 'package:universal_html/html.dart';
 
-class RemoteWebSocketDatasourceImpl implements WebSocketDatasource {
+class WebSocketDatasourceImpl implements WebSocketDatasource {
   final WebSocketApi _webSocketApi;
   final ExceptionThrower _exceptionThrower;
   
-  const RemoteWebSocketDatasourceImpl(this._webSocketApi, this._exceptionThrower);
+  const WebSocketDatasourceImpl(this._webSocketApi, this._exceptionThrower);
+
+  static const String _webSocketClosed = 'webSocketClosed';
 
   @override
   Stream getWebSocketChannel(Session session, AccountId accountId) {
@@ -30,32 +35,31 @@ class RemoteWebSocketDatasourceImpl implements WebSocketDatasource {
   Stream _getWebSocketChannel(
     Session session,
     AccountId accountId,
-    [int retryRemained = 3]
-    ) async* {
+  ) async* {
+    final broadcastChannel = BroadcastChannel(Constant.wsServiceWorkerBroadcastChannel);
     try {
       _verifyWebSocketCapabilities(session, accountId);
       final webSocketTicket = await _webSocketApi.getWebSocketTicket(session, accountId);
       final webSocketUri = _getWebSocketUri(session, accountId);
+      window.navigator.serviceWorker?.controller?.postMessage(ConnectWebSocketMessage(
+        webSocketAction: WebSocketAction.connect,
+        webSocketUrl: webSocketUri.toString(),
+        webSocketTicket: webSocketTicket
+      ).toJson());
       
-      final webSocketChannel = WebSocketChannel.connect(
-        Uri.parse('$webSocketUri?ticket=$webSocketTicket'));
-      await webSocketChannel.ready;
-      webSocketChannel.sink.add(jsonEncode({"@type": "WebSocketPushEnable"}));
-      
-      yield* webSocketChannel.stream;
+      yield* _webSocketListener(broadcastChannel);
     } catch (e) {
       logError('RemoteWebSocketDatasourceImpl::getWebSocketChannel():error: $e');
-      if (retryRemained > 0) {
-        yield* _getWebSocketChannel(session, accountId, retryRemained - 1);
-      } else {
-        rethrow;
-      }
+      rethrow;
     }
   }
 
   void _verifyWebSocketCapabilities(Session session, AccountId accountId) {
     if (!CapabilityIdentifier.jmapWebSocket.isSupported(session, accountId)
       || !CapabilityIdentifier.jmapWebSocketTicket.isSupported(session, accountId)
+      || session.getCapabilityProperties<WebSocketCapability>(
+        accountId,
+        CapabilityIdentifier.jmapWebSocket)?.supportsPush != true
     ) {
       throw WebSocketPushNotSupportedException();
     }
@@ -73,4 +77,14 @@ class RemoteWebSocketDatasourceImpl implements WebSocketDatasource {
 
     return webSocketUri;
   }
+
+  Stream _webSocketListener(BroadcastChannel broadcastChannel) {
+    return broadcastChannel.onMessage.map((event) {
+      if (event.data == _webSocketClosed) {
+        throw WebSocketClosedException();
+      }
+
+      return event.data;
+    });
+  }
 }

lib/features/push_notification/data/model/connect_web_socket_message.dart

@@ -0,0 +1,28 @@
+import 'package:equatable/equatable.dart';
+import 'package:json_annotation/json_annotation.dart';
+import 'package:tmail_ui_user/features/push_notification/domain/model/web_socket_action.dart';
+
+part 'connect_web_socket_message.g.dart';
+
+@JsonSerializable()
+class ConnectWebSocketMessage with EquatableMixin {
+  @JsonKey(name: 'action')
+  final WebSocketAction webSocketAction;
+  @JsonKey(name: 'url')
+  final String webSocketUrl;
+  @JsonKey(name: 'ticket')
+  final String webSocketTicket;
+
+  ConnectWebSocketMessage({
+    required this.webSocketAction,
+    required this.webSocketUrl,
+    required this.webSocketTicket,
+  });
+
+  factory ConnectWebSocketMessage.fromJson(Map<String, dynamic> json) 
+    => _$ConnectWebSocketMessageFromJson(json);
+  Map<String, dynamic> toJson() => _$ConnectWebSocketMessageToJson(this);
+  
+  @override
+  List<Object?> get props => [webSocketAction, webSocketUrl, webSocketTicket];
+}

lib/features/push_notification/data/model/web_socket_echo.dart

@@ -0,0 +1,31 @@
+import 'package:json_annotation/json_annotation.dart';
+
+part 'web_socket_echo.g.dart';
+
+@JsonSerializable(includeIfNull: false)
+class WebSocketEcho {
+  @JsonKey(name: '@type')
+  final String? type;
+  final String? requestId;
+  final List<List<dynamic>>? methodResponses;
+
+  WebSocketEcho({
+    this.type,
+    this.requestId,
+    this.methodResponses,
+  });
+
+  factory WebSocketEcho.fromJson(Map<String, dynamic> json) => _$WebSocketEchoFromJson(json);
+
+  Map<String, dynamic> toJson() => _$WebSocketEchoToJson(this);
+
+  static bool isValid(Map<String, dynamic> json) {
+    try {
+      final webSocketEcho = WebSocketEcho.fromJson(json);
+      final listResponses = webSocketEcho.methodResponses?.firstOrNull;
+      return listResponses?.contains('Core/echo') ?? false;
+    } catch (_) {
+      return false;
+    }
+  }
+}