Merge pull request #1023 from solidsnakedev/main

Add wrap constructor to V1/V2 StakeCredential
aiken-lang · Sep 20, 2024 · 7155b4e · 7155b4e
2 parents 3fddab4 + 75b5332
commit 7155b4e
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/crates/uplc/src/tx/to_plutus_data.rs b/crates/uplc/src/tx/to_plutus_data.rs
@@ -872,9 +872,10 @@ impl<'a> ToPlutusData for WithWrappedTransactionId<'a, ScriptPurpose> {
             ScriptPurpose::Spending(out_ref, ()) => {
                 wrap_with_constr(1, WithWrappedTransactionId(out_ref).to_plutus_data())
             }
-            ScriptPurpose::Rewarding(stake_credential) => {
-                wrap_with_constr(2, stake_credential.to_plutus_data())
-            }
+            ScriptPurpose::Rewarding(stake_credential) => wrap_with_constr(
+                2,
+                WithWrappedStakeCredential(stake_credential).to_plutus_data(),
+            ),
             ScriptPurpose::Certifying(_, dcert) => {
                 wrap_with_constr(3, WithPartialCertificates(dcert).to_plutus_data())
             }
@@ -1243,6 +1244,12 @@ impl<'a> ToPlutusData for WithWrappedStakeCredential<'a, KeyValuePairs<Address,
     }
 }
 
+impl<'a> ToPlutusData for WithWrappedStakeCredential<'a, StakeCredential> {
+    fn to_plutus_data(&self) -> PlutusData {
+        wrap_with_constr(0, self.0.to_plutus_data())
+    }
+}
+
 impl ToPlutusData for Voter {
     fn to_plutus_data(&self) -> PlutusData {
         match self {