forc-wallet 0.14.0

use crate::{
    DEFAULT_CACHE_ACCOUNTS,
    account::derive_and_cache_addresses,
    utils::{
        ensure_no_wallet_exists, request_new_password, write_wallet_from_mnemonic_and_password,
    },
};
use anyhow::{Result, bail};
use clap::Args;
use fuels::{accounts::signers::derivation::DEFAULT_DERIVATION_PATH, crypto::SecretKey};
use std::io::stdin;

#[derive(Debug, Args)]
pub struct Import {
    /// Forces wallet creation, removing any existing wallet file
    #[clap(short, long)]
    pub force: bool,
    /// How many accounts to cache by default (Default 10)
    #[clap(short, long)]
    pub cache_accounts: Option<usize>,
}

/// Check if given mnemonic is valid by trying to create a [SecretKey] from it
fn check_mnemonic(mnemonic: &str) -> Result<()> {
    // Check users's phrase by trying to create secret key from it
    if SecretKey::new_from_mnemonic_phrase_with_path(mnemonic, DEFAULT_DERIVATION_PATH).is_err() {
        bail!("Cannot generate a wallet from provided mnemonics, please check your mnemonic phrase")
    }
    Ok(())
}

pub async fn import_wallet_cli(ctx: &crate::CliContext, import: Import) -> Result<()> {
    ensure_no_wallet_exists(&ctx.wallet_path, import.force, stdin().lock())?;
    let mnemonic = rpassword::prompt_password("Please enter your mnemonic phrase: ")?;
    check_mnemonic(&mnemonic)?;
    let password = request_new_password();
    write_wallet_from_mnemonic_and_password(&ctx.wallet_path, &mnemonic, &password)?;
    derive_and_cache_addresses(
        ctx,
        &mnemonic,
        0..import.cache_accounts.unwrap_or(DEFAULT_CACHE_ACCOUNTS),
    )
    .await?;
    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::utils::test_utils::TEST_MNEMONIC;

    #[test]
    fn check_mnemonic_should_succeed() {
        assert!(check_mnemonic(TEST_MNEMONIC).is_ok())
    }

    #[test]
    fn check_mnemonic_should_fail() {
        let invalid_mnemonic = "this is an invalid mnemonic";
        assert!(check_mnemonic(invalid_mnemonic).is_err())
    }
}