lssa/wallet/src/lib.rs

use std::{fs::File, io::Write, path::PathBuf, sync::Arc};

use common::{
    execution_input::PublicNativeTokenSend,
    sequencer_client::{json::SendTxResponse, SequencerClient},
    transaction::Transaction,
    ExecutionFailureKind,
};

use accounts::account_core::{address::AccountAddress, Account};
use anyhow::Result;
use chain_storage::NodeChainStore;
use common::transaction::TransactionBody;
use config::NodeConfig;
use log::info;
use sc_core::proofs_circuits::pedersen_commitment_vec;

use clap::{Parser, Subcommand};

use crate::helperfunctions::{
    fetch_config, fetch_persistent_accounts, get_home, produce_account_addr_from_hex,
};

pub const HOME_DIR_ENV_VAR: &str = "NSSA_WALLET_HOME_DIR";
pub const BLOCK_GEN_DELAY_SECS: u64 = 20;

pub mod chain_storage;
pub mod config;
pub mod helperfunctions;
pub mod requests_structs;

pub struct NodeCore {
    pub storage: NodeChainStore,
    pub node_config: NodeConfig,
    pub sequencer_client: Arc<SequencerClient>,
}

impl NodeCore {
    pub async fn start_from_config_update_chain(config: NodeConfig) -> Result<Self> {
        let client = Arc::new(SequencerClient::new(config.sequencer_addr.clone())?);

        let mut storage = NodeChainStore::new(config.clone())?;
        for acc in config.clone().initial_accounts {
            storage.acc_map.insert(acc.address, acc);
        }

        //Persistent accounts take precedence for initial accounts
        let persistent_accounts = fetch_persistent_accounts()?;
        for acc in persistent_accounts {
            storage.acc_map.insert(acc.address, acc);
        }

        Ok(Self {
            storage,
            node_config: config.clone(),
            sequencer_client: client.clone(),
        })
    }

    pub async fn get_roots(&self) -> [[u8; 32]; 2] {
        [
            self.storage
                .utxo_commitments_store
                .get_root()
                .unwrap_or([0; 32]),
            self.storage.pub_tx_store.get_root().unwrap_or([0; 32]),
        ]
    }

    pub async fn create_new_account(&mut self) -> AccountAddress {
        let account = Account::new();
        account.log();

        let addr = account.address;

        self.storage.acc_map.insert(account.address, account);

        addr
    }

    pub async fn send_public_native_token_transfer(
        &self,
        from: AccountAddress,
        nonce: u64,
        to: AccountAddress,
        balance_to_move: u64,
    ) -> Result<SendTxResponse, ExecutionFailureKind> {
        let tx_roots = self.get_roots().await;

        let public_context = self.storage.produce_context(from);

        let (tweak, secret_r, commitment) = pedersen_commitment_vec(
            //Will not panic, as public context is serializable
            public_context.produce_u64_list_from_context().unwrap(),
        );

        let sc_addr = hex::encode([0; 32]);

        let tx: TransactionBody =
            sc_core::transaction_payloads_tools::create_public_transaction_payload(
                serde_json::to_vec(&PublicNativeTokenSend {
                    from,
                    nonce,
                    to,
                    balance_to_move,
                })
                .unwrap(),
                commitment,
                tweak,
                secret_r,
                sc_addr,
            );
        tx.log();

        let account = self.storage.acc_map.get(&from);

        if let Some(account) = account {
            let key_to_sign_transaction = account.key_holder.get_pub_account_signing_key();

            let signed_transaction = Transaction::new(tx, key_to_sign_transaction);

            Ok(self
                .sequencer_client
                .send_tx(signed_transaction, tx_roots)
                .await?)
        } else {
            Err(ExecutionFailureKind::AmountMismatchError)
        }
    }

    ///Dumps all accounts from acc_map at `path`
    ///
    ///Currently storing everything in one file
    ///
    ///ToDo: extend storage
    pub fn store_present_accounts_at_path(&self, path: PathBuf) -> Result<PathBuf> {
        let dump_path = path.join("curr_accounts.json");

        let curr_accs: Vec<Account> = self
            .storage
            .acc_map
            .values()
            .cloned()
            .collect();
        let accs_serialized = serde_json::to_vec_pretty(&curr_accs)?;

        let mut acc_file = File::create(&dump_path).unwrap();
        acc_file.write_all(&accs_serialized).unwrap();

        Ok(dump_path)
    }

    ///Dumps all accounts from acc_map at `NSSA_WALLET_HOME_DIR`
    ///
    ///Currently storing everything in one file
    ///
    ///ToDo: extend storage
    pub fn store_present_accounts_at_home(&self) -> Result<PathBuf> {
        let home = get_home()?;
        self.store_present_accounts_at_path(home)
    }
}

///Represents CLI command for a wallet
#[derive(Subcommand, Debug, Clone)]
#[clap(about)]
pub enum Command {
    ///Send native token transfer from `from` to `to` for `amount`
    SendNativeTokenTransfer {
        ///from - valid 32 byte hex string
        #[arg(long)]
        from: String,
        ///to - valid 32 byte hex string
        #[arg(long)]
        to: String,
        ///amount - amount of balance to move
        #[arg(long)]
        amount: u64,
    },
    ///Dump accounts at destination
    DumpAccountsOnDisc {
        ///Dump path for accounts
        #[arg(short, long)]
        dump_path: PathBuf,
    },
}

///To execute commands, env var NSSA_WALLET_HOME_DIR must be set into directory with config
#[derive(Parser, Debug)]
#[clap(version, about)]
pub struct Args {
    /// Wallet command
    #[command(subcommand)]
    pub command: Command,
}

pub async fn execute_subcommand(command: Command) -> Result<()> {
    match command {
        Command::SendNativeTokenTransfer { from, to, amount } => {
            let node_config = fetch_config()?;

            let from = produce_account_addr_from_hex(from)?;
            let to = produce_account_addr_from_hex(to)?;

            let wallet_core = NodeCore::start_from_config_update_chain(node_config).await?;

            //ToDo: Nonce management
            let res = wallet_core
                .send_public_native_token_transfer(from, 0, to, amount)
                .await?;

            info!("Results of tx send is {res:#?}");
        }
        Command::DumpAccountsOnDisc { dump_path } => {
            let node_config = fetch_config()?;

            let wallet_core = NodeCore::start_from_config_update_chain(node_config).await?;

            wallet_core.store_present_accounts_at_path(dump_path.clone())?;

            info!("Accounts stored at path {dump_path:#?}");
        }
    }

    Ok(())
}