miden-client-cli 0.14.5

use std::fs::{self, File};
use std::io::Read;
use std::path::PathBuf;

use miden_client::account::{AccountFile, AccountId};
use miden_client::keystore::Keystore;
use miden_client::note::NoteFile;
use miden_client::utils::Deserializable;
use miden_client::{Client, ClientError};
use tracing::info;

use crate::commands::account::set_default_account_if_unset;
use crate::errors::CliError;
use crate::{FilesystemKeyStore, Parser};

#[derive(Debug, Parser, Clone)]
#[command(about = "Import notes or accounts")]
pub struct ImportCmd {
    /// Paths to the files that contains the account/note data.
    #[arg()]
    filenames: Vec<PathBuf>,
    /// Only relevant for accounts. If set, the account will be overwritten if it already exists.
    #[arg(short, long, default_value_t = false)]
    overwrite: bool,
}

impl ImportCmd {
    pub async fn execute<AUTH: Keystore + Sync + 'static>(
        &self,
        mut client: Client<AUTH>,
        keystore: FilesystemKeyStore,
    ) -> Result<(), CliError> {
        validate_paths(&self.filenames)?;
        for filename in &self.filenames {
            let note_file = read_note_file(filename.clone());

            if let Ok(note_file) = note_file {
                let note_id = client.import_notes(&[note_file]).await?[0];
                println!("Successfully imported note {}", note_id.to_hex());
            } else {
                info!(
                    "Attempting to import account data from {}...",
                    fs::canonicalize(filename)?.as_path().display()
                );
                let account_file = AccountFile::read(filename)?;
                let account_id =
                    import_account(&mut client, &keystore, account_file, self.overwrite).await?;

                println!("Successfully imported account {account_id}");

                if account_id.is_regular_account() {
                    set_default_account_if_unset(&mut client, account_id).await?;
                }
            }
        }
        Ok(())
    }
}

// IMPORT ACCOUNT
// ================================================================================================

/// Imports an account file to the client.
///
/// This implies:
///
/// - Reading all secret keys, and importing them to the CLI keystore with account association
/// - Adding the [account][`miden_client::account::Account`] to the client
async fn import_account<AUTH>(
    client: &mut Client<AUTH>,
    keystore: &FilesystemKeyStore,
    account_file: AccountFile,
    overwrite: bool,
) -> Result<AccountId, CliError> {
    let account_id = account_file.account.id();
    let AccountFile { account, auth_secret_keys } = account_file;

    for key in auth_secret_keys {
        // Use the Keystore trait method which handles both key storage and account association
        keystore.add_key(&key, account_id).await.map_err(CliError::KeyStore)?;
    }

    client.add_account(&account, overwrite).await?;

    Ok(account_id)
}

// IMPORT NOTE
// ================================================================================================

fn read_note_file(filename: PathBuf) -> Result<NoteFile, CliError> {
    let mut contents = vec![];
    let mut _file = File::open(filename).and_then(|mut f| f.read_to_end(&mut contents))?;

    NoteFile::read_from_bytes(&contents)
        .map_err(|err| ClientError::DataDeserializationError(err).into())
}

// HELPERS
// ================================================================================================

/// Checks that all files exist, otherwise returns an error. It also ensures that all files have a
/// specific extension.
fn validate_paths(paths: &[PathBuf]) -> Result<(), CliError> {
    let invalid_path = paths.iter().find(|path| !path.exists());

    if let Some(path) = invalid_path {
        Err(CliError::Input(format!("The path `{}` does not exist", path.to_string_lossy())))
    } else {
        Ok(())
    }
}