lssa-zkvm-testing/risc0-selective-privacy-poc/examples/happy_path.rs

use core::{bytes_to_words, types::Address, visibility::AccountVisibility};

use nssa::program::PinataProgram;

use crate::mocked_components::sequencer::{print_accounts, MockedSequencer, PINATA_ADDRESS};
use crate::mocked_components::{client::MockedClient, USER_CLIENTS};

mod mocked_components;

fn main() {
    let mut sequencer = MockedSequencer::new();
    let addresses: [Address; 3] = USER_CLIENTS.map(|client| client.user_address());
    println!("📝 Initial balances");
    print_accounts(&sequencer, &[]);

    // A public execution of the Transfer Program
    // User1 sends 3 tokens to another account
    USER_CLIENTS[1]
        .transfer_public(&[0xdeadbeef; 8], 3, &mut sequencer)
        .unwrap();
    println!("📝 Balances after transfer");
    print_accounts(&sequencer, &[]);

    // A public execution of the Transfer Program
    // User1 sends 51 tokens to the piñata account
    USER_CLIENTS[1]
        .transfer_public(&PINATA_ADDRESS, 51, &mut sequencer)
        .unwrap();
    println!("📝 Balances after transfer");
    print_accounts(&sequencer, &[]);

    // A shielded execution of the Transfer Program
    // User0 shields 15 tokens to a new private account of User1
    let private_account_user_1 = USER_CLIENTS[0]
        .transfer_shielded(&addresses[1], 15, &mut sequencer)
        .unwrap();
    println!("📝 Balances after shielded execution");
    print_accounts(&sequencer, &[&private_account_user_1]);

    // A private execution of the Transfer Program
    // User1 uses it's private account to send 8 tokens to a new private account of User2
    let [private_account_user_1, private_account_user_2] = USER_CLIENTS[1]
        .transfer_private(private_account_user_1, &addresses[2], 8, &mut sequencer)
        .unwrap();
    println!("📝 Balances after private execution");
    print_accounts(&sequencer, &[&private_account_user_1, &private_account_user_2]);

    // A deshielded execution of the Transfer Program
    // User2 deshields 1 token to the public account of User0
    let private_account_user_2 = USER_CLIENTS[2]
        .transfer_deshielded(private_account_user_2, &addresses[0], 1, &mut sequencer)
        .unwrap();
    println!("📝 Balances after deshielded execution");
    print_accounts(&sequencer, &[&private_account_user_1, &private_account_user_2]);

    // A public execution of the Piñata program
    // User2 claims the prize of the Piñata program to its public account
    let preimage = bytes_to_words(b"NSSA Selective privacy is great!").to_vec();
    sequencer
        .process_public_execution::<PinataProgram>(&[PINATA_ADDRESS, addresses[2]], preimage)
        .unwrap();
    println!("📝 Balances after public piñata execution");
    print_accounts(&sequencer, &[&private_account_user_1, &private_account_user_2]);

    // A shielded execution of the Piñata program
    // User1 claims the prize of the Piñata program to a new self-owned private account
    let another_private_account_user_1 = {
        // All of this is executed locally by the User1
        let pinata_account = sequencer.get_account(&PINATA_ADDRESS);
        let receiver_account = MockedClient::fresh_account_for_mint(USER_CLIENTS[1].user_address());
        let visibilities = [AccountVisibility::Public, AccountVisibility::Private(None)];
        let preimage = bytes_to_words(b"NSSA Selective privacy is great!").to_vec();

        let private_outputs = MockedClient::prove_and_send_to_sequencer::<PinataProgram>(
            &[pinata_account, receiver_account],
            preimage,
            &visibilities,
            sequencer.get_commitment_tree_root(),
            &mut sequencer,
        )
        .unwrap();
        let [private_account_user_1] = private_outputs.try_into().unwrap();
        private_account_user_1
    };
    println!("📝 Balances after private piñata execution");
    print_accounts(
        &sequencer,
        &[
            &private_account_user_1,
            &private_account_user_2,
            &another_private_account_user_1,
        ],
    );

    println!("Ok!");
}