Support fee on transfer tokens

Signed-off-by: Francesco <[email protected]>
flood-protocol · Dec 18, 2024 · 813a1da · 813a1da
1 parent 6ff56b6
commit 813a1da
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/TokenFlow.sol b/src/TokenFlow.sol
@@ -52,9 +52,10 @@ contract TokenFlow is ITokenFlow {
 
      /// @inheritdoc ITokenFlow
     function moveOut(address token, uint128 amount) external requireScope(INTERNAL_SCOPE) {
-        TransientNetflows.add(token, int256(uint256(amount)));
-
+        uint balanceBefore = token.balanceOf(payer);
         token.safeTransferFrom(msg.sender, payer, amount);
+        uint received = token.balanceOf(payer) - balanceBefore;
+        TransientNetflows.add(token, int256(uint256(received)));
     }
 
     /// @inheritdoc ITokenFlow