Combine all chains provider into grk provider

covalenthq · Dec 5, 2023 · 3165274 · 3165274
1 parent 264a0b9
commit 3165274
Show file tree

Hide file tree

Showing 7 changed files with 848 additions and 471 deletions.
diff --git a/.storybook/preview.tsx b/.storybook/preview.tsx
@@ -57,9 +57,7 @@ const preview: Preview = {
             <CovalentProvider
                 apikey={import.meta.env.STORYBOOK_COVALENT_API_KEY}
             >
-                <ChainsProvider>
-                    <Story />
-                </ChainsProvider>
+                <Story />
             </CovalentProvider>
         ),
     ],

diff --git a/package-lock.json b/package-lock.json
diff --git a/src/components/Organisms/TokenBalances/TokenBalancesListView/TokenBalancesListView.tsx b/src/components/Organisms/TokenBalances/TokenBalancesListView/TokenBalancesListView.tsx
@@ -41,7 +41,6 @@ import { TableHeaderSorting } from "@/components/ui/tableHeaderSorting";
 import { sum } from "lodash";
 import { IconWrapper } from "@/components/Atoms/IconWrapper/IconWrapper";
 import { GRK_SIZES } from "@/utils/constants/shared.constants";
-import { useChains } from "@/utils/store/Chains";
 import { useCovalent } from "@/utils/store/Covalent";
 import {
     type CrossChainBalanceItem,
@@ -55,8 +54,7 @@ export const TokenBalancesListView: React.FC<TokenBalancesListViewProps> = ({
     hide_small_balances,
     on_transfer_click,
 }) => {
-    const { covalentClient } = useCovalent();
-    const { chains } = useChains();
+    const { covalentClient, chains } = useCovalent();
 
     const [sorting, setSorting] = useState<SortingState>([
         {

diff --git a/src/components/Organisms/TokenBalances/TokenTransfersListView/TokenTransfersListView.tsx b/src/components/Organisms/TokenBalances/TokenTransfersListView/TokenTransfersListView.tsx
@@ -52,7 +52,6 @@ import {
     type BlockTransactionWithContractTransfersWithDelta,
     type TokenTransferMeta,
 } from "@/utils/types/organisms.types";
-import { useChains } from "@/utils/store/Chains";
 import { useCovalent } from "@/utils/store/Covalent";
 
 const columns: ColumnDef<BlockTransactionWithContractTransfers>[] = [
@@ -210,8 +209,7 @@ export const TokenTransfersListView: React.FC<TokenTransfersListViewProps> = ({
     address,
     contract_address,
 }) => {
-    const { covalentClient } = useCovalent();
-    const { chains } = useChains();
+    const { covalentClient, chains } = useCovalent();
 
     const [sorting, setSorting] = useState<SortingState>([]);
     const [rowSelection, setRowSelection] = useState({});

diff --git a/src/index.ts b/src/index.ts
@@ -2,4 +2,4 @@ export { AddressActivityListView } from "./components/Organisms/TokenBalances/Ad
 export { TokenBalancesListView } from "./components/Organisms/TokenBalances/TokenBalancesListView/TokenBalancesListView";
 export { TokenTransfersListView } from "./components/Organisms/TokenBalances/TokenTransfersListView/TokenTransfersListView";
 export { NFTWalletTokenListView } from "./components/Organisms/NFTs/NFTWalletTokenListView/NFTWalletTokenListView";
-export { CovalentProvider as GoldRushProvider } from "./utils/store/Covalent";
+export { CovalentProvider as GoldRushProvider } from "./utils/store/Covalent";
diff --git a/src/utils/store/Chains.tsx b/src/utils/store/Chains.tsx
diff --git a/src/utils/store/Covalent.tsx b/src/utils/store/Covalent.tsx
@@ -1,9 +1,10 @@
-import { createContext, useContext, useMemo } from "react";
-import { CovalentClient } from "@covalenthq/client-sdk";
+import { createContext, useContext, useEffect, useMemo, useState } from "react";
+import { ChainItem, CovalentClient } from "@covalenthq/client-sdk";
 import { Toaster } from "@/components/ui/toaster";
 
 interface CovalentContextType {
     covalentClient: CovalentClient;
+    chains: ChainItem[] | null;
 }
 
 interface CovalentProviderProps {
@@ -23,13 +24,27 @@ export const CovalentProvider: React.FC<CovalentProviderProps> = ({
         () => new CovalentClient(apikey),
         [apikey]
     );
+    const [chains, setChains] = useState<ChainItem[] | null>(null);
+
+    useEffect(() => {
+        (async () => {
+            try {
+                const allChainsResp =
+                    await covalentClient.BaseService.getAllChains();
+                setChains(allChainsResp.data.items);
+            } catch (error) {
+                console.error(error);
+            }
+        })();
+    }, []);
 
     return (
-        <CovalentContext.Provider value={{ covalentClient: covalentClient }}>
+        <CovalentContext.Provider value={{ covalentClient: covalentClient, chains: chains  }}>
             {children}
             <Toaster />
         </CovalentContext.Provider>
     );
 };
 
 export const useCovalent = () => useContext(CovalentContext);
+