fix: pwat not setting correct spender (#6320)

* fix: pwat not setting correct spender

* todo

* responding to comments

src/nft/components/bag/BagFooter.tsx

@@ -334,7 +334,8 @@ export const BagFooter = ({ setModalIsOpen, eventProperties }: BagFooterProps) =
   const { allowance, isAllowancePending, isApprovalLoading, updateAllowance } = usePermit2Approval(
     trade?.inputAmount.currency.isToken ? (trade?.inputAmount as CurrencyAmount<Token>) : undefined,
     maximumAmountIn,
-    shouldUsePayWithAnyToken
+    shouldUsePayWithAnyToken,
+    true
   )
   usePayWithAnyTokenSwap(trade, allowance, allowedSlippage)
   const priceImpact = usePriceImpact(trade)

src/nft/hooks/usePermit2Approval.test.ts

+import { renderHook } from '@testing-library/react'
+import { CurrencyAmount } from '@uniswap/sdk-core'
+import { USDC_MAINNET } from '@uniswap/smart-order-router'
+import usePermit2Allowance, { AllowanceState } from 'hooks/usePermit2Allowance'
+
+import usePermit2Approval from './usePermit2Approval'
+
+const USDCAmount = CurrencyAmount.fromRawAmount(USDC_MAINNET, '10000')
+const NFT_UNIVERSAL_ROUTER_MAINNET_ADDRESS = '0x4c60051384bd2d3c01bfc845cf5f4b44bcbe9de5'
+
+jest.mock('@web3-react/core', () => {
+  return {
+    useWeb3React: () => ({
+      chainId: 1,
+    }),
+  }
+})
+jest.mock('hooks/usePermit2Allowance')
+
+const mockUsePermit2Allowance = usePermit2Allowance as jest.MockedFunction<typeof usePermit2Allowance>
+
+describe('usePermit2Approval', () => {
+  it('sets spender of the correct UR contract from NFT side', async () => {
+    mockUsePermit2Allowance.mockReturnValue({ state: AllowanceState.LOADING })
+    renderHook(() => usePermit2Approval(USDCAmount, undefined, true, true))
+    expect(mockUsePermit2Allowance).toHaveBeenCalledWith(USDCAmount, NFT_UNIVERSAL_ROUTER_MAINNET_ADDRESS)
+  })
+})

src/nft/hooks/usePermit2Approval.ts

@@ -7,16 +7,24 @@ import usePermit2Allowance, { AllowanceState } from 'hooks/usePermit2Allowance'
 import { useCallback, useMemo, useState } from 'react'
 import invariant from 'tiny-invariant'
 
+// TODO: This should be removed when the sdk is updated to include the new UR address
+const NFT_UNIVERSAL_ROUTER_MAINNET_ADDRESS = '0x4c60051384bd2d3c01bfc845cf5f4b44bcbe9de5'
+
 export default function usePermit2Approval(
   amount?: CurrencyAmount<Token>,
   maximumAmount?: CurrencyAmount<Token>,
-  enabled?: boolean
+  enabled?: boolean,
+  shouldUseNftRouter?: boolean
 ) {
   const { chainId } = useWeb3React()
 
   const allowance = usePermit2Allowance(
     enabled ? maximumAmount ?? (amount?.currency.isToken ? (amount as CurrencyAmount<Token>) : undefined) : undefined,
-    enabled && chainId ? UNIVERSAL_ROUTER_ADDRESS(chainId) : undefined
+    enabled && chainId
+      ? shouldUseNftRouter && chainId === 1
+        ? NFT_UNIVERSAL_ROUTER_MAINNET_ADDRESS
+        : UNIVERSAL_ROUTER_ADDRESS(chainId)
+      : undefined
   )
   const isApprovalLoading = allowance.state === AllowanceState.REQUIRED && allowance.isApprovalLoading
   const [isAllowancePending, setIsAllowancePending] = useState(false)