From 644321bda15f23b1d78733bab261bbd30c2739d1 Mon Sep 17 00:00:00 2001
From: h0ngcha0 <lbs.lhc@gmail.com>
Date: Mon, 22 Jul 2024 06:14:32 +0200
Subject: [PATCH] Cache fetching token metadata

---
 .../src/ui/features/accountNfts/alephium-nft.service.ts    | 7 ++++---
 .../src/ui/features/accountTokens/tokens.state.ts          | 2 +-
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/packages/extension/src/ui/features/accountNfts/alephium-nft.service.ts b/packages/extension/src/ui/features/accountNfts/alephium-nft.service.ts
index 72d28058..a9cd3bc0 100644
--- a/packages/extension/src/ui/features/accountNfts/alephium-nft.service.ts
+++ b/packages/extension/src/ui/features/accountNfts/alephium-nft.service.ts
@@ -2,6 +2,7 @@ import { NFT, CollectionAndNFTMap } from "./alephium-nft.model"
 import { NFTCollection } from "./alephium-nft.model"
 import { Network } from "../../../shared/network"
 import { ExplorerProvider, NodeProvider } from "@alephium/web3"
+import { fetchImmutable } from "../../../shared/utils/fetchImmutable"
 
 export const fetchCollectionAndNfts = async (
   nonFungibleTokenIds: string[],
@@ -25,7 +26,7 @@ export const fetchCollectionAndNfts = async (
     const nodeProvider = new NodeProvider(network.nodeUrl)
     const otherNonFungibleTokenIds = nonFungibleTokenIds.filter(id => nftMetadataz.findIndex((m) => m.id == id) == -1)
     for (const tokenId of otherNonFungibleTokenIds) {
-      const nftMetadata = await nodeProvider.fetchNFTMetaData(tokenId)
+      const nftMetadata = await fetchImmutable(`${tokenId}-nft-metadata`, () => nodeProvider.fetchNFTMetaData(tokenId))
       const collectionId = nftMetadata.collectionId
       if (parentAndTokenIds[collectionId]) {
         parentAndTokenIds[collectionId].push(tokenId)
@@ -60,7 +61,7 @@ export const fetchNFTCollection = async (
 export async function getNFT(collectionId: string, nftId: string, network: Network): Promise<NFT | undefined> {
   try {
     const nodeProvider = new NodeProvider(network.nodeUrl)
-    const nftMetadata = await nodeProvider.fetchNFTMetaData(nftId)
+    const nftMetadata = await fetchImmutable(`${nftId}-nft-metadata`, () => nodeProvider.fetchNFTMetaData(nftId))
     const metadataResponse = await fetch(nftMetadata.tokenUri)
     const metadata = await metadataResponse.json()
     return {
@@ -81,7 +82,7 @@ async function getCollectionMetadata(
   collectionId: string,
   nodeProvider: NodeProvider
 ) {
-  const metadata = await nodeProvider.fetchNFTCollectionMetaData(collectionId)
+  const metadata = await fetchImmutable(`${collectionId}-nft-collection-metadata`, () => nodeProvider.fetchNFTCollectionMetaData(collectionId))
   const metadataResponse = await fetch(metadata.collectionUri)
   return await metadataResponse.json()
 }
diff --git a/packages/extension/src/ui/features/accountTokens/tokens.state.ts b/packages/extension/src/ui/features/accountTokens/tokens.state.ts
index 20f23e23..b14597fa 100644
--- a/packages/extension/src/ui/features/accountTokens/tokens.state.ts
+++ b/packages/extension/src/ui/features/accountTokens/tokens.state.ts
@@ -337,7 +337,7 @@ async function fetchFungibleTokenFromFullNode(network: Network, tokenId: string)
       return undefined
     }
 
-    const metadata = await nodeProvider.fetchFungibleTokenMetaData(tokenId)
+    const metadata = await fetchImmutable(`${tokenId}-token-metadata`, () => nodeProvider.fetchFungibleTokenMetaData(tokenId))
     const token: Token = {
       id: tokenId,
       networkId: network.id,