hyli-contract-sdk 0.15.0

//! # Hyli Contract SDK
//!
//! This crate provides helper tools for writing smart contracts that run inside a Rust-based zkVM,
//! such as [Risc0](https://github.com/risc0/risc0) or [SP1](https://github.com/succinctlabs/sp1).
//!
//! ## Building a contract on Hyli
//!
//! To write a contract for Hyli:
//!
//! - Create a library crate that defines a struct implementing the [`ZkContract`] trait.
//! - Build a corresponding zkVM binary that runs your contract logic.
//!
//! See [`crate::guest`] for details on how to configure and run your contract inside the zkVM.
//!
//! You can use our app scaffold to get started:
//!
//! - [scaffold](https://github.com/hyli-org/app-scaffold/)
//! - [Quickstart documentation](https://docs.hyli.org/quickstart/)
//!
//! ## Contract interoperability
//!
//! If your contract needs to exchange data with others, refer to [`StructuredBlobData`].
//! More documentation about this will follow.

#![cfg_attr(not(test), no_std)]

extern crate alloc;

use alloc::string::String;
use alloc::vec::Vec;

pub mod caller;
pub mod guest;
#[cfg(feature = "smt")]
pub mod merkle_utils;
pub mod secp256k1;
pub mod utils;

use caller::ExecutionContext;
// re-export hyli-model
pub use hyli_model::*;

pub use hyli_model::utils as hyli_model_utils;

#[cfg(feature = "tracing")]
pub use tracing;

// Si la feature "tracing" est activée, on redirige vers `tracing::info!`
#[cfg(feature = "tracing")]
#[macro_export]
macro_rules! info {
    ($($arg:tt)*) => {
        $crate::tracing::info!($($arg)*);
    }
}

// Si la feature "tracing" est activée, on redirige vers `tracing::error!`
#[cfg(feature = "tracing")]
#[macro_export]
macro_rules! error {
    ($($arg:tt)*) => {
        $crate::tracing::error!($($arg)*);
    }
}

// Si la feature "tracing" n'est pas activée, on redirige vers la fonction env::log
#[cfg(all(not(feature = "tracing"), feature = "risc0"))]
#[macro_export]
macro_rules! info {
    ($($arg:tt)*) => {
        risc0_zkvm::guest::env::log(&format!($($arg)*));
    }
}

#[cfg(all(not(feature = "tracing"), not(feature = "risc0")))]
#[macro_export]
macro_rules! info {
    ($($arg:tt)*) => {
        // no_std: no-op; in test builds (which have std), print normally
        #[cfg(test)]
        println!($($arg)*);
    }
}

// Si la feature "tracing" n'est pas activée, on redirige vers la fonction env::log
#[cfg(all(not(feature = "tracing"), feature = "risc0"))]
#[macro_export]
macro_rules! error {
    ($($arg:tt)*) => {
        risc0_zkvm::guest::env::log(&format!($($arg)*));
    }
}

#[cfg(all(not(feature = "tracing"), not(feature = "risc0")))]
#[macro_export]
macro_rules! error {
    ($($arg:tt)*) => {
        // no_std: no-op; in test builds (which have std), print normally
        #[cfg(test)]
        println!($($arg)*);
    }
}

pub type RunResult = Result<(Vec<u8>, ExecutionContext, Vec<OnchainEffect>), String>;

/**
This trait is used to define the contract's entrypoint.
By using it and the [execute](function@crate::guest::execute) function, you let the sdk
generate for you the [HyliOutput] struct with correct fields.

The [Calldata] struct is built by the application backend and given as input to
the program that runs in the zkvm.

The calldata is generic to any contract, and holds all the blobs of the blob transaction
being proved. These blobs are stored as vec of bytes, so contract need to parse them into the
expected type. For this, it can call either [utils::parse_raw_calldata] or
[utils::parse_calldata]. Check the [utils] documentation for details on these functions.

## Example of execute implementation:

```rust
use hyli_contract_sdk::{StateCommitment, RunResult, ZkContract};
use hyli_contract_sdk::utils::parse_raw_calldata;
use hyli_model::Calldata;

use borsh::{BorshSerialize, BorshDeserialize};

struct MyContract{}
#[derive(BorshSerialize, BorshDeserialize)]
enum MyContractAction{
    DoSomething
}

impl ZkContract for MyContract {
    fn execute(&mut self, calldata: &Calldata) -> RunResult {
        let (action, exec_ctx) = parse_raw_calldata(calldata)?;

        let output = self.execute_action(action)?;

        Ok((output.into_bytes(), exec_ctx, vec![]))
    }
    fn commit(&self) -> StateCommitment {
        StateCommitment(vec![])
    }
}

impl MyContract {
    fn execute_action(&mut self, action: MyContractAction) -> Result<String, String> {
        /// Execute contract's logic
        Ok("Done.".to_string())
    }
}

```
*/
pub trait ZkContract {
    /// Entry point of the contract
    /// Execution is based solely on the contract's commitment metadata.
    /// Exemple: the merkle root for a contract's state based on a MerkleTrie. The execute function will only update the roothash of the trie.
    fn execute(&mut self, calldata: &Calldata) -> RunResult;

    fn commit(&self) -> StateCommitment;

    /// A function executed before the contract is executed.
    /// This might be used to do verifications on the state before validating calldatas.
    fn initialize(&mut self) -> Result<(), String> {
        Ok(())
    }
}

pub trait TransactionalZkContract
where
    Self: ZkContract,
{
    type State;
    fn initial_state(&self) -> Self::State;

    fn revert(&mut self, initial_state: Self::State);

    fn on_success(&mut self) -> StateCommitment {
        self.commit()
    }
}

pub trait FullStateRevert {}

impl<T> TransactionalZkContract for T
where
    T: FullStateRevert,
    Self: Sized + Clone + ZkContract,
{
    type State = Self;

    fn initial_state(&self) -> Self::State {
        self.clone()
    }

    fn revert(&mut self, initial_state: Self::State) {
        *self = initial_state;
    }
}

pub const fn to_u8_array(val: &[u32; 8]) -> [u8; 32] {
    [
        (val[0] & 0xFF) as u8,
        ((val[0] >> 8) & 0xFF) as u8,
        ((val[0] >> 16) & 0xFF) as u8,
        ((val[0] >> 24) & 0xFF) as u8,
        (val[1] & 0xFF) as u8,
        ((val[1] >> 8) & 0xFF) as u8,
        ((val[1] >> 16) & 0xFF) as u8,
        ((val[1] >> 24) & 0xFF) as u8,
        (val[2] & 0xFF) as u8,
        ((val[2] >> 8) & 0xFF) as u8,
        ((val[2] >> 16) & 0xFF) as u8,
        ((val[2] >> 24) & 0xFF) as u8,
        (val[3] & 0xFF) as u8,
        ((val[3] >> 8) & 0xFF) as u8,
        ((val[3] >> 16) & 0xFF) as u8,
        ((val[3] >> 24) & 0xFF) as u8,
        (val[4] & 0xFF) as u8,
        ((val[4] >> 8) & 0xFF) as u8,
        ((val[4] >> 16) & 0xFF) as u8,
        ((val[4] >> 24) & 0xFF) as u8,
        (val[5] & 0xFF) as u8,
        ((val[5] >> 8) & 0xFF) as u8,
        ((val[5] >> 16) & 0xFF) as u8,
        ((val[5] >> 24) & 0xFF) as u8,
        (val[6] & 0xFF) as u8,
        ((val[6] >> 8) & 0xFF) as u8,
        ((val[6] >> 16) & 0xFF) as u8,
        ((val[6] >> 24) & 0xFF) as u8,
        (val[7] & 0xFF) as u8,
        ((val[7] >> 8) & 0xFF) as u8,
        ((val[7] >> 16) & 0xFF) as u8,
        ((val[7] >> 24) & 0xFF) as u8,
    ]
}

const fn byte_to_u8(byte: u8) -> u8 {
    match byte {
        b'0'..=b'9' => byte - b'0',
        b'a'..=b'f' => byte - b'a' + 10,
        b'A'..=b'F' => byte - b'A' + 10,
        _ => 0,
    }
}

#[allow(
    clippy::indexing_slicing,
    reason = "const block, shouldn't be used at runtime."
)]
pub const fn str_to_u8(s: &str) -> [u8; 32] {
    let mut bytes = [0u8; 32];
    let chrs = s.as_bytes();
    let mut i = 0;
    while i < 32 {
        bytes[i] = (byte_to_u8(chrs[i * 2]) << 4) | byte_to_u8(chrs[i * 2 + 1]);
        i += 1;
    }
    bytes
}

#[cfg(test)]
mod tests {
    use super::*;
    use alloc::{format, string::ToString, vec};

    #[test]
    fn test_identity_from_string() {
        let identity_str = "test_identity".to_string();
        let identity = Identity::from(identity_str.clone());
        assert_eq!(identity.0, identity_str);
    }

    #[test]
    fn test_identity_from_str() {
        let identity_str = "test_identity";
        let identity = Identity::from(identity_str);
        assert_eq!(identity.0, identity_str.to_string());
    }

    #[test]
    fn test_txhash_from_string() {
        let txhash_str = "746573745f747868617368".to_string();
        let txhash = TxHash::from_hex(&txhash_str).expect("txhash hex");
        assert_eq!(txhash.0, b"test_txhash".to_vec());
    }

    #[test]
    fn test_txhash_from_str() {
        let txhash_str = "746573745f747868617368";
        let txhash = TxHash::from_hex(txhash_str).expect("txhash hex");
        assert_eq!(txhash.0, b"test_txhash".to_vec());
    }

    #[test]
    fn test_txhash_new() {
        let txhash = TxHash::new(b"test_txhash".to_vec());
        assert_eq!(txhash.0, b"test_txhash".to_vec());
    }

    #[test]
    fn test_blobindex_from_u32() {
        let index = 42;
        let blob_index = BlobIndex::from(index);
        assert_eq!(blob_index.0, index);
    }

    #[test]
    fn test_txhash_display() {
        let txhash_str = "746573745f747868617368";
        let txhash = TxHash::from_hex(txhash_str).expect("txhash hex");
        assert_eq!(format!("{txhash}"), txhash_str);
    }

    #[test]
    fn test_blobindex_display() {
        let index = 42;
        let blob_index = BlobIndex::from(index);
        assert_eq!(format!("{blob_index}"), index.to_string());
    }

    #[test]
    fn test_state_commitment_encoding() {
        let state_commitment = StateCommitment(vec![1, 2, 3, 4]);
        let encoded = borsh::to_vec(&state_commitment).expect("Failed to encode StateCommitment");
        let decoded: StateCommitment =
            borsh::from_slice(&encoded).expect("Failed to decode StateCommitment");
        assert_eq!(state_commitment, decoded);
    }

    #[test]
    fn test_identity_encoding() {
        let identity = Identity::new("test_identity");
        let encoded = borsh::to_vec(&identity).expect("Failed to encode Identity");
        let decoded: Identity = borsh::from_slice(&encoded).expect("Failed to decode Identity");
        assert_eq!(identity, decoded);
    }

    #[test]
    fn test_txhash_encoding() {
        let txhash = TxHash::from_hex("746573745f747868617368").expect("txhash hex");
        let encoded = borsh::to_vec(&txhash).expect("Failed to encode TxHash");
        let decoded: TxHash = borsh::from_slice(&encoded).expect("Failed to decode TxHash");
        assert_eq!(txhash, decoded);
    }

    #[test]
    fn test_blobindex_encoding() {
        let blob_index = BlobIndex(42);
        let encoded = borsh::to_vec(&blob_index).expect("Failed to encode BlobIndex");
        let decoded: BlobIndex = borsh::from_slice(&encoded).expect("Failed to decode BlobIndex");
        assert_eq!(blob_index, decoded);
    }

    #[test]
    fn test_blobdata_encoding() {
        let blob_data = BlobData(vec![1, 2, 3, 4]);
        let encoded = borsh::to_vec(&blob_data).expect("Failed to encode BlobData");
        let decoded: BlobData = borsh::from_slice(&encoded).expect("Failed to decode BlobData");
        assert_eq!(blob_data, decoded);
    }
}