[TokenPaymaster] Support chains without WETH9

For chains like Celo we can swap directly to it without unwrapping
thirdweb-dev · Oct 30, 2024 · 91dbdaa · 91dbdaa
1 parent cf51bfa
commit 91dbdaa
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/contracts/prebuilts/account/utils/UniswapHelper.sol b/contracts/prebuilts/account/utils/UniswapHelper.sol
@@ -19,6 +19,7 @@ abstract contract UniswapHelper {
         uint256 minSwapAmount;
         uint24 uniswapPoolFee;
         uint8 slippage;
+        bool wethIsNativeAsset;
     }
 
     /// @notice The Uniswap V3 SwapRouter contract
@@ -85,6 +86,9 @@ abstract contract UniswapHelper {
     }
 
     function unwrapWeth(uint256 amount) internal {
+        if (uniswapHelperConfig.wethIsNativeAsset) {
+            return;
+        }
         IPeripheryPayments(address(uniswap)).unwrapWETH9(amount, address(this));
     }
 

diff --git a/src/test/smart-wallet/token-paymaster/TokenPaymaster.t.sol b/src/test/smart-wallet/token-paymaster/TokenPaymaster.t.sol
@@ -115,7 +115,8 @@ contract TokenPaymasterTest is BaseTest {
         UniswapHelper.UniswapHelperConfig memory uniswapHelperConfig = UniswapHelper.UniswapHelperConfig({
             minSwapAmount: 1,
             slippage: 5,
-            uniswapPoolFee: 3
+            uniswapPoolFee: 3,
+            wethIsNativeAsset: false
         });
 
         paymaster = new TokenPaymaster(