@@ -7,25 +7,29 @@ import Link from "next/link";
77import { usePathname } from "next/navigation" ;
88import { useEffect , useRef , useState } from "react" ;
99import useSWR from "swr" ;
10- import { useAccount , usePublicClient } from "wagmi" ;
11- import withClientConnected from "components/HighOrder/withClientConnected" ;
10+ import { useAccount , useChainId , useConfig } from "wagmi" ;
1211import useWeb3Loaded from "hooks/useWeb3Loaded" ;
1312import DesktopNavigation from "./DesktopNavigation" ;
1413import MobileMenu from "./MobileMenu" ;
1514import Options from "./Options" ;
1615import WalletSection from "./WalletSection" ;
1716
18- interface IHeader extends JSX . IntrinsicAttributes {
17+ interface IHeader {
1918 policy : string ;
2019}
2120
22- export default withClientConnected ( function Header ( { policy } : IHeader ) {
21+ export default function Header ( { policy } : IHeader ) {
2322 const [ menuOpen , setMenuOpen ] = useState < boolean > ( false ) ;
2423 const [ isDarkMode , setIsDarkMode ] = useState < boolean > ( false ) ;
2524 const menuRef = useRef < HTMLDivElement > ( null ) ;
2625 const pathname = usePathname ( ) ;
2726 const { isConnected, address } = useAccount ( ) ;
28- const { chain } = usePublicClient ( ) ;
27+ const chainId = useChainId ( )
28+
29+ const config = useConfig ( )
30+ const chains = config . chains
31+
32+ const chain = chains . find ( chain => chain . id === chainId )
2933 const web3Loaded = useWeb3Loaded ( ) ;
3034 const { data : me } = useSWR ( address , getMyData ) ;
3135
@@ -100,7 +104,7 @@ export default withClientConnected(function Header({ policy }: IHeader) {
100104
101105 < div className = "lg:absolute lg:left-1/2 lg:-translate-x-1/2 lg:transform" >
102106 < DesktopNavigation
103- { ...{ address, me, policy, pathname, chain, web3Loaded } }
107+ { ...{ address, me, policy, pathname, chain : chain ! , web3Loaded } }
104108 />
105109 </ div >
106110
@@ -113,12 +117,12 @@ export default withClientConnected(function Header({ policy }: IHeader) {
113117
114118 < div className = "flex flex-row items-center" >
115119 < div className = "hidden md:block" >
116- < WalletSection { ...{ chain, address, isConnected, web3Loaded } } />
120+ < WalletSection { ...{ chain : chain ! , address, isConnected, web3Loaded } } />
117121 </ div >
118122 < div className = "hidden md:block" >
119123 < Options />
120124 </ div >
121125 </ div >
122126 </ header >
123127 ) ;
124- } ) ;
128+ }
0 commit comments