add to (#70)

Author: anna-carroll

.gas-snapshot

@@ -1,20 +1,20 @@
 OrderOriginPermit2Test:test_fillPermit2() (gas: 225289)
 OrderOriginPermit2Test:test_fillPermit2_multi() (gas: 1019134)
-OrderOriginPermit2Test:test_initiatePermit2() (gas: 235752)
-OrderOriginPermit2Test:test_initiatePermit2_multi() (gas: 989274)
-OrdersTest:test_fill_ERC20() (gas: 70537)
-OrdersTest:test_fill_ETH() (gas: 68498)
-OrdersTest:test_fill_both() (gas: 166773)
-OrdersTest:test_fill_multiETH() (gas: 132119)
-OrdersTest:test_fill_underflowETH() (gas: 115403)
+OrderOriginPermit2Test:test_initiatePermit2() (gas: 235774)
+OrderOriginPermit2Test:test_initiatePermit2_multi() (gas: 992096)
+OrdersTest:test_fill_ERC20() (gas: 70559)
+OrdersTest:test_fill_ETH() (gas: 68520)
+OrdersTest:test_fill_both() (gas: 166795)
+OrdersTest:test_fill_multiETH() (gas: 132141)
+OrdersTest:test_fill_underflowETH() (gas: 115425)
 OrdersTest:test_initiate_ERC20() (gas: 81636)
 OrdersTest:test_initiate_ETH() (gas: 45150)
 OrdersTest:test_initiate_both() (gas: 118911)
 OrdersTest:test_initiate_multiERC20() (gas: 722642)
 OrdersTest:test_initiate_multiETH() (gas: 75538)
 OrdersTest:test_orderExpired() (gas: 28106)
-OrdersTest:test_sweepERC20() (gas: 60491)
-OrdersTest:test_sweepETH() (gas: 82186)
+OrdersTest:test_sweepERC20() (gas: 59721)
+OrdersTest:test_sweepETH() (gas: 82348)
 OrdersTest:test_underflowETH() (gas: 63690)
 PassagePermit2Test:test_disallowedEnterPermit2() (gas: 699630)
 PassagePermit2Test:test_enterTokenPermit2() (gas: 145449)

src/orders/OrderOrigin.sol

@@ -74,17 +74,14 @@ abstract contract OrderOrigin is IOrders, OrdersPermit2 {
     /// @param token - The token to transfer.
     /// @custom:emits Sweep
     /// @custom:reverts OnlyBuilder if called by non-block builder
-    function sweep(address recipient, address token) external {
+    function sweep(address recipient, address token, uint256 amount) external {
         // send ETH or tokens
-        uint256 balance;
         if (token == address(0)) {
-            balance = address(this).balance;
-            payable(recipient).transfer(balance);
+            payable(recipient).transfer(amount);
         } else {
-            balance = IERC20(token).balanceOf(address(this));
-            IERC20(token).transfer(recipient, balance);
+            IERC20(token).transfer(recipient, amount);
         }
-        emit Sweep(recipient, token, balance);
+        emit Sweep(recipient, token, amount);
     }
 
     /// @notice Transfer the Order inputs to this contract, where they can be collected by the Order filler via `sweep`.

test/Orders.t.sol

@@ -157,7 +157,7 @@ contract OrdersTest is Test {
         // sweep ETH
         vm.expectEmit();
         emit Sweep(recipient, address(0), amount);
-        target.sweep(recipient, address(0));
+        target.sweep(recipient, address(0), amount);
 
         assertEq(recipient.balance, amount);
     }
@@ -173,7 +173,7 @@ contract OrdersTest is Test {
         vm.expectEmit();
         emit Sweep(recipient, token, amount);
         vm.expectCall(token, abi.encodeWithSelector(ERC20.transfer.selector, recipient, amount));
-        target.sweep(recipient, token);
+        target.sweep(recipient, token, amount);
     }
 
     function test_fill_ETH() public {