feat: added WebSocket support in AtLookup

packages/at_lookup/lib/at_lookup.dart

@@ -6,9 +6,12 @@ library at_lookup;
 export 'src/at_lookup.dart';
 export 'src/at_lookup_impl.dart';
 export 'src/connection/outbound_connection.dart';
+export 'src/connection/outbound_websocket_connection.dart';
 export 'src/connection/outbound_connection_impl.dart';
 export 'src/exception/at_lookup_exception.dart';
 export 'src/monitor_client.dart';
 export 'src/cache/secondary_address_finder.dart';
 export 'src/cache/cacheable_secondary_address_finder.dart';
 export 'src/util/secure_socket_util.dart';
+export 'src/connection/outbound_websocket_connection_impl.dart';
+export 'src/connection/at_lookup_outbound_connection_factory.dart';

packages/at_lookup/lib/src/at_lookup_impl.dart

@@ -5,25 +5,38 @@ import 'dart:convert';
 import 'dart:io';
 import 'dart:typed_data';
 
+import 'package:at_chops/at_chops.dart';
 import 'package:at_commons/at_builders.dart';
 import 'package:at_commons/at_commons.dart';
 import 'package:at_lookup/at_lookup.dart';
+import 'package:at_lookup/src/connection/at_connection.dart';
 import 'package:at_lookup/src/connection/outbound_message_listener.dart';
 import 'package:at_utils/at_logger.dart';
 import 'package:crypto/crypto.dart';
 import 'package:crypton/crypton.dart';
 import 'package:mutex/mutex.dart';
-import 'package:at_chops/at_chops.dart';
 
 class AtLookupImpl implements AtLookUp {
   final logger = AtSignLogger('AtLookup');
 
   /// Listener for reading verb responses from the remote server
   late OutboundMessageListener messageListener;
 
-  OutboundConnection? _connection;
+  AtConnection?
+      _connection; // Single variable for both socket and WebSocket connections
 
-  OutboundConnection? get connection => _connection;
+  AtConnection? get connection => _connection;
+
+  /// Setters or methods to initialize the connection
+  void setSocketConnection(OutboundConnection socketConnection) {
+    _connection = socketConnection;
+  }
+
+  void setWebSocketConnection(OutboundWebSocketConnection webSocketConnection) {
+    _connection = webSocketConnection;
+  }
+
+  late AtLookupOutboundConnectionFactory atOutboundConnectionFactory;
 
   @override
   late SecondaryAddressFinder secondaryAddressFinder;
@@ -44,12 +57,6 @@ class AtLookupImpl implements AtLookUp {
 
   late SecureSocketConfig _secureSocketConfig;
 
-  late final AtLookupSecureSocketFactory socketFactory;
-
-  late final AtLookupSecureSocketListenerFactory socketListenerFactory;
-
-  late AtLookupOutboundConnectionFactory outboundConnectionFactory;
-
   /// Represents the client configurations.
   late Map<String, dynamic> _clientConfig;
 
@@ -61,9 +68,10 @@ class AtLookupImpl implements AtLookUp {
       SecondaryAddressFinder? secondaryAddressFinder,
       SecureSocketConfig? secureSocketConfig,
       Map<String, dynamic>? clientConfig,
-      AtLookupSecureSocketFactory? secureSocketFactory,
-      AtLookupSecureSocketListenerFactory? socketListenerFactory,
-      AtLookupOutboundConnectionFactory? outboundConnectionFactory}) {
+      AtLookupOutboundConnectionFactory? atOutboundConnectionFactory}) {
+    // Default to secure socket factory
+    this.atOutboundConnectionFactory =
+        atOutboundConnectionFactory ?? AtLookupSecureSocketFactory();
     _currentAtSign = atSign;
     _rootDomain = rootDomain;
     _rootPort = rootPort;
@@ -73,11 +81,6 @@ class AtLookupImpl implements AtLookUp {
     // Stores the client configurations.
     // If client configurations are not available, defaults to empty map
     _clientConfig = clientConfig ?? {};
-    socketFactory = secureSocketFactory ?? AtLookupSecureSocketFactory();
-    this.socketListenerFactory =
-        socketListenerFactory ?? AtLookupSecureSocketListenerFactory();
-    this.outboundConnectionFactory =
-        outboundConnectionFactory ?? AtLookupOutboundConnectionFactory();
   }
 
   @Deprecated('use CacheableSecondaryAddressFinder')
@@ -240,22 +243,24 @@ class AtLookupImpl implements AtLookUp {
 
   Future<void> createConnection() async {
     if (!isConnectionAvailable()) {
-      if (_connection != null) {
+      if (connection != null) {
         // Clean up the connection before creating a new one
         logger.finer('Closing old connection');
-        await _connection!.close();
+        await connection!.close();
       }
       logger.info('Creating new connection');
-      //1. find secondary url for atsign from lookup library
+
+      // 1. Find secondary URL for the atsign from the lookup library
       SecondaryAddress secondaryAddress =
           await secondaryAddressFinder.findSecondary(_currentAtSign);
       var host = secondaryAddress.host;
       var port = secondaryAddress.port;
-      //2. create a connection to secondary server
-      await createOutBoundConnection(
-          host, port.toString(), _currentAtSign, _secureSocketConfig);
-      //3. listen to server response
-      messageListener = socketListenerFactory.createListener(_connection!);
+
+      // 2. Create a connection to the secondary server
+      await createOutboundConnection(
+          host, port.toString(), _secureSocketConfig);
+
+      // 3. Listen to server response
       messageListener.listen();
       logger.info('New connection created OK');
     }
@@ -436,7 +441,7 @@ class AtLookupImpl implements AtLookUp {
     await createConnection();
     try {
       await _pkamAuthenticationMutex.acquire();
-      if (!_connection!.getMetaData()!.isAuthenticated) {
+      if (!connection!.getMetaData()!.isAuthenticated) {
         await _sendCommand((FromVerbBuilder()
               ..atSign = _currentAtSign
               ..clientConfig = _clientConfig)
@@ -458,13 +463,13 @@ class AtLookupImpl implements AtLookUp {
         var pkamResponse = await messageListener.read();
         if (pkamResponse == 'data:success') {
           logger.info('auth success');
-          _connection!.getMetaData()!.isAuthenticated = true;
+          connection!.getMetaData()!.isAuthenticated = true;
         } else {
           throw UnAuthenticatedException(
               'Failed connecting to $_currentAtSign. $pkamResponse');
         }
       }
-      return _connection!.getMetaData()!.isAuthenticated;
+      return connection!.getMetaData()!.isAuthenticated;
     } finally {
       _pkamAuthenticationMutex.release();
     }
@@ -524,31 +529,40 @@ class AtLookupImpl implements AtLookUp {
     await createConnection();
     try {
       await _cramAuthenticationMutex.acquire();
+
       if (!_connection!.getMetaData()!.isAuthenticated) {
+        // Use the connection and message listener dynamically
         await _sendCommand((FromVerbBuilder()
               ..atSign = _currentAtSign
               ..clientConfig = _clientConfig)
             .buildCommand());
+
         var fromResponse = await messageListener.read(
             transientWaitTimeMillis: 4000, maxWaitMilliSeconds: 10000);
         logger.info('from result:$fromResponse');
+
         if (fromResponse.isEmpty) {
           return false;
         }
+
         fromResponse = fromResponse.trim().replaceAll('data:', '');
+
         var digestInput = '$secret$fromResponse';
         var bytes = utf8.encode(digestInput);
         var digest = sha512.convert(bytes);
+
         await _sendCommand('cram:$digest\n');
         var cramResponse = await messageListener.read(
             transientWaitTimeMillis: 4000, maxWaitMilliSeconds: 10000);
+
         if (cramResponse == 'data:success') {
           logger.info('auth success');
           _connection!.getMetaData()!.isAuthenticated = true;
         } else {
           throw UnAuthenticatedException('Auth failed');
         }
       }
+
       return _connection!.getMetaData()!.isAuthenticated;
     } finally {
       _cramAuthenticationMutex.release();
@@ -628,19 +642,33 @@ class AtLookupImpl implements AtLookUp {
         !(_connection!.getMetaData()!.isAuthenticated);
   }
 
-  Future<bool> createOutBoundConnection(String host, String port,
-      String toAtSign, SecureSocketConfig secureSocketConfig) async {
+  Future<bool> createOutboundConnection(
+      String host, String port, SecureSocketConfig secureSocketConfig) async {
     try {
-      SecureSocket secureSocket =
-          await socketFactory.createSocket(host, port, secureSocketConfig);
-      _connection =
-          outboundConnectionFactory.createOutboundConnection(secureSocket);
+      // Create the socket connection using the factory
+      final underlying = await atOutboundConnectionFactory.createUnderlying(
+          host, port, secureSocketConfig);
+
+      // Create the outbound connection and listener using the factory's methods
+      final outboundConnection =
+          atOutboundConnectionFactory.outBoundConnectionFactory(underlying);
+      messageListener = atOutboundConnectionFactory
+          .atLookupSocketListenerFactory(outboundConnection);
+
+      // Set the connection type in `_webSocketConnection` or `_connection`
+      if (underlying is WebSocket) {
+        _connection = outboundConnection as OutboundWebSocketConnection;
+      } else if (underlying is SecureSocket) {
+        _connection = outboundConnection as OutboundConnection;
+      }
+
+      // Set idle time if applicable
       if (outboundConnectionTimeout != null) {
-        _connection!.setIdleTime(outboundConnectionTimeout);
+        outboundConnection.setIdleTime(outboundConnectionTimeout);
       }
     } on SocketException {
       throw SecondaryConnectException(
-          'unable to connect to secondary $toAtSign on $host:$port');
+          'Unable to connect to secondary $_currentAtSign on $host:$port');
     }
     return true;
   }
@@ -661,7 +689,7 @@ class AtLookupImpl implements AtLookUp {
   Future<void> _sendCommand(String command) async {
     await createConnection();
     logger.finer('SENDING: $command');
-    await _connection!.write(command);
+    _connection!.write(command);
   }
 
   @override
@@ -682,23 +710,3 @@ class AtLookupImpl implements AtLookUp {
   @override
   String? enrollmentId;
 }
-
-class AtLookupSecureSocketFactory {
-  Future<SecureSocket> createSocket(
-      String host, String port, SecureSocketConfig socketConfig) async {
-    return await SecureSocketUtil.createSecureSocket(host, port, socketConfig);
-  }
-}
-
-class AtLookupSecureSocketListenerFactory {
-  OutboundMessageListener createListener(
-      OutboundConnection outboundConnection) {
-    return OutboundMessageListener(outboundConnection);
-  }
-}
-
-class AtLookupOutboundConnectionFactory {
-  OutboundConnection createOutboundConnection(SecureSocket secureSocket) {
-    return OutboundConnectionImpl(secureSocket);
-  }
-}

packages/at_lookup/lib/src/cache/cacheable_secondary_address_finder.dart

@@ -128,10 +128,10 @@ class SecondaryAddressCacheEntry {
 class SecondaryUrlFinder {
   final String _rootDomain;
   final int _rootPort;
-  late final AtLookupSecureSocketFactory _socketFactory;
+  late final AtLookupOutboundConnectionFactory _socketFactory;
 
   SecondaryUrlFinder(this._rootDomain, this._rootPort,
-      {AtLookupSecureSocketFactory? socketFactory}) {
+      {AtLookupOutboundConnectionFactory? socketFactory}) {
     _socketFactory = socketFactory ?? AtLookupSecureSocketFactory();
   }
 
@@ -188,11 +188,11 @@ class SecondaryUrlFinder {
       var prompt = false;
       var once = true;
 
-      socket = await _socketFactory.createSocket(
+      socket = await _socketFactory.createUnderlying(
           _rootDomain, '$_rootPort', SecureSocketConfig());
       _logger.finer('findSecondaryUrl: connection to root server established');
       // listen to the received data event stream
-      socket.listen((List<int> event) async {
+      socket!.listen((List<int> event) async {
         _logger.finest('root socket listener received: $event');
         answer = utf8.decode(event);
 

packages/at_lookup/lib/src/connection/at_connection.dart

@@ -1,14 +1,12 @@
-import 'dart:io';
+abstract class AtConnection<T> {
+  /// The underlying connection
+  T get underlying;
 
-abstract class AtConnection {
   /// Write a data to the underlying socket of the connection
   /// @param - data - Data to write to the socket
   /// @throws [AtIOException] for any exception during the operation
   void write(String data);
 
-  /// Retrieves the socket of underlying connection
-  Socket getSocket();
-
   /// closes the underlying connection
   Future<void> close();