Remove MintContextProvider

src/swap/components/Info.tsx

@@ -10,7 +10,7 @@ import PopupState, { bindTrigger, bindPopover } from "material-ui-popup-state";
 import { PublicKey } from "@solana/web3.js";
 import { useTokenMap } from "../context/TokenList";
 import { useSwapContext, useSwapFair } from "../context/Swap";
-import { useMint } from "../context/Mint";
+import { useMint } from "../context/Token";
 import { useRoute, useMarketName, useFair } from "../context/Dex";
 
 const useStyles = makeStyles((theme) => ({

src/swap/components/OpenOrdersDialog.tsx

@@ -24,8 +24,7 @@ import {
 import { Close } from "@material-ui/icons";
 import { useMarket, useOpenOrders, useDexContext } from "../context/Dex";
 import { useTokenMap } from "../context/TokenList";
-import { useMint } from "../context/Mint";
-import { useOwnedTokenAccount } from "../context/Token";
+import { useMint, useOwnedTokenAccount } from "../context/Token";
 
 const useStyles = makeStyles((theme) => ({
   table: {},

src/swap/components/Swap.tsx

@@ -17,9 +17,8 @@ import {
   useRouteVerbose,
   useMarket,
 } from "../context/Dex";
-import { useMint } from "../context/Mint";
 import { useTokenMap } from "../context/TokenList";
-import { useOwnedTokenAccount } from "../context/Token";
+import { useMint, useOwnedTokenAccount } from "../context/Token";
 import { useCanSwap } from "../context/Swap";
 import TokenDialog from "./TokenDialog";
 import { SettingsButton } from "./Settings";

src/swap/context/Mint.tsx

@@ -1,64 +0,0 @@
-import React, { useContext, useState } from "react";
-import { useAsync } from "react-async-hook";
-import { Provider } from "@project-serum/anchor";
-import { PublicKey, Account } from "@solana/web3.js";
-import { MintInfo, Token, TOKEN_PROGRAM_ID } from "@solana/spl-token";
-
-const _MintContext = React.createContext<null | MintContext>(null);
-type MintContext = {
-  mintCache: Map<string, MintInfo>;
-  setMintCache: (m: Map<string, MintInfo>) => void;
-  provider: Provider;
-};
-
-export function MintContextProvider(props: any) {
-  const provider = props.provider;
-  const [mintCache, setMintCache] = useState(new Map<string, MintInfo>());
-
-  return (
-    <_MintContext.Provider
-      value={{
-        mintCache,
-        setMintCache,
-        provider,
-      }}
-    >
-      {props.children}
-    </_MintContext.Provider>
-  );
-}
-
-export function useMint(mint?: PublicKey): MintInfo | undefined | null {
-  const ctx = useContext(_MintContext);
-  if (ctx === null) {
-    throw new Error("Mint context not found");
-  }
-
-  // Lazy load the mint account if needeed.
-  const asyncMintInfo = useAsync(async () => {
-    if (!mint) {
-      return undefined;
-    }
-    if (ctx.mintCache.get(mint.toString())) {
-      return ctx.mintCache.get(mint.toString());
-    }
-    const mintClient = new Token(
-      ctx.provider.connection,
-      mint,
-      TOKEN_PROGRAM_ID,
-      new Account()
-    );
-    const mintInfo = await mintClient.getMintInfo();
-
-    let cache = new Map(ctx.mintCache);
-    cache.set(mint.toString(), mintInfo);
-    ctx.setMintCache(cache);
-
-    return mintInfo;
-  }, [ctx.provider.connection, mint]);
-
-  if (asyncMintInfo.result) {
-    return asyncMintInfo.result;
-  }
-  return undefined;
-}

src/swap/context/Token.tsx

@@ -1,9 +1,17 @@
 import React, { useContext, useState, useEffect } from "react";
-import { PublicKey } from "@solana/web3.js";
-import { AccountInfo as TokenAccount } from "@solana/spl-token";
+import { useAsync } from "react-async-hook";
+import { Provider } from "@project-serum/anchor";
+import { PublicKey, Account } from "@solana/web3.js";
+import {
+  MintInfo,
+  AccountInfo as TokenAccount,
+  Token,
+  TOKEN_PROGRAM_ID,
+} from "@solana/spl-token";
 import { getOwnedTokenAccounts } from "../utils/tokens";
 
 export type TokenContext = {
+  provider: Provider;
   ownedTokenAccounts:
     | { publicKey: PublicKey; account: TokenAccount }[]
     | undefined;
@@ -29,6 +37,7 @@ export function TokenContextProvider(props: any) {
     <_TokenContext.Provider
       value={{
         ownedTokenAccounts,
+        provider,
       }}
     >
       {props.children}
@@ -36,15 +45,20 @@ export function TokenContextProvider(props: any) {
   );
 }
 
+function useTokenContext() {
+  const ctx = useContext(_TokenContext);
+  if (ctx === null) {
+    throw new Error("Context not available");
+  }
+  return ctx;
+}
+
 // Null => none exists.
 // Undefined => loading.
 export function useOwnedTokenAccount(
   mint?: PublicKey
 ): { publicKey: PublicKey; account: TokenAccount } | null | undefined {
-  const ctx = useContext(_TokenContext);
-  if (ctx === null) {
-    throw new Error("Context not available");
-  }
+  const ctx = useTokenContext();
   if (mint === undefined) {
     return mint;
   }
@@ -69,3 +83,34 @@ export function useOwnedTokenAccount(
   );
   return tokenAccounts[0];
 }
+
+// Cache storing all previously fetched mint infos.
+const _MINT_CACHE = new Map<string, MintInfo>();
+
+export function useMint(mint?: PublicKey): MintInfo | undefined | null {
+  const { provider } = useTokenContext();
+  // Lazy load the mint account if needeed.
+  const asyncMintInfo = useAsync(async () => {
+    if (!mint) {
+      return undefined;
+    }
+    if (_MINT_CACHE.get(mint.toString())) {
+      return _MINT_CACHE.get(mint.toString());
+    }
+
+    const mintClient = new Token(
+      provider.connection,
+      mint,
+      TOKEN_PROGRAM_ID,
+      new Account()
+    );
+    const mintInfo = await mintClient.getMintInfo();
+    _MINT_CACHE.set(mint.toString(), mintInfo);
+    return mintInfo;
+  }, [provider.connection, mint]);
+
+  if (asyncMintInfo.result) {
+    return asyncMintInfo.result;
+  }
+  return undefined;
+}

src/swap/index.tsx

@@ -4,7 +4,6 @@ import { Provider } from "@project-serum/anchor";
 import { Swap as SwapClient } from "@project-serum/swap";
 import { SwapContextProvider } from "./context/Swap";
 import { DexContextProvider } from "./context/Dex";
-import { MintContextProvider } from "./context/Mint";
 import { TokenListContextProvider } from "./context/TokenList";
 import { TokenContextProvider } from "./context/Token";
 import SwapCard from "./components/Swap";
@@ -31,20 +30,18 @@ export default function Swap({
   const swapClient = new SwapClient(provider, tokenList);
   return (
     <TokenListContextProvider tokenList={tokenList}>
-      <MintContextProvider provider={provider}>
-        <TokenContextProvider provider={provider}>
-          <DexContextProvider swapClient={swapClient}>
-            <SwapContextProvider
-              fromMint={fromMint}
-              toMint={toMint}
-              fromAmount={fromAmount}
-              toAmount={toAmount}
-            >
-              <SwapCard style={style} />
-            </SwapContextProvider>
-          </DexContextProvider>
-        </TokenContextProvider>
-      </MintContextProvider>
+      <TokenContextProvider provider={provider}>
+        <DexContextProvider swapClient={swapClient}>
+          <SwapContextProvider
+            fromMint={fromMint}
+            toMint={toMint}
+            fromAmount={fromAmount}
+            toAmount={toAmount}
+          >
+            <SwapCard style={style} />
+          </SwapContextProvider>
+        </DexContextProvider>
+      </TokenContextProvider>
     </TokenListContextProvider>
   );
 }