Add special validation when JCLI TX builder finds owned stake delegation

Author: CodeSandwich

jcli/src/jcli_app/transaction/mod.rs

@@ -131,6 +131,11 @@ custom_error! { pub Error
     ExpectingOnlyOneSigningKey { got: usize }
         = "expecting only one signing keys but got {got}",
     CertificateError { error: certificate::Error } = "certificate error {error}",
+
+    TxWithOwnerStakeDelegationMultiInputs { inputs: usize }
+        = "transaction has owner stake delegation, but has {inputs} inputs, should have 1",
+    TxWithOwnerStakeDelegationHasUtxoInput = "transaction has owner stake delegation, but has UTxO input",
+    TxWithOwnerStakeDelegationHasOutputs = "transaction has owner stake delegation, but has outputs",
 }
 
 /*

jcli/src/jcli_app/transaction/staging.rs

@@ -192,7 +192,7 @@ impl Staging {
         payload: &P,
         fee_algorithm: &FA,
         output_policy: chain::transaction::OutputPolicy,
-    ) -> Result<chain::transaction::Balance, Error>
+    ) -> Result<Balance, Error>
     where
         FA: FeeAlgorithm,
         P: Payload,
@@ -215,7 +215,7 @@ impl Staging {
         &mut self,
         fee_algorithm: &FA,
         output_policy: chain::transaction::OutputPolicy,
-    ) -> Result<chain::transaction::Balance, Error>
+    ) -> Result<Balance, Error>
     where
         FA: FeeAlgorithm,
     {
@@ -241,7 +241,24 @@ impl Staging {
                     self.finalize_payload(&c, fee_algorithm, output_policy)
                 }
                 Certificate::OwnerStakeDelegation(c) => {
-                    self.finalize_payload(&c, fee_algorithm, output_policy)
+                    let balance = self.finalize_payload(&c, fee_algorithm, output_policy)?;
+                    match self.inputs() {
+                        [input] => match input.input {
+                            interfaces::TransactionInputType::Account(_) => (),
+                            interfaces::TransactionInputType::Utxo(_, _) => {
+                                return Err(Error::TxWithOwnerStakeDelegationHasUtxoInput)
+                            }
+                        },
+                        inputs @ _ => {
+                            return Err(Error::TxWithOwnerStakeDelegationMultiInputs {
+                                inputs: inputs.len(),
+                            })
+                        }
+                    };
+                    if self.outputs().is_empty() == false {
+                        return Err(Error::TxWithOwnerStakeDelegationHasOutputs);
+                    }
+                    Ok(balance)
                 }
             },
         }