From ca9aaf3dbf7e42a336d497759b6d1bdf568ad043 Mon Sep 17 00:00:00 2001
From: Wukong Sun <swkatmask@gmail.com>
Date: Wed, 5 Mar 2025 14:54:40 +0800
Subject: [PATCH 1/2] fix: enter to search

---
 .../UI/components/AssetsManagement/CollectionHeader.tsx   | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/packages/shared/src/UI/components/AssetsManagement/CollectionHeader.tsx b/packages/shared/src/UI/components/AssetsManagement/CollectionHeader.tsx
index 8bbc49224003..d96b608aa3fb 100644
--- a/packages/shared/src/UI/components/AssetsManagement/CollectionHeader.tsx
+++ b/packages/shared/src/UI/components/AssetsManagement/CollectionHeader.tsx
@@ -157,13 +157,13 @@ export const CollectionHeader = memo(function CollectionHeader({
                             autoFocus
                             fullWidth
                             value={pendingKeyword}
-                            onKeyDown={(event: KeyboardEvent<HTMLInputElement>) => {
-                                if (event.key !== 'Enter') return
-                                setSearchKeyword(event.currentTarget.value)
-                            }}
                             onChange={(event) => setPendingKeyword(event.target.value)}
                             InputProps={{
                                 startAdornment: <Icons.Search size={16} />,
+                                onKeyDown: (event: KeyboardEvent<HTMLInputElement>) => {
+                                    if (event.key !== 'Enter') return
+                                    setSearchKeyword(pendingKeyword)
+                                },
                             }}
                         />
                         <Button

From a9a9ea97ca62839199785b6e0b3236c1d6dcb810 Mon Sep 17 00:00:00 2001
From: Wukong Sun <swkatmask@gmail.com>
Date: Wed, 5 Mar 2025 15:25:04 +0800
Subject: [PATCH 2/2] fix: align chain list

---
 .../SiteAdaptor/views/SelectCollectibles.tsx  | 11 +++++--
 .../AssetsManagement/ChainRuntimeProvider.tsx | 33 +++++++------------
 .../UI/components/AssetsManagement/index.ts   |  1 +
 .../AssetsManagement/useAssetsNetworks.ts     | 26 +++++++++++++++
 .../AddCollectiblesDialog.tsx                 | 12 +++++--
 .../UI/modals/AddCollectiblesModal/index.tsx  |  8 +++--
 6 files changed, 63 insertions(+), 28 deletions(-)
 create mode 100644 packages/shared/src/UI/components/AssetsManagement/useAssetsNetworks.ts

diff --git a/packages/plugins/RedPacket/src/SiteAdaptor/views/SelectCollectibles.tsx b/packages/plugins/RedPacket/src/SiteAdaptor/views/SelectCollectibles.tsx
index 2f0f7d0c31ea..98d0ed2de575 100644
--- a/packages/plugins/RedPacket/src/SiteAdaptor/views/SelectCollectibles.tsx
+++ b/packages/plugins/RedPacket/src/SiteAdaptor/views/SelectCollectibles.tsx
@@ -1,5 +1,5 @@
 import { Trans } from '@lingui/react/macro'
-import { AddCollectiblesModal, CollectionList, UserAssetsProvider } from '@masknet/shared'
+import { AddCollectiblesModal, CollectionList, useAssetsNetworks, UserAssetsProvider } from '@masknet/shared'
 import { NetworkPluginID } from '@masknet/shared-base'
 import { makeStyles } from '@masknet/theme'
 import type { Web3Helper } from '@masknet/web3-helpers'
@@ -7,12 +7,12 @@ import { useChainContext, useNetworkContext, useWeb3Connection, useWeb3Hub } fro
 import { isSameAddress } from '@masknet/web3-shared-base'
 import type { ChainId } from '@masknet/web3-shared-evm'
 import { alpha, Box, Button, DialogActions } from '@mui/material'
+import { compact, uniqBy } from 'lodash-es'
 import { useCallback, useEffect, useMemo, useState } from 'react'
 import { useNavigate } from 'react-router-dom'
 import { NFT_DEFAULT_CHAINS, NFT_RED_PACKET_MAX_SHARES } from '../../constants.js'
 import { useRedPacket } from '../contexts/RedPacketContext.js'
 import { emitter } from '../emitter.js'
-import { compact, uniqBy } from 'lodash-es'
 
 const useStyles = makeStyles()((theme) => ({
     container: {
@@ -48,6 +48,7 @@ export function SelectCollectibles() {
     const { pluginID } = useNetworkContext()
     const Web3 = useWeb3Connection(pluginID)
     const Hub = useWeb3Hub(pluginID)
+    const networks = useAssetsNetworks(pluginID)
     const { selectedNfts, setSelectedNfts, setCollection } = useRedPacket()
     const [pendingNfts, setPendingNfts] = useState<Web3Helper.NonFungibleAssetAll[]>(selectedNfts)
     const handleItemClick = useCallback((nft: Web3Helper.NonFungibleAssetAll) => {
@@ -71,10 +72,14 @@ export function SelectCollectibles() {
     const [pendingTokenCount, setPendingTokenCount] = useState(0)
     const [tokens, setTokens] = useState<Web3Helper.NonFungibleAssetAll[]>([])
     const handleAddCollectibles = useCallback(async () => {
+        const chainWhiteList = networks
+            .filter((x) => NFT_DEFAULT_CHAINS.includes(x.chainId as ChainId))
+            .map((x) => x.chainId)
         const results = await AddCollectiblesModal.openAndWaitForClose({
             pluginID,
             chainId: assetChainId || chainId,
             account,
+            chainWhiteList,
         })
         if (!results) return
         const [contract, tokenIds] = results
@@ -108,7 +113,7 @@ export function SelectCollectibles() {
         setTokens((originalTokens) => {
             return uniqBy([...originalTokens, ...tokens], (x) => `${x.contract?.address}.${x.tokenId}`)
         })
-    }, [pluginID, assetChainId, chainId, account])
+    }, [pluginID, assetChainId, chainId, account, networks])
 
     const handleSelect = useCallback((assets: Web3Helper.NonFungibleAssetAll[]) => {
         setPendingNfts(assets.length > NFT_RED_PACKET_MAX_SHARES ? assets.slice(0, NFT_RED_PACKET_MAX_SHARES) : assets)
diff --git a/packages/shared/src/UI/components/AssetsManagement/ChainRuntimeProvider.tsx b/packages/shared/src/UI/components/AssetsManagement/ChainRuntimeProvider.tsx
index 9e5c9ff4a466..ff1d3afb2957 100644
--- a/packages/shared/src/UI/components/AssetsManagement/ChainRuntimeProvider.tsx
+++ b/packages/shared/src/UI/components/AssetsManagement/ChainRuntimeProvider.tsx
@@ -1,22 +1,21 @@
 import { EMPTY_LIST, NetworkPluginID } from '@masknet/shared-base'
 import type { Web3Helper } from '@masknet/web3-helpers'
-import { CHAIN_DESCRIPTORS, ChainId, type NetworkType, type SchemaType } from '@masknet/web3-shared-evm'
+import type { ReasonableNetwork } from '@masknet/web3-shared-base'
+import { type ChainId } from '@masknet/web3-shared-evm'
 import { type ChainId as FlowChainId } from '@masknet/web3-shared-flow'
-import { noop, sortBy } from 'lodash-es'
 import { type ChainId as SolanaChainId } from '@masknet/web3-shared-solana'
-import { useNetworks } from '@masknet/web3-hooks-base'
+import { noop } from 'lodash-es'
 import {
     createContext,
     memo,
+    useContext,
     useMemo,
-    type PropsWithChildren,
     useState,
-    type SetStateAction,
     type Dispatch,
-    useContext,
+    type PropsWithChildren,
+    type SetStateAction,
 } from 'react'
-import type { ReasonableNetwork } from '@masknet/web3-shared-base'
-import { SimpleHashSupportedChains } from '../../../constants.js'
+import { useAssetsNetworks } from './useAssetsNetworks.js'
 
 interface ChainRuntimeOptions {
     pluginID: NetworkPluginID
@@ -46,21 +45,13 @@ export const ChainRuntimeProvider = memo<PropsWithChildren<ChainRuntimeProviderP
     children,
 }) {
     const [chainId, setChainId] = useState<Web3Helper.ChainIdAll>()
-    const allNetworks = useNetworks(pluginID, true)
+    const assetsNetworks = useAssetsNetworks(pluginID)
 
     const networks = useMemo(() => {
-        const supported = SimpleHashSupportedChains[pluginID]
-        const networks = allNetworks.filter(
-            (x) => (x.network === 'mainnet' || x.isCustomized) && supported.includes(x.chainId),
-        )
-        // hard-coded for Zora
-        if (pluginID === NetworkPluginID.PLUGIN_EVM) {
-            const zora = CHAIN_DESCRIPTORS.find((x) => x.chainId === ChainId.Zora)
-            if (zora) networks.push(zora as ReasonableNetwork<ChainId, SchemaType, NetworkType>)
-        }
-        const list = sortBy(networks, (x) => supported.indexOf(x.chainId))
-        return chainWhiteList?.length ? list.filter((x) => chainWhiteList.includes(x.chainId as ChainId)) : list
-    }, [allNetworks, pluginID, chainWhiteList])
+        return chainWhiteList?.length ?
+                assetsNetworks.filter((x) => chainWhiteList.includes(x.chainId as ChainId))
+            :   assetsNetworks
+    }, [chainWhiteList, assetsNetworks])
 
     const currentChainId = chainId ?? defaultChainId ?? (networks.length === 1 ? networks[0].chainId : chainId)
 
diff --git a/packages/shared/src/UI/components/AssetsManagement/index.ts b/packages/shared/src/UI/components/AssetsManagement/index.ts
index b610fef44114..3f91d57c9b61 100644
--- a/packages/shared/src/UI/components/AssetsManagement/index.ts
+++ b/packages/shared/src/UI/components/AssetsManagement/index.ts
@@ -2,3 +2,4 @@ export type * from './types.js'
 export * from './CollectionList.js'
 export * from './UserAssetsContext.js'
 export * from './CollectibleCard.js'
+export * from './useAssetsNetworks.js'
diff --git a/packages/shared/src/UI/components/AssetsManagement/useAssetsNetworks.ts b/packages/shared/src/UI/components/AssetsManagement/useAssetsNetworks.ts
new file mode 100644
index 000000000000..58449d344318
--- /dev/null
+++ b/packages/shared/src/UI/components/AssetsManagement/useAssetsNetworks.ts
@@ -0,0 +1,26 @@
+import { SimpleHashSupportedChains } from '@masknet/shared'
+import { NetworkPluginID } from '@masknet/shared-base'
+import { useNetworks } from '@masknet/web3-hooks-base'
+import type { ReasonableNetwork } from '@masknet/web3-shared-base'
+import { CHAIN_DESCRIPTORS, ChainId, type SchemaType, type NetworkType } from '@masknet/web3-shared-evm'
+import { sortBy } from 'lodash-es'
+import { useMemo } from 'react'
+
+export function useAssetsNetworks(pluginID: NetworkPluginID) {
+    const allNetworks = useNetworks(pluginID, true)
+    const networks = useMemo(() => {
+        const supported = SimpleHashSupportedChains[pluginID]
+        const networks = allNetworks.filter(
+            (x) => (x.network === 'mainnet' || x.isCustomized) && supported.includes(x.chainId),
+        )
+        // hard-coded for Zora
+        if (pluginID === NetworkPluginID.PLUGIN_EVM) {
+            const zora = CHAIN_DESCRIPTORS.find((x) => x.chainId === ChainId.Zora)
+            if (zora) networks.push(zora as ReasonableNetwork<ChainId, SchemaType, NetworkType>)
+        }
+        const list = sortBy(networks, (x) => supported.indexOf(x.chainId))
+        return list
+    }, [allNetworks, pluginID])
+
+    return networks
+}
diff --git a/packages/shared/src/UI/modals/AddCollectiblesModal/AddCollectiblesDialog.tsx b/packages/shared/src/UI/modals/AddCollectiblesModal/AddCollectiblesDialog.tsx
index 92483a90b8d3..70f2fbba4b29 100644
--- a/packages/shared/src/UI/modals/AddCollectiblesModal/AddCollectiblesDialog.tsx
+++ b/packages/shared/src/UI/modals/AddCollectiblesModal/AddCollectiblesDialog.tsx
@@ -2,10 +2,12 @@ import { Trans } from '@lingui/react/macro'
 import { type NetworkPluginID } from '@masknet/shared-base'
 import { makeStyles } from '@masknet/theme'
 import { useNetworks } from '@masknet/web3-hooks-base'
+import type { ChainId } from '@masknet/web3-shared-evm'
 import { DialogContent } from '@mui/material'
-import { memo, useState } from 'react'
+import { memo, useMemo, useState } from 'react'
 import { AddCollectibles, SelectNetworkSidebar, type AddCollectiblesProps } from '../../components/index.js'
 import { InjectedDialog } from '../../contexts/components/index.js'
+import type { Web3Helper } from '@masknet/web3-helpers'
 
 const useStyles = makeStyles()((theme) => ({
     content: {
@@ -28,6 +30,7 @@ const useStyles = makeStyles()((theme) => ({
 
 interface AddCollectiblesDialogProps<T extends NetworkPluginID = NetworkPluginID> extends AddCollectiblesProps<T> {
     open: boolean
+    chainWhiteList?: Web3Helper.ChainIdAll[]
 }
 
 export const AddCollectiblesDialog = memo(function AddCollectiblesDialog({
@@ -35,12 +38,17 @@ export const AddCollectiblesDialog = memo(function AddCollectiblesDialog({
     pluginID,
     chainId: defaultChainId,
     account,
+    chainWhiteList,
     onAdd,
 }: AddCollectiblesDialogProps) {
     const { classes } = useStyles()
 
     const [chainId, setChainId] = useState(defaultChainId)
     const allNetworks = useNetworks(pluginID, true)
+    const networks = useMemo(() => {
+        if (!chainWhiteList?.length) return allNetworks
+        return allNetworks.filter((x) => chainWhiteList.includes(x.chainId as ChainId))
+    }, [allNetworks, chainWhiteList])
 
     return (
         <InjectedDialog
@@ -54,7 +62,7 @@ export const AddCollectiblesDialog = memo(function AddCollectiblesDialog({
                     chainId={chainId}
                     onChainChange={setChainId}
                     pluginID={pluginID}
-                    networks={allNetworks}
+                    networks={networks}
                     hideAllButton
                 />
                 <AddCollectibles
diff --git a/packages/shared/src/UI/modals/AddCollectiblesModal/index.tsx b/packages/shared/src/UI/modals/AddCollectiblesModal/index.tsx
index f5e7c32768f9..84fbbd5ac1f0 100644
--- a/packages/shared/src/UI/modals/AddCollectiblesModal/index.tsx
+++ b/packages/shared/src/UI/modals/AddCollectiblesModal/index.tsx
@@ -1,8 +1,8 @@
-import { useState } from 'react'
-import { NetworkPluginID, type SingletonModalProps } from '@masknet/shared-base'
+import { EMPTY_LIST, NetworkPluginID, type SingletonModalProps } from '@masknet/shared-base'
 import { useSingletonModal } from '@masknet/shared-base-ui'
 import type { Web3Helper } from '@masknet/web3-helpers'
 import type { NonFungibleTokenContract } from '@masknet/web3-shared-base'
+import { useState } from 'react'
 import { AddCollectiblesDialog } from './AddCollectiblesDialog.js'
 
 export interface AddCollectiblesModalOpenProps {
@@ -13,6 +13,7 @@ export interface AddCollectiblesModalOpenProps {
      * For example, in PFP, we can add collectibles from verified wallets if no wallet connected.
      */
     account?: string
+    chainWhiteList?: Web3Helper.ChainIdAll[]
 }
 
 export type AddCollectiblesModalCloseProps =
@@ -25,12 +26,14 @@ export function AddCollectiblesModal({
     const [pluginID, setPluginID] = useState<NetworkPluginID>(NetworkPluginID.PLUGIN_EVM)
     const [chainId, setChainId] = useState<Web3Helper.ChainIdAll>()
     const [account, setAccount] = useState<string>()
+    const [chainWhiteList, setChainWhiteList] = useState<Web3Helper.ChainIdAll[]>(EMPTY_LIST)
 
     const [open, dispatch] = useSingletonModal(ref, {
         onOpen(props) {
             setPluginID(props.pluginID ?? NetworkPluginID.PLUGIN_EVM)
             setChainId(props.chainId)
             setAccount(props.account)
+            setChainWhiteList(props.chainWhiteList ?? EMPTY_LIST)
         },
     })
 
@@ -42,6 +45,7 @@ export function AddCollectiblesModal({
             pluginID={pluginID}
             account={account}
             chainId={chainId}
+            chainWhiteList={chainWhiteList}
         />
     )
 }