Merge pull request #925 from multiversx/simple-lock-exit-farm-with-pe…

…nalty-fix Simple lock exit farm with penalty fix
multiversx · Jun 20, 2024 · 4320a90 · 4320a90
2 parents a639089 + 81be5bb
commit 4320a90
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/locked-asset/simple-lock/src/proxy_farm.rs b/locked-asset/simple-lock/src/proxy_farm.rs
@@ -214,7 +214,7 @@ pub trait ProxyFarmModule:
             farm_address,
             farm_proxy_token_attributes.farm_token_id,
             farm_proxy_token_attributes.farm_token_nonce,
-            payment.amount,
+            payment.amount.clone(),
             caller.clone(),
         );
         require!(
@@ -236,6 +236,15 @@ pub trait ProxyFarmModule:
             &lp_proxy_token_payment.amount,
         );
 
+        if payment.amount > lp_proxy_token_payment.amount {
+            let penalty_amount = &payment.amount - &lp_proxy_token_payment.amount;
+
+            self.lp_proxy_token().nft_burn(
+                farm_proxy_token_attributes.farming_token_locked_nonce,
+                &penalty_amount,
+            );
+        }
+
         if exit_farm_result.reward_tokens.amount > 0 {
             self.send().direct_esdt(
                 &caller,