🐛 Allow user to search pool with UCO in the pair

archethic-foundation · Feb 6, 2024 · 9f6e568 · 9f6e568
1 parent a084e21
commit 9f6e568
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 8 deletions.
diff --git a/lib/application/pool/dex_pool.g.dart b/lib/application/pool/dex_pool.g.dart
diff --git a/lib/application/pool/dex_pool_list.dart b/lib/application/pool/dex_pool_list.dart
@@ -57,16 +57,27 @@ Future<List<DexPool>> _getPoolListForSearch(
 ) async {
   final dexPools = <DexPool>[];
   final poolList = await ref.read(_getPoolListProvider.future);
+  final poolsListDatasource = await HivePoolsListDatasource.getInstance();
 
   for (final pool in poolList) {
-    if (pool.poolAddress.toUpperCase() == searchText.toUpperCase() ||
-        pool.pair.token1.address!.toUpperCase() == searchText.toUpperCase() ||
-        pool.pair.token2.address!.toUpperCase() == searchText.toUpperCase()) {
-      final poolWithInfos = await ref.read(
-        DexPoolProviders.getPoolInfos(pool).future,
-      );
-      dexPools.add(poolWithInfos!);
+    if ((pool.poolAddress.toUpperCase() == searchText.toUpperCase() ||
+            pool.pair.token1.address!.toUpperCase() ==
+                searchText.toUpperCase() ||
+            pool.pair.token2.address!.toUpperCase() ==
+                searchText.toUpperCase()) ||
+        (searchText.toUpperCase() == 'UCO' &&
+            (pool.pair.token1.isUCO || pool.pair.token2.isUCO))) {
+      final poolHive = poolsListDatasource.getPool(pool.poolAddress);
+      if (poolHive == null) {
+        final poolWithInfos = await ref.read(
+          DexPoolProviders.getPoolInfos(pool).future,
+        );
+        dexPools.add(poolWithInfos!);
+      } else {
+        dexPools.add(poolHive.toDexPool());
+      }
     }
   }
+
   return dexPools;
 }