logos-execution-zone/wallet/src/token_transfers/deshielded.rs

use common::{ExecutionFailureKind, sequencer_client::json::SendTxResponse};
use k256::elliptic_curve::rand_core::{OsRng, RngCore};
use nssa::Address;
use nssa_core::{SharedSecretKey, encryption::EphemeralPublicKey};

use crate::WalletCore;

impl WalletCore {
    pub async fn send_deshielded_native_token_transfer(
        &self,
        from: Address,
        to: Address,
        balance_to_move: u128,
    ) -> Result<(SendTxResponse, nssa_core::SharedSecretKey), ExecutionFailureKind> {
        let Some((from_keys, mut from_acc)) =
            self.storage.user_data.get_private_account(&from).cloned()
        else {
            return Err(ExecutionFailureKind::KeyNotFoundError);
        };

        let Ok(to_acc) = self.get_account(to).await else {
            return Err(ExecutionFailureKind::KeyNotFoundError);
        };

        if from_acc.balance >= balance_to_move {
            let program = nssa::program::Program::authenticated_transfer_program();

            from_acc.program_owner = program.id();

            let sender_commitment =
                nssa_core::Commitment::new(&from_keys.nullifer_public_key, &from_acc);

            let sender_pre = nssa_core::account::AccountWithMetadata {
                account: from_acc.clone(),
                is_authorized: true,
                account_id: (&from_keys.nullifer_public_key).into(),
            };
            let recipient_pre = nssa_core::account::AccountWithMetadata {
                account: to_acc.clone(),
                is_authorized: false,
                account_id: to,
            };

            //Move into different function
            let mut esk = [0; 32];
            OsRng.fill_bytes(&mut esk);
            let shared_secret = SharedSecretKey::new(&esk, &from_keys.incoming_viewing_public_key);
            let epk = EphemeralPublicKey::from_scalar(esk);

            let (output, proof) = nssa::privacy_preserving_transaction::circuit::execute_and_prove(
                &[sender_pre, recipient_pre],
                &nssa::program::Program::serialize_instruction(balance_to_move).unwrap(),
                &[1, 0],
                &[from_acc.nonce + 1],
                &[(from_keys.nullifer_public_key.clone(), shared_secret.clone())],
                &[(
                    from_keys.private_key_holder.nullifier_secret_key,
                    self.sequencer_client
                        .get_proof_for_commitment(sender_commitment)
                        .await
                        .unwrap()
                        .unwrap(),
                )],
                &program,
            )
            .unwrap();

            let message =
                nssa::privacy_preserving_transaction::message::Message::try_from_circuit_output(
                    vec![to],
                    vec![],
                    vec![(
                        from_keys.nullifer_public_key.clone(),
                        from_keys.incoming_viewing_public_key.clone(),
                        epk,
                    )],
                    output,
                )
                .unwrap();

            let witness_set =
                nssa::privacy_preserving_transaction::witness_set::WitnessSet::for_message(
                    &message,
                    proof,
                    &[],
                );

            let tx = nssa::privacy_preserving_transaction::PrivacyPreservingTransaction::new(
                message,
                witness_set,
            );

            Ok((
                self.sequencer_client.send_tx_private(tx).await?,
                shared_secret,
            ))
        } else {
            Err(ExecutionFailureKind::InsufficientFundsError)
        }
    }
}