paymaster: fix transfer_callback

multiversx · Nov 20, 2024 · 18a6ad3 · 18a6ad3
1 parent 648957e
commit 18a6ad3
Showing 1 changed file with 16 additions and 19 deletions.
diff --git a/contracts/paymaster/src/forward_call.rs b/contracts/paymaster/src/forward_call.rs
@@ -36,32 +36,29 @@ pub trait ForwardCall {
         payments: PaymentsVec<Self::Api>,
         #[call_result] result: ManagedAsyncCallResult<MultiValueEncoded<ManagedBuffer>>,
     ) -> MultiValueEncoded<ManagedBuffer> {
-        let back_transfers = self.blockchain().get_back_transfers();
-
-        // Send the original input tokens back to the original caller
-        if !back_transfers.esdt_payments.is_empty() {
-            self.tx()
-                .to(&original_caller)
-                .payment(&back_transfers.esdt_payments)
-                .transfer();
-        }
-        if back_transfers.total_egld_amount > 0 {
-            self.tx()
-                .to(&original_caller)
-                .egld(back_transfers.total_egld_amount)
-                .transfer();
-        }
-
         match result {
             ManagedAsyncCallResult::Ok(return_values) => {
+                let back_transfers = self.blockchain().get_back_transfers();
+
+                // Send the original input tokens back to the original caller
+                if !back_transfers.esdt_payments.is_empty() {
+                    self.tx()
+                        .to(&original_caller)
+                        .payment(&back_transfers.esdt_payments)
+                        .transfer();
+                }
+                if back_transfers.total_egld_amount > 0 {
+                    self.tx()
+                        .to(&original_caller)
+                        .egld(back_transfers.total_egld_amount)
+                        .transfer();
+                }
                 // Send the resulted tokens to the original caller
                 return_values
             }
             ManagedAsyncCallResult::Err(err) => {
-                let egld_value = self.call_value().egld_value();
                 // Send the resulted tokens to the original caller
-                self.tx().to(&original_caller).payment(&payments).transfer();
-                self.tx().to(&original_caller).egld(egld_value).transfer();
+                self.tx().to(&original_caller).payment(payments).transfer();
 
                 let mut err_result = MultiValueEncoded::new();
                 err_result.push(ManagedBuffer::new_from_bytes(ERR_CALLBACK_MSG));