exclude namewrapper nfts from nft avatar selection

ensdomains · Jan 22, 2024 · b666bd2 · b666bd2
1 parent d4a8abf
commit b666bd2
Show file tree

Hide file tree

Showing 5 changed files with 40 additions and 3 deletions.
diff --git a/src/components/@molecules/ProfileEditor/Avatar/AvatarNFT.tsx b/src/components/@molecules/ProfileEditor/Avatar/AvatarNFT.tsx
@@ -10,6 +10,8 @@ import MagnifyingGlassSVG from '@app/assets/MagnifyingGlass.svg'
 import { InnerDialog } from '@app/components/@atoms/InnerDialog'
 import { ScrollBoxWithSpinner, SpinnerRow } from '@app/components/@molecules/ScrollBoxWithSpinner'
 import { useChainName } from '@app/hooks/chain/useChainName'
+import { usePublicClient } from '@app/hooks/usePublicClient'
+import { getSupportedChainContractAddress } from '@app/utils/getSupportedChainContractAddress'
 
 type OwnedNFT = {
   contract: {
@@ -168,6 +170,8 @@ export const AvatarNFT = ({
   const { address: _address } = useAccount()
   const address = _address!
 
+  const publicClient = usePublicClient()
+
   const {
     data: NFTPages,
     fetchNextPage,
@@ -191,7 +195,16 @@ export const AvatarNFT = ({
         ownedNfts: response.ownedNfts.filter(
           (nft) =>
             (nft.media[0].thumbnail || nft.media[0].gateway) &&
-            nft.contract.address !== '0x57f1887a8bf19b14fc0df6fd9b2acc9af147ea85',
+            nft.contract.address !==
+              getSupportedChainContractAddress({
+                client: publicClient,
+                contract: 'ensBaseRegistrarImplementation',
+              }) &&
+            nft.contract.address !==
+              getSupportedChainContractAddress({
+                client: publicClient,
+                contract: 'ensNameWrapper',
+              }),
         ),
       } as NFTResponse
     },

diff --git a/src/constants/chains.ts b/src/constants/chains.ts
@@ -70,3 +70,5 @@ export const localhostWithEns = {
 export const mainnetWithEns = addEnsContracts(mainnet)
 export const goerliWithEns = addEnsContracts(goerli)
 export const sepoliaWithEns = addEnsContracts(sepolia)
+
+export type SupportedChain = typeof mainnetWithEns | typeof goerliWithEns | typeof sepoliaWithEns
diff --git a/src/hooks/usePublicClient.ts b/src/hooks/usePublicClient.ts
@@ -3,6 +3,8 @@ import { usePublicClient as usePublicClient_ } from 'wagmi'
 
 import { PublicClientWithChain } from '@app/types'
 
-export const usePublicClient = usePublicClient_ as <TPublicClient extends PublicClientWithChain>(
+export const usePublicClient = usePublicClient_ as <
+  TPublicClient extends PublicClientWithChain = PublicClientWithChain,
+>(
   args?: Partial<GetPublicClientArgs>,
 ) => TPublicClient
diff --git a/src/types/index.ts b/src/types/index.ts
@@ -22,6 +22,8 @@ import {
 } from '@ensdomains/ensjs/utils'
 import { Helper, Space } from '@ensdomains/thorin'
 
+import { SupportedChain } from '@app/constants/chains'
+
 export type Profile = Partial<
   GetRecordsReturnType &
     Pick<NonNullable<GetSubgraphRecordsReturnType>, 'isMigrated' | 'createdAt'> & {
@@ -161,7 +163,7 @@ export type Prettify<T> = {
   [K in keyof T]: T[K]
 } & {}
 
-export type PublicClientWithChain = PublicClient<Transport, ChainWithEns>
+export type PublicClientWithChain = PublicClient<Transport, SupportedChain>
 export type WalletClientWithAccount = WalletClient<Transport, ChainWithEns, Account>
 
 export type QueryConfig<TData, TError, TSelectData = TData> = Pick<

diff --git a/src/utils/getSupportedChainContractAddress.ts b/src/utils/getSupportedChainContractAddress.ts
@@ -0,0 +1,18 @@
+import { Client, getChainContractAddress, Transport } from 'viem'
+
+import { SupportedChain } from '@app/constants/chains'
+
+export const getSupportedChainContractAddress = <
+  const TClient extends Client<Transport, SupportedChain>,
+  TContract extends Extract<keyof TClient['chain']['contracts'], string>,
+>({
+  client,
+  contract,
+}: {
+  client: TClient
+  contract: TContract
+}) =>
+  getChainContractAddress({
+    chain: client.chain,
+    contract,
+  })