Fix: don't burn the fee on ft_on_transfer (#168)

* burn only transfered amount * mint tokens on cliam fee
Near-One · Dec 26, 2024 · 3bf06d5 · 3bf06d5
1 parent 3d16aed
commit 3bf06d5
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/near/omni-bridge/src/lib.rs b/near/omni-bridge/src/lib.rs
@@ -590,12 +590,20 @@ impl Contract {
         );
 
         if fee > 0 {
-            PromiseOrValue::Promise(
-                ext_token::ext(token)
-                    .with_static_gas(FT_TRANSFER_GAS)
-                    .with_attached_deposit(ONE_YOCTO)
-                    .ft_transfer(fin_transfer.fee_recipient, U128(fee), None),
-            )
+            if self.deployed_tokens.contains(&token) {
+                PromiseOrValue::Promise(ext_token::ext(token).with_static_gas(MINT_TOKEN_GAS).mint(
+                    fin_transfer.fee_recipient,
+                    U128(fee),
+                    None,
+                ))
+            } else {
+                PromiseOrValue::Promise(
+                    ext_token::ext(token)
+                        .with_static_gas(FT_TRANSFER_GAS)
+                        .with_attached_deposit(ONE_YOCTO)
+                        .ft_transfer(fin_transfer.fee_recipient, U128(fee), None),
+                )
+            }
         } else {
             PromiseOrValue::Value(())
         }