execution-core: Hide transaction payload structs from API

circuits/license/Cargo.toml

@@ -5,7 +5,7 @@ edition = "2021"
 
 [dependencies]
 dusk-plonk = { version = "0.19", default-features = false, features = ["rkyv-impl", "alloc"] }
-zk-citadel = "0.12.0-rc"
+zk-citadel = "0.13.0-rc"
 
 [dev-dependencies]
 rand = "0.8"

contracts/license/Cargo.toml

@@ -24,7 +24,7 @@ license-circuits = { version = "0.1", path = "../../circuits/license" }
 rkyv = { version = "0.7", default-features = false, features = ["size_32"] }
 hex = "0.4"
 rand = "0.8"
-zk-citadel = "0.12.0-rc"
+zk-citadel = "0.13.0-rc"
 ff = { version = "0.13", default-features = false }
 
 [build-dependencies]

contracts/stake/Cargo.toml

@@ -16,6 +16,7 @@ rusk-abi = { version = "0.13.0-rc", path = "../../rusk-abi" }
 
 [dev-dependencies]
 rusk-profile = { version = "0.6", path = "../../rusk-profile" }
+once_cell = { version = "1.9" }
 rusk-abi = { version = "0.13.0-rc", path = "../../rusk-abi", default-features = false, features = ["host"] }
 execution-core = { version = "0.1.0", path = "../../execution-core", features = ["zk"] }
 rkyv = { version = "0.7", default-features = false, features = ["size_32"] }

contracts/stake/tests/common/mod.rs

@@ -6,4 +6,5 @@
 
 pub mod assert;
 pub mod init;
+pub mod prove;
 pub mod utils;

contracts/stake/tests/common/prove.rs

@@ -0,0 +1,132 @@
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at http://mozilla.org/MPL/2.0/.
+//
+// Copyright (c) DUSK NETWORK. All rights reserved.
+
+use dusk_bytes::Serializable;
+use execution_core::{
+    plonk::Prover as PlonkProver,
+    transfer::phoenix::{Prove, TxCircuit, TxCircuitVec, NOTES_TREE_DEPTH},
+};
+use once_cell::sync::Lazy;
+
+use rand::rngs::StdRng;
+use rand::SeedableRng;
+
+static PHOENIX_TX_1_2_PROVER: Lazy<PlonkProver> =
+    Lazy::new(|| fetch_prover("ExecuteCircuitOneTwo"));
+
+static PHOENIX_TX_2_2_PROVER: Lazy<PlonkProver> =
+    Lazy::new(|| fetch_prover("ExecuteCircuitTwoTwo"));
+
+static PHOENIX_TX_3_2_PROVER: Lazy<PlonkProver> =
+    Lazy::new(|| fetch_prover("ExecuteCircuitThreeTwo"));
+
+static PHOENIX_TX_4_2_PROVER: Lazy<PlonkProver> =
+    Lazy::new(|| fetch_prover("ExecuteCircuitFourTwo"));
+
+fn fetch_prover(circuit_name: &str) -> PlonkProver {
+    let circuit_profile = rusk_profile::Circuit::from_name(circuit_name)
+        .unwrap_or_else(|_| {
+            panic!("There should be circuit data stored for {}", circuit_name)
+        });
+    let pk = circuit_profile.get_prover().unwrap_or_else(|_| {
+        panic!("there should be a prover key stored for {}", circuit_name)
+    });
+
+    PlonkProver::try_from_bytes(pk).expect("Prover key is expected to by valid")
+}
+
+pub struct CachedProver();
+
+impl Prove for CachedProver {
+    fn prove(circuit: TxCircuitVec) -> Vec<u8> {
+        let rng = &mut StdRng::seed_from_u64(0xbeef);
+
+        // fetch the prover from the cache and crate the circuit
+        let (proof, _pi) = match circuit.input_notes_info.len() {
+            1 => PHOENIX_TX_1_2_PROVER
+                .prove(rng, &tx_circuit_1_2(circuit))
+                .expect("the circuit should be correct"),
+            2 => PHOENIX_TX_2_2_PROVER
+                .prove(rng, &tx_circuit_2_2(circuit))
+                .expect("the circuit should be correct"),
+            3 => PHOENIX_TX_3_2_PROVER
+                .prove(rng, &tx_circuit_3_2(circuit))
+                .expect("the circuit should be correct"),
+            4 => PHOENIX_TX_4_2_PROVER
+                .prove(rng, &tx_circuit_4_2(circuit))
+                .expect("the circuit should be correct"),
+            _ => panic!(
+                "The `TxCircuit` is only implemented for 1,
+            2, 3 or 4 input-notes."
+            ),
+        };
+        proof.to_bytes().to_vec()
+    }
+}
+
+fn tx_circuit_1_2(circuit: TxCircuitVec) -> TxCircuit<NOTES_TREE_DEPTH, 1> {
+    TxCircuit {
+        input_notes_info: circuit
+            .input_notes_info
+            .try_into()
+            .expect("There should be exactly one input"),
+        output_notes_info: circuit.output_notes_info,
+        payload_hash: circuit.payload_hash,
+        root: circuit.root,
+        deposit: circuit.deposit,
+        max_fee: circuit.max_fee,
+        sender_pk: circuit.sender_pk,
+        signatures: circuit.signatures,
+    }
+}
+
+fn tx_circuit_2_2(circuit: TxCircuitVec) -> TxCircuit<NOTES_TREE_DEPTH, 2> {
+    TxCircuit {
+        input_notes_info: circuit
+            .input_notes_info
+            .try_into()
+            .expect("There should be exactly two inputs"),
+        output_notes_info: circuit.output_notes_info,
+        payload_hash: circuit.payload_hash,
+        root: circuit.root,
+        deposit: circuit.deposit,
+        max_fee: circuit.max_fee,
+        sender_pk: circuit.sender_pk,
+        signatures: circuit.signatures,
+    }
+}
+
+fn tx_circuit_3_2(circuit: TxCircuitVec) -> TxCircuit<NOTES_TREE_DEPTH, 3> {
+    TxCircuit {
+        input_notes_info: circuit
+            .input_notes_info
+            .try_into()
+            .expect("There should be exactly three inputs"),
+        output_notes_info: circuit.output_notes_info,
+        payload_hash: circuit.payload_hash,
+        root: circuit.root,
+        deposit: circuit.deposit,
+        max_fee: circuit.max_fee,
+        sender_pk: circuit.sender_pk,
+        signatures: circuit.signatures,
+    }
+}
+
+fn tx_circuit_4_2(circuit: TxCircuitVec) -> TxCircuit<NOTES_TREE_DEPTH, 4> {
+    TxCircuit {
+        input_notes_info: circuit
+            .input_notes_info
+            .try_into()
+            .expect("There should be exactly four inputs"),
+        output_notes_info: circuit.output_notes_info,
+        payload_hash: circuit.payload_hash,
+        root: circuit.root,
+        deposit: circuit.deposit,
+        max_fee: circuit.max_fee,
+        sender_pk: circuit.sender_pk,
+        signatures: circuit.signatures,
+    }
+}