forc-wallet 0.1.3

A forc plugin for generating or importing wallets using BIP39 phrases.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

use crate::{
    utils::{
        create_vault, default_vault_path, display_string_discreetly, request_new_password,
        save_phrase_to_disk, validate_vault_path,
    },
    Error,
};
use fuels::signers::wallet::generate_mnemonic_phrase;
use std::path::{Path, PathBuf};

fn init_wallet(path: &Path, password: &str) -> Result<String, Error> {
    // Generate mnemonic phrase
    let mnemonic = generate_mnemonic_phrase(&mut rand::thread_rng(), 24)?;
    // Encrypt and store it
    save_phrase_to_disk(path, &mnemonic, password);
    Ok(mnemonic)
}

pub(crate) fn init_wallet_cli(path_opt: Option<PathBuf>) -> Result<(), Error> {
    let path = path_opt.unwrap_or_else(default_vault_path);
    validate_vault_path(&path)?;
    create_vault(&path)?;
    let password = request_new_password();
    let mnemonic = init_wallet(&path, &password)?;
    let mnemonic_string = format!("Wallet mnemonic phrase: {}\n", mnemonic);
    display_string_discreetly(
        &mnemonic_string,
        "### Do not share or lose this mnemonic phrase! Press any key to complete. ###",
    )?;
    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::utils::test_utils::{with_tmp_folder, TEST_PASSWORD};
    use fuels::signers::WalletUnlocked;
    use serial_test::serial;

    #[serial]
    #[test]
    fn initialize_wallet() {
        with_tmp_folder(|tmp_folder| {
            let mnemonic = init_wallet(tmp_folder, TEST_PASSWORD).unwrap();
            let wallet_success = WalletUnlocked::new_from_mnemonic_phrase(&mnemonic, None).is_ok();
            assert!(wallet_success)
        })
    }
}