From 54a872b49668c50ab6235b27722ff7a2d24baa1a Mon Sep 17 00:00:00 2001
From: Nikola Pavlov <nikola@txfusion.io>
Date: Wed, 25 Dec 2024 15:13:53 +0100
Subject: [PATCH] feat: add search for token balances

---
 pages/balances.vue | 48 +++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 43 insertions(+), 5 deletions(-)

diff --git a/pages/balances.vue b/pages/balances.vue
index 8cab3fa10..674b53395 100644
--- a/pages/balances.vue
+++ b/pages/balances.vue
@@ -14,11 +14,20 @@
           {{ balanceError.message }}
         </CommonErrorBlock>
       </CommonCardWithLineButtons>
-      <div v-else>
-        <div v-for="(group, index) in balanceGroups" :key="index">
-          <TypographyCategoryLabel v-if="group.title">
-            {{ group.title }}
-          </TypographyCategoryLabel>
+      <div v-else class="space-y-4">
+        <CommonInputSearch
+          v-model.trim="search"
+          class="mb-block-padding-1/4"
+          placeholder="Search by symbol or address"
+          autofocus="desktop"
+        >
+          <template #icon>
+            <MagnifyingGlassIcon aria-hidden="true" />
+          </template>
+        </CommonInputSearch>
+
+        <div v-for="(group, index) in filteredBalances" :key="index">
+          <TypographyCategoryLabel v-if="group.title"> {{ group.title }} </TypographyCategoryLabel>
           <CommonCardWithLineButtons>
             <TokenBalance
               v-for="item in group.balances"
@@ -37,6 +46,8 @@
 </template>
 
 <script lang="ts" setup>
+import type { TokenAmount } from "@/types";
+
 const onboardStore = useOnboardStore();
 const walletEraStore = useZkSyncWalletStore();
 const { isConnected } = storeToRefs(onboardStore);
@@ -45,8 +56,35 @@ const { eraNetwork } = storeToRefs(useZkSyncProviderStore());
 
 const { loading, reset: resetSingleLoading } = useSingleLoading(computed(() => balanceInProgress.value));
 
+const search = ref("");
 const balanceGroups = groupBalancesByAmount(balance);
 
+const filterBalanceGroups = (
+  balancesGroups: globalThis.ComputedRef<
+    {
+      title: string | null;
+      balances: TokenAmount[];
+    }[]
+  >
+) => {
+  const lowercaseSearch = search.value.toLowerCase();
+
+  return balancesGroups.value.map((balanceGroup) => {
+    const filteredGroupBalances = balanceGroup.balances.filter(({ symbol, name, l1Address, address }) => {
+      return Object.values({ address, name, symbol, l1Address })
+        .filter((e) => typeof e === "string")
+        .some((value) => value!.toLowerCase().includes(lowercaseSearch));
+    });
+
+    return {
+      title: balanceGroup.title,
+      balances: filteredGroupBalances,
+    };
+  });
+};
+
+const filteredBalances = computed(() => filterBalanceGroups(balanceGroups));
+
 const fetch = () => {
   walletEraStore.requestBalance();
 };