feat: Support SettleAndMintRefund in periphery

.forge-snapshots/BinFungiblePositionManager_AddLiquidityTest#testAddLiquidityOutsideActiveId_ExistingId.snap

@@ -1 +1 @@
-79699
+80435

.forge-snapshots/BinFungiblePositionManager_AddLiquidityTest#testAddLiquidityOutsideActiveId_NewId.snap

@@ -1 +1 @@
-627067
+627803

.forge-snapshots/BinFungiblePositionManager_AddLiquidityTest#testAddLiquidityWithActiveId.snap

@@ -1 +1 @@
-918147
+919685

.forge-snapshots/BinFungiblePositionManager_AddLiquidityTest#testAddLiquidityWithActiveId_WithHook.snap

@@ -1 +1 @@
-1268812
+1270351

.forge-snapshots/BinFungiblePositionManager_RemoveLiquidityTest#testRemoveLiquidityOutsideActiveId_ThreeBins.snap

@@ -1 +1 @@
-80824
+80759

.forge-snapshots/BinFungiblePositionManager_RemoveLiquidityTest#testRemoveLiquidityWithActiveId_ThreeBins.snap

@@ -1 +1 @@
-117826
+117761

.forge-snapshots/BinFungiblePositionManager_RemoveLiquidityTest#testRemoveLiquidity_Half.snap

@@ -1 +1 @@
-114631
+114566

.forge-snapshots/BinFungibleTokenTest#testBatchTransferFrom_FromOwner.snap

@@ -1 +1 @@
-29390
+9490

.forge-snapshots/BinFungibleTokenTest#testMint.snap

@@ -1 +1 @@
-43594
+43591

.forge-snapshots/BinSwapRouterTest#testExactInputSingle_DifferentRecipient.snap

@@ -1 +1 @@
-138752
+139491

.forge-snapshots/BinSwapRouterTest#testExactInputSingle_EthPool_SwapEthForToken.snap

@@ -1 +1 @@
-131484
+132217

.forge-snapshots/BinSwapRouterTest#testExactInputSingle_EthPool_SwapTokenForEth.snap

@@ -1 +1 @@
-136297
+137030

.forge-snapshots/BinSwapRouterTest#testExactInputSingle_SwapForY_1.snap

@@ -1 +1 @@
-132258
+132997

.forge-snapshots/BinSwapRouterTest#testExactInputSingle_SwapForY_2.snap

@@ -1 +1 @@
-132333
+133072

.forge-snapshots/BinSwapRouterTest#testExactInput_MultiHopDifferentRecipient.snap

@@ -1 +1 @@
-167590
+168327

.forge-snapshots/BinSwapRouterTest#testExactOutputSingle_DifferentRecipient.snap

@@ -1 +1 @@
-144076
+144815

.forge-snapshots/BinSwapRouterTest#testExactOutputSingle_SwapForY_1.snap

@@ -1 +1 @@
-137582
+138321

.forge-snapshots/BinSwapRouterTest#testExactOutputSingle_SwapForY_2.snap

@@ -1 +1 @@
-137669
+138408

.forge-snapshots/BinSwapRouterTest#testExactOutput_MultiHopDifferentRecipient.snap

@@ -1 +1 @@
-172111
+172844

.forge-snapshots/BinSwapRouterTest#testExactOutput_SingleHop.snap

@@ -1 +1 @@
-130040
+130773

.forge-snapshots/CLSwapRouterTest#ExactInput.snap

@@ -1 +1 @@
-250131
+250868

.forge-snapshots/CLSwapRouterTest#ExactInputSingle.snap

@@ -1 +1 @@
-180878
+181611

.forge-snapshots/CLSwapRouterTest#ExactOutput.snap

@@ -1 +1 @@
-256134
+256867

.forge-snapshots/CLSwapRouterTest#ExactOutputSingle.snap

@@ -1 +1 @@
-180439
+181172

.forge-snapshots/NonfungiblePositionManager#collect.snap

@@ -1 +1 @@
-199252
+199079

.forge-snapshots/NonfungiblePositionManager#decreaseLiquidity.snap

@@ -1 +1 @@
-75212
+75144

.forge-snapshots/NonfungiblePositionManager#increaseLiquidity.snap

@@ -1 +1 @@
-80908
+82440

.forge-snapshots/NonfungiblePositionManager#mint.snap

@@ -1 +1 @@
-624939
+626477

package.json

@@ -10,7 +10,7 @@
     "test": "forge test",
     "dev": "forge test -vvv -w",
     "snapshot": "rm -fr .forge-snapshots && forge test",
-    "prettier": "forge fmt contracts/ && forge fmt test/ && forge fmt script/",
+    "prettier": "forge fmt src/ && forge fmt test/ && forge fmt script/",
     "prettier-check": "forge fmt --check"
   }
 }

src/SwapRouterBase.sol

@@ -40,7 +40,7 @@ abstract contract SwapRouterBase is ISwapRouterBase {
 
     function _payAndSettle(Currency currency, address msgSender, int128 settleAmount) internal virtual {
         _pay(currency, msgSender, address(vault), uint256(uint128(settleAmount)));
-        vault.settle(currency);
+        vault.settleAndMintRefund(currency, msgSender);
     }
 
     function _pay(Currency currency, address payer, address recipient, uint256 amount) internal virtual;

src/pool-bin/BinFungiblePositionManager.sol

@@ -227,14 +227,14 @@ contract BinFungiblePositionManager is
     function _settleDeltas(address user, PoolKey memory poolKey, BalanceDelta delta) internal {
         if (delta.amount0() > 0) {
             pay(poolKey.currency0, user, address(vault), uint256(int256(delta.amount0())));
-            vault.settle(poolKey.currency0);
+            vault.settleAndMintRefund(poolKey.currency0, user);
         } else if (delta.amount0() < 0) {
             vault.take(poolKey.currency0, user, uint128(-delta.amount0()));
         }
 
         if (delta.amount1() > 0) {
             pay(poolKey.currency1, user, address(vault), uint256(int256(delta.amount1())));
-            vault.settle(poolKey.currency1);
+            vault.settleAndMintRefund(poolKey.currency1, user);
         } else if (delta.amount1() < 0) {
             vault.take(poolKey.currency1, user, uint128(-delta.amount1()));
         }

src/pool-cl/base/LiquidityManagement.sol

@@ -113,13 +113,13 @@ abstract contract LiquidityManagement is CLPeripheryImmutableState, PeripheryPay
     function settleDeltas(address sender, PoolKey memory poolKey, BalanceDelta delta) internal {
         if (delta.amount0() > 0) {
             pay(poolKey.currency0, sender, address(vault), uint256(int256(delta.amount0())));
-            vault.settle(poolKey.currency0);
+            vault.settleAndMintRefund(poolKey.currency0, sender);
         } else if (delta.amount0() < 0) {
             vault.take(poolKey.currency0, sender, uint128(-delta.amount0()));
         }
         if (delta.amount1() > 0) {
             pay(poolKey.currency1, sender, address(vault), uint256(int256(delta.amount1())));
-            vault.settle(poolKey.currency1);
+            vault.settleAndMintRefund(poolKey.currency1, sender);
         } else if (delta.amount1() < 0) {
             vault.take(poolKey.currency1, sender, uint128(-delta.amount1()));
         }

test/pool-bin/BinFungiblePositionManager_AddLiquidity.t.sol

@@ -538,4 +538,41 @@ contract BinFungiblePositionManager_AddLiquidityTest is Test, GasSnapshot, Liqui
             deadline: block.timestamp + 600
         });
     }
+
+    function testSettleAndMintRefund() public {
+        // transfer excess token to vault
+        uint256 excessTokenAmount = 1 ether;
+        address hacker = address(1);
+        token0.mint(hacker, excessTokenAmount);
+        vm.startPrank(hacker);
+        token0.transfer(address(vault), excessTokenAmount);
+        vm.stopPrank();
+
+        // pre-test, verify alice has 1e18 token0 and token1
+        token0.mint(alice, 1 ether);
+        token1.mint(alice, 1 ether);
+        assertEq(token0.balanceOf(alice), 1 ether);
+        assertEq(token1.balanceOf(alice), 1 ether);
+
+        vm.startPrank(alice);
+        uint24[] memory binIds = getBinIds(activeId, 3);
+        IBinFungiblePositionManager.AddLiquidityParams memory params;
+        params = _getAddParams(key1, binIds, 1 ether, 1 ether, activeId, alice);
+        (,, uint256[] memory tokenIds,) = binFungiblePositionManager.addLiquidity(params);
+
+        for (uint256 i; i < tokenIds.length; i++) {
+            (Currency curr0, Currency curr1, uint24 fee, uint24 binId) =
+                binFungiblePositionManager.positions(tokenIds[i]);
+            assertEq(Currency.unwrap(curr0), Currency.unwrap(key1.currency0));
+            assertEq(Currency.unwrap(curr1), Currency.unwrap(key1.currency1));
+            assertEq(fee, key1.fee);
+            assertEq(binId, binIds[i]);
+        }
+
+        // check currency balance in vault
+        {
+            uint256 currency0Balance = vault.balanceOf(alice, currency0);
+            assertEq(currency0Balance, excessTokenAmount, "Unexpected currency0 balance in vault");
+        }
+    }
 }

test/pool-bin/BinSwapRouter.t.sol

@@ -766,6 +766,47 @@ contract BinSwapRouterTest is Test, GasSnapshot, LiquidityParamsHelper {
         assertEq(token0.balanceOf(alice), abi.decode(result[0], (uint256)));
     }
 
+    function testSettleAndMintRefund() public {
+        // transfer excess token to vault
+        uint256 excessTokenAmount = 1 ether;
+        address hacker = address(1);
+        token0.mint(hacker, excessTokenAmount);
+        vm.startPrank(hacker);
+        token0.transfer(address(vault), excessTokenAmount);
+        vm.stopPrank();
+
+        vm.startPrank(alice);
+        token0.mint(alice, 1 ether);
+
+        ISwapRouterBase.PathKey[] memory path = new ISwapRouterBase.PathKey[](1);
+        path[0] = ISwapRouterBase.PathKey({
+            intermediateCurrency: Currency.wrap(address(token1)),
+            fee: key.fee,
+            hooks: key.hooks,
+            hookData: new bytes(0),
+            poolManager: key.poolManager,
+            parameters: key.parameters
+        });
+
+        uint256 amountOut = router.exactInput(
+            IBinSwapRouterBase.V4BinExactInputParams({
+                currencyIn: Currency.wrap(address(token0)),
+                path: path,
+                recipient: alice,
+                amountIn: 1 ether,
+                amountOutMinimum: 0
+            }),
+            block.timestamp + 60
+        );
+        assertEq(token1.balanceOf(alice), amountOut);
+
+        // check currency balance in vault
+        {
+            uint256 currency0Balance = vault.balanceOf(alice, Currency.wrap(address(token0)));
+            assertEq(currency0Balance, excessTokenAmount, "Unexpected currency0 balance in vault");
+        }
+    }
+
     // function testExactOutput_InsufficientAmountOut() public {
     //     //todo: in order to simulate this error, require
     //     // 1. hooks at beforeSwap do something funny on the pool resulting in actual amountOut lesser

test/pool-cl/CLSwapRouter.t.sol

@@ -689,6 +689,40 @@ contract CLSwapRouterTest is TokenFixture, Test, GasSnapshot {
         snapEnd();
     }
 
+    function testSettleAndMintRefund() external {
+        // transfer excess token to vault
+        uint256 excessTokenAmount = 1 ether;
+        address hacker = address(1);
+        MockERC20(Currency.unwrap(currency0)).mint(hacker, excessTokenAmount);
+        vm.startPrank(hacker);
+        MockERC20(Currency.unwrap(currency0)).transfer(address(vault), excessTokenAmount);
+        vm.stopPrank();
+
+        uint256 amountOut = router.exactInputSingle(
+            ICLSwapRouterBase.V4CLExactInputSingleParams({
+                poolKey: poolKey0,
+                zeroForOne: true,
+                recipient: makeAddr("recipient"),
+                amountIn: 0.01 ether,
+                amountOutMinimum: 0,
+                sqrtPriceLimitX96: 0,
+                hookData: new bytes(0)
+            }),
+            block.timestamp + 100
+        );
+
+        uint256 received = IERC20(Currency.unwrap(currency1)).balanceOf(makeAddr("recipient"));
+        assertEq(received, amountOut);
+        // considering slippage and fee, tolerance is 1%
+        assertApproxEqAbs(amountOut, 1 ether, amountOut / 100);
+
+        // check currency balance in vault
+        {
+            uint256 currency0Balance = vault.balanceOf(address(this), currency0);
+            assertEq(currency0Balance, excessTokenAmount, "Unexpected currency0 balance in vault");
+        }
+    }
+
     // allow refund of ETH
     receive() external payable {}
 }