Merge pull request #157 from crisdut/fix/rgb-inputs

Mark as rgb-consumer only transition inputs
RGB-WG · May 1, 2024 · 0f65b19 · 0f65b19
2 parents ce07eb7 + 326112d
commit 0f65b19
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/psbt/src/lib.rs b/psbt/src/lib.rs
@@ -77,10 +77,11 @@ impl RgbPsbt for Psbt {
             let contract_id = info.transition.contract_id;
             let mut inputs = info.inputs.into_inner();
             for input in self.inputs_mut() {
-                inputs.remove(&XChain::Bitcoin(input.prevout().outpoint()));
-                input
-                    .set_rgb_consumer(contract_id, info.id)
-                    .map_err(|_| EmbedError::PsbtRepeatedInputs)?;
+                if inputs.remove(&XChain::Bitcoin(input.prevout().outpoint())) {
+                    input
+                        .set_rgb_consumer(contract_id, info.id)
+                        .map_err(|_| EmbedError::PsbtRepeatedInputs)?;
+                }
             }
             if !inputs.is_empty() {
                 return Err(EmbedError::AbsentInputs);