hamed/minor_refactor

Author: hamed-rezaee

lib/state/connection/connection_cubit.dart

@@ -29,9 +29,7 @@ class ConnectionCubit extends Cubit<ConnectionState> {
 
     _api = Injector()<BaseAPI>();
 
-    _connectionInformation = connectionInformation;
-
-    _connect(_connectionInformation);
+    connect(_connectionInformation = connectionInformation);
   }
 
   final String _key = '${UniqueKey()}';
@@ -60,27 +58,25 @@ class ConnectionCubit extends Cubit<ConnectionState> {
   /// Gets app id of websocket.
   static String get appId => _connectionInformation.appId;
 
-  /// Reconnect to Websocket with new [connectionInformation].
-  Future<void> reconnect({ConnectionInformation? connectionInformation}) async {
-    if (connectionInformation != null) {
-      _connectionInformation = connectionInformation;
-    }
-
-    await _connect(_connectionInformation);
-  }
+  /// Connect to Websocket with new [connectionInformation].
+  Future<void> connect([ConnectionInformation? connectionInformation]) async {
+    try {
+      if (connectionInformation != null) {
+        _connectionInformation = connectionInformation;
+      }
 
-  Future<void> _connect(ConnectionInformation connectionInformation) async {
-    if (state is ConnectionConnectingState) {
-      return;
-    }
+      await _api!.disconnect();
 
-    await _api!.connect(
-      _connectionInformation,
-      printResponse: enableDebug && printResponse,
-      onError: (String error) => emit(ConnectionErrorState(error)),
-    );
+      await _api!.connect(
+        _connectionInformation,
+        printResponse: enableDebug && printResponse,
+        onError: (String error) => emit(ConnectionErrorState(error)),
+      );
 
-    _handleConnectionStatus();
+      _handleConnectionStatus();
+    } on Exception catch (e) {
+      emit(ConnectionErrorState('$e'));
+    }
   }
 
   void _handleConnectionStatus() {