lssa/wallet/src/program_facades/native_token_transfer/public.rs

use common::{HashType, transaction::NSSATransaction};
use keycard_wallet::KeycardWallet;
use nssa::{
    AccountId, PublicTransaction,
    program::Program,
    public_transaction::{Message, WitnessSet},
};
use sequencer_service_rpc::RpcClient as _;

use super::NativeTokenTransfer;
use crate::{ExecutionFailureKind, WalletCore};

impl NativeTokenTransfer<'_> {
    pub async fn send_public_transfer(
        &self,
        from: AccountId,
        to: AccountId,
        balance_to_move: u128,
        pin: &Option<String>,
        key_path: &Option<String>,
    ) -> Result<HashType, ExecutionFailureKind> {
        let balance = self
            .0
            .get_account_balance(from)
            .await
            .map_err(ExecutionFailureKind::SequencerError)?;

        if balance >= balance_to_move {
            let account_ids = vec![from, to];
            let program_id = Program::authenticated_transfer_program().id();

            let mut sign_ids = Vec::new();
            sign_ids.push(from);

            let mut nonces = self
                .0
                .get_accounts_nonces(vec![from])
                .await
                .map_err(ExecutionFailureKind::SequencerError)?;
            let to_signing_key = self.0.storage.user_data.get_pub_account_signing_key(to);
            if let Some(_to_signing_key) = to_signing_key {
                sign_ids.push(to);
                let to_nonces = self
                    .0
                    .get_accounts_nonces(vec![to])
                    .await
                    .map_err(ExecutionFailureKind::SequencerError)?;
                nonces.extend(to_nonces);
            } else {
                println!(
                    "Receiver's account ({to}) private key not found in wallet. Proceeding with only sender's key."
                );
            }

            let message =
                Message::try_new(program_id, account_ids, nonces, balance_to_move).unwrap();

            let witness_set = if pin.is_none() {
                WalletCore::sign_public_message(self.0, &message, &sign_ids)
                    .expect("Expect a valid signature")
            } else {
                let pub_key = KeycardWallet::get_public_key_for_path_with_connect(
                    pin.as_ref().expect("Expect a pin as a String."),
                    key_path.as_ref().expect("Expect a key path String."),
                );
                let signature = KeycardWallet::sign_message_for_path_with_connect(
                    pin.as_ref().expect("Expect a pin as a String."),
                    key_path.as_ref().expect("Expect a key path String."),
                    &message.hash_message(),
                )
                .expect("Expect valid signature");
                WitnessSet::from_list(&[signature], &[pub_key])
            };

            let tx = PublicTransaction::new(message, witness_set);

            Ok(self
                .0
                .sequencer_client
                .send_transaction(NSSATransaction::Public(tx))
                .await?)
        } else {
            Err(ExecutionFailureKind::InsufficientFundsError)
        }
    }

    pub async fn register_account(
        &self,
        from: AccountId,
        pin: &Option<String>,      // Used by Keycard.
        key_path: &Option<String>, // Used by Keycard.
    ) -> Result<HashType, ExecutionFailureKind> {
        let nonces = self
            .0
            .get_accounts_nonces(vec![from])
            .await
            .map_err(ExecutionFailureKind::SequencerError)?;

        let instruction: u128 = 0;
        let account_ids = vec![from];
        let program_id = Program::authenticated_transfer_program().id();
        let message = Message::try_new(program_id, account_ids, nonces, instruction)
            .expect("Expect a valid Message");

        let witness_set = if pin.is_none() {
            let signing_key = self.0.storage.user_data.get_pub_account_signing_key(from);

            let Some(signing_key) = signing_key else {
                return Err(ExecutionFailureKind::KeyNotFoundError);
            };

            WitnessSet::for_message(&message, &[signing_key])
        } else {
            let pub_key = KeycardWallet::get_public_key_for_path_with_connect(
                pin.as_ref().expect("Expect a pin as a String."),
                key_path.as_ref().expect("Expect a key path String."),
            );

            let signature = KeycardWallet::sign_message_for_path_with_connect(
                pin.as_ref().as_ref().expect("Expect a pin as a String."),
                key_path.as_ref().expect("Expect a key path String."),
                &message.hash_message(),
            )
            .expect("Expect a valid Signature.");
            WitnessSet::from_list(&[signature], &[pub_key])
        };

        let tx = PublicTransaction::new(message, witness_set);

        Ok(self
            .0
            .sequencer_client
            .send_transaction(NSSATransaction::Public(tx))
            .await?)
    }
}