From a0235bd6a9179d244754ae44d14f2dcb2adef8f5 Mon Sep 17 00:00:00 2001
From: mejango <mejango@protonmail.com>
Date: Fri, 8 Dec 2023 17:58:41 -0500
Subject: [PATCH] process held fees on migrate

---
 src/JBMultiTerminal.sol                    | 85 ++++++++++++----------
 src/interfaces/terminal/IJBFeeTerminal.sol |  2 +-
 2 files changed, 49 insertions(+), 38 deletions(-)

diff --git a/src/JBMultiTerminal.sol b/src/JBMultiTerminal.sol
index 1db47c0c6..5ca185aca 100644
--- a/src/JBMultiTerminal.sol
+++ b/src/JBMultiTerminal.sol
@@ -476,6 +476,9 @@ contract JBMultiTerminal is JBPermissioned, Ownable, ERC2771Context, IJBMultiTer
             revert TERMINAL_TOKENS_INCOMPATIBLE();
         }
 
+        // Process any held fees.
+        _processHeldFeesOf(projectId, token);
+
         // Record the migration in the store.
         balance = STORE.recordTerminalMigration(projectId, token);
 
@@ -505,7 +508,8 @@ contract JBMultiTerminal is JBPermissioned, Ownable, ERC2771Context, IJBMultiTer
     /// @dev Only a project's owner, an operator with the `PROCESS_FEES` permission from that owner, or this terminal's
     /// owner can process held fees.
     /// @param projectId The ID of the project to process held fees for.
-    function processHeldFees(uint256 projectId, address token) external virtual override {
+    /// @param token The token to process held fees for.
+    function processHeldFeesOf(uint256 projectId, address token) external virtual override {
         // Enforce permissions.
         _requirePermissionAllowingOverride({
             account: PROJECTS.ownerOf(projectId),
@@ -514,42 +518,7 @@ contract JBMultiTerminal is JBPermissioned, Ownable, ERC2771Context, IJBMultiTer
             alsoGrantAccessIf: _msgSender() == owner()
         });
 
-        // Get a reference to the project's held fees.
-        JBFee[] memory heldFees = _heldFeesOf[projectId][token];
-
-        // Delete the held fees.
-        delete _heldFeesOf[projectId][token];
-
-        // Keep a reference to the amount.
-        uint256 amount;
-
-        // Keep a reference to the number of held fees.
-        uint256 numberOfHeldFees = heldFees.length;
-
-        // Keep a reference to the fee being iterated on.
-        JBFee memory heldFee;
-
-        // Keep a reference to the terminal that'll receive the fees.
-        IJBTerminal feeTerminal = DIRECTORY.primaryTerminalOf(_FEE_BENEFICIARY_PROJECT_ID, token);
-
-        // Process each fee.
-        for (uint256 i; i < numberOfHeldFees; ++i) {
-            // Keep a reference to the held fee being iterated on.
-            heldFee = heldFees[i];
-
-            // Get the fee amount.
-            amount = JBFees.feeAmountIn(heldFee.amount, FEE);
-
-            // Process the fee.
-            _processFee({
-                projectId: projectId,
-                token: token,
-                amount: heldFee.amount,
-                beneficiary: heldFee.beneficiary,
-                feeTerminal: feeTerminal,
-                wasHeld: true
-            });
-        }
+        _processHeldFeesOf(projectId, token);
     }
 
     /// @notice Sets an address as feeless or not feeless for this terminal.
@@ -1611,6 +1580,48 @@ contract JBMultiTerminal is JBPermissioned, Ownable, ERC2771Context, IJBMultiTer
         }
     }
 
+    /// @notice Process any fees that are being held for the project.
+    /// @param projectId The ID of the project to process held fees for.
+    /// @param token The token to process held fees for.
+    function _processHeldFeesOf(uint256 projectId, address token) private {
+        // Get a reference to the project's held fees.
+        JBFee[] memory heldFees = _heldFeesOf[projectId][token];
+
+        // Delete the held fees.
+        delete _heldFeesOf[projectId][token];
+
+        // Keep a reference to the amount.
+        uint256 amount;
+
+        // Keep a reference to the number of held fees.
+        uint256 numberOfHeldFees = heldFees.length;
+
+        // Keep a reference to the fee being iterated on.
+        JBFee memory heldFee;
+
+        // Keep a reference to the terminal that'll receive the fees.
+        IJBTerminal feeTerminal = DIRECTORY.primaryTerminalOf(_FEE_BENEFICIARY_PROJECT_ID, token);
+
+        // Process each fee.
+        for (uint256 i; i < numberOfHeldFees; ++i) {
+            // Keep a reference to the held fee being iterated on.
+            heldFee = heldFees[i];
+
+            // Get the fee amount.
+            amount = JBFees.feeAmountIn(heldFee.amount, FEE);
+
+            // Process the fee.
+            _processFee({
+                projectId: projectId,
+                token: token,
+                amount: heldFee.amount,
+                beneficiary: heldFee.beneficiary,
+                feeTerminal: feeTerminal,
+                wasHeld: true
+            });
+        }
+    }
+
     /// @notice Process a fee of the specified amount from a project.
     /// @param projectId The ID of the project paying the fee.
     /// @param token The token the fee is being paid in.
diff --git a/src/interfaces/terminal/IJBFeeTerminal.sol b/src/interfaces/terminal/IJBFeeTerminal.sol
index 4a5dc5c09..6f003b6b6 100644
--- a/src/interfaces/terminal/IJBFeeTerminal.sol
+++ b/src/interfaces/terminal/IJBFeeTerminal.sol
@@ -49,7 +49,7 @@ interface IJBFeeTerminal is IJBTerminal {
 
     function isFeelessAddress(address account) external view returns (bool);
 
-    function processHeldFees(uint256 projectId, address token) external;
+    function processHeldFeesOf(uint256 projectId, address token) external;
 
     function setFeelessAddress(address account, bool flag) external;
 }