From 03ef6f766b6a4a907b39b7ff6347dba51f44c348 Mon Sep 17 00:00:00 2001
From: Yashvardhan Jagnani <60016972+jagnani73@users.noreply.github.com>
Date: Fri, 19 Apr 2024 23:09:26 +0530
Subject: [PATCH] feat(chain-selector): filter options + loading (#159)

---
 .../ChainSelector/ChainSelector.stories.tsx   | 17 ++--
 .../Molecules/ChainSelector/ChainSelector.tsx | 81 ++++++++++++-------
 src/utils/types/molecules.types.ts            |  1 +
 3 files changed, 63 insertions(+), 36 deletions(-)

diff --git a/src/components/Molecules/ChainSelector/ChainSelector.stories.tsx b/src/components/Molecules/ChainSelector/ChainSelector.stories.tsx
index 80bf8d26..ee43f5e3 100644
--- a/src/components/Molecules/ChainSelector/ChainSelector.stories.tsx
+++ b/src/components/Molecules/ChainSelector/ChainSelector.stories.tsx
@@ -6,18 +6,17 @@ type Story = StoryObj<typeof ChainSelectorComponent>;
 const meta: Meta<typeof ChainSelectorComponent> = {
     title: "Molecules/Chain Selector",
     component: ChainSelectorComponent,
-    argTypes: {
-        onChangeChain: {
-            control: false,
-            table: {
-                disable: true,
-            },
-        },
-    },
 };
 
 export default meta;
 
 export const ChainSelector: Story = {
-    args: {},
+    args: {
+        chain_options: [
+            "eth-mainnet",
+            "matic-mainnet",
+            "arbitrum-mainnet",
+            "solana-mainnet",
+        ],
+    },
 };
diff --git a/src/components/Molecules/ChainSelector/ChainSelector.tsx b/src/components/Molecules/ChainSelector/ChainSelector.tsx
index ad98dd96..b9b4918c 100644
--- a/src/components/Molecules/ChainSelector/ChainSelector.tsx
+++ b/src/components/Molecules/ChainSelector/ChainSelector.tsx
@@ -11,16 +11,33 @@ import {
     PopoverContent,
     PopoverTrigger,
 } from "@/components/ui/popover";
-import { useState } from "react";
+import { useMemo, useState } from "react";
 import { CheckIcon, DoubleArrowDownIcon } from "@radix-ui/react-icons";
 import { Button } from "@/components/ui/button";
 import { type ChainSelectorProps } from "@/utils/types/molecules.types";
+import { type ChainItem, type Chain } from "@covalenthq/client-sdk";
+import { CommandLoading } from "cmdk";
+import { Skeleton } from "@/components/ui/skeleton";
+import { GRK_SIZES } from "@/utils/constants/shared.constants";
 
 export const ChainSelector: React.FC<ChainSelectorProps> = ({
+    chain_options = [],
     onChangeChain,
 }) => {
     const { chains, selectedChain, setSelectedChain } = useGoldRush();
-    const [open, setOpen] = useState<boolean>(false);
+    const [open, setOpen] = useState<boolean>(!false);
+
+    const dropdownChains = useMemo<ChainItem[] | null>(() => {
+        if (!chains) {
+            return null;
+        }
+        const chainOptionsSet = new Set(chain_options);
+        if (chainOptionsSet.size === 0) {
+            return chains;
+        }
+
+        return chains.filter(({ name }) => chainOptionsSet.has(name as Chain));
+    }, [chains, chain_options]);
 
     return (
         <Popover open={open} onOpenChange={setOpen}>
@@ -40,31 +57,41 @@ export const ChainSelector: React.FC<ChainSelectorProps> = ({
                 <Command className="w-72">
                     <CommandInput placeholder="Search chain..." />
                     <CommandEmpty>No chain found.</CommandEmpty>
-                    <CommandGroup>
-                        {chains?.map((chain) => (
-                            <CommandItem
-                                key={chain.name}
-                                value={chain.name}
-                                className="cursor-pointer bg-background-light dark:bg-background-dark"
-                                onSelect={() => {
-                                    setSelectedChain(chain);
-                                    setOpen(false);
-                                    if (onChangeChain) {
-                                        onChangeChain(chain);
-                                    }
-                                }}
-                            >
-                                {chain.label}
-                                <CheckIcon
-                                    className={`w-4" ml-auto h-4 ${
-                                        chain.name === selectedChain?.name
-                                            ? "opacity-100"
-                                            : "opacity-0"
-                                    }`}
-                                />
-                            </CommandItem>
-                        ))}
-                    </CommandGroup>
+                    {!dropdownChains ? (
+                        <CommandLoading>
+                            <div className="flex flex-col gap-2 px-4 pt-2">
+                                <Skeleton size={GRK_SIZES.LARGE} />
+                                <Skeleton size={GRK_SIZES.LARGE} />
+                                <Skeleton size={GRK_SIZES.LARGE} />
+                            </div>
+                        </CommandLoading>
+                    ) : (
+                        <CommandGroup>
+                            {dropdownChains.map((chain) => (
+                                <CommandItem
+                                    key={chain.name}
+                                    value={chain.name}
+                                    className="cursor-pointer bg-background-light dark:bg-background-dark"
+                                    onSelect={() => {
+                                        setSelectedChain(chain);
+                                        setOpen(false);
+                                        if (onChangeChain) {
+                                            onChangeChain(chain);
+                                        }
+                                    }}
+                                >
+                                    {chain.label}
+                                    <CheckIcon
+                                        className={`w-4" ml-auto h-4 ${
+                                            chain.name === selectedChain?.name
+                                                ? "opacity-100"
+                                                : "opacity-0"
+                                        }`}
+                                    />
+                                </CommandItem>
+                            ))}
+                        </CommandGroup>
+                    )}
                 </Command>
             </PopoverContent>
         </Popover>
diff --git a/src/utils/types/molecules.types.ts b/src/utils/types/molecules.types.ts
index df52599e..512ad5bc 100644
--- a/src/utils/types/molecules.types.ts
+++ b/src/utils/types/molecules.types.ts
@@ -42,6 +42,7 @@ export interface AddressDetailsProps {
 }
 
 export interface ChainSelectorProps {
+    chain_options?: Chain[];
     onChangeChain?: (chain: ChainItem) => unknown;
 }