From ea23046ab0fe72cb3722e0cee37e8ebdc41b30d0 Mon Sep 17 00:00:00 2001
From: Joaquim Verges <joaquim.verges@gmail.com>
Date: Tue, 17 Dec 2024 21:40:16 +1300
Subject: [PATCH] fix: Handle NFT fetch failures gracefully and add Treasure
 chain

---
 apps/wallet-ui/src/components/ConnectButton.tsx | 11 ++++++++++-
 apps/wallet-ui/src/lib/assets/erc721.ts         |  6 +++++-
 apps/wallet-ui/src/util/simplehash.ts           |  1 +
 3 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/apps/wallet-ui/src/components/ConnectButton.tsx b/apps/wallet-ui/src/components/ConnectButton.tsx
index e5cb72e0338..10f0e3997c2 100644
--- a/apps/wallet-ui/src/components/ConnectButton.tsx
+++ b/apps/wallet-ui/src/components/ConnectButton.tsx
@@ -5,6 +5,7 @@ import {
   arbitrum,
   base,
   blast,
+  defineChain,
   mainnet,
   optimism,
   zkSync,
@@ -19,7 +20,15 @@ export default function ConnectButton({
 
   return (
     <ThirdwebConnectButton
-      chains={[mainnet, base, optimism, arbitrum, blast, zkSync]}
+      chains={[
+        mainnet,
+        base,
+        optimism,
+        arbitrum,
+        blast,
+        zkSync,
+        defineChain(61166), // Treasure mainnet
+      ]}
       wallets={[ecosystemWallet(ecosystem)]}
       client={client}
       theme={theme === "light" ? "light" : "dark"}
diff --git a/apps/wallet-ui/src/lib/assets/erc721.ts b/apps/wallet-ui/src/lib/assets/erc721.ts
index 529a08d2122..0ffabfd5411 100644
--- a/apps/wallet-ui/src/lib/assets/erc721.ts
+++ b/apps/wallet-ui/src/lib/assets/erc721.ts
@@ -61,7 +61,11 @@ export async function getErc721Tokens({
 
   if (!response.ok) {
     response.body?.cancel();
-    throw new Error("Failed to fetch NFTs");
+    console.error("Failed to fetch NFTs");
+    return {
+      nextCursor: undefined,
+      tokens: [],
+    };
   }
   const data = await response.json();
 
diff --git a/apps/wallet-ui/src/util/simplehash.ts b/apps/wallet-ui/src/util/simplehash.ts
index b4f9c9315f2..79d7a15c819 100644
--- a/apps/wallet-ui/src/util/simplehash.ts
+++ b/apps/wallet-ui/src/util/simplehash.ts
@@ -26,6 +26,7 @@ const chainIdNameMap = [
   { id: 660279, name: "xai", label: "Xai" },
   { id: 324, name: "zksync-era", label: "zkSync Era" },
   { id: 7777777, name: "zora", label: "Zora" },
+  { id: 61166, name: "treasure", label: "Treasure" },
 ];
 
 if (process.env.NEXT_PUBLIC_INCLUDE_TESTNETS) {