ergo-rustkit 0.5.0

A SDK for building applications on the Ergo blockchain
Documentation 
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

use ergo_lib::wallet::{Wallet, secret_key::SecretKey, ext_secret_key::ExtSecretKey, mnemonic::Mnemonic, derivation_path::{DerivationPath, ChildIndexHardened, ChildIndexNormal}, ext_pub_key::ExtPubKey};
use ergotree_ir::{chain::{address::{Address, AddressEncoder, NetworkPrefix}, ergo_box::ErgoBox}};

use crate::{utils::format::remove_quotes, config::{file::Config}};


/// RustKit wallet
pub struct RustKitWallet {
    secret_key: SecretKey,
    pub index_0_address: String,
    pub wallet: Wallet,
    config: Config,
}

impl RustKitWallet {
    pub fn new(config: &Config) -> Self {
        let config: Config = config.to_owned();
        let mnemonic: String = config.get_mnemonic();
        let mnemonic_password: String = config.get_mnemonic_password();
        let seed = Mnemonic::to_seed(&mnemonic, &mnemonic_password);
        let extended_secret_key: ExtSecretKey = ExtSecretKey::derive_master(seed).unwrap();

        let path: DerivationPath = DerivationPath::new(ChildIndexHardened::from_31_bit(0).unwrap(), vec![ChildIndexNormal::normal(0).unwrap()]);
        let derived_extended_secret_key: ExtSecretKey = extended_secret_key.derive(path).unwrap();
        let secret_key: SecretKey = derived_extended_secret_key.secret_key();
        let public_key: ExtPubKey = derived_extended_secret_key.public_key().unwrap();
        let address: Address = Address::from(public_key);
        let wallet: Wallet = Wallet::from_secrets(vec![secret_key.clone()]);
        match config.network.as_str() {
            "mainnet" => {
                let address_encoder: AddressEncoder = AddressEncoder::new(NetworkPrefix::Mainnet);
                let index_0_address: String = address_encoder.address_to_str(&address);
                let index_0_address: String = remove_quotes(index_0_address);
                RustKitWallet { secret_key, index_0_address, wallet, config }
            },
            "testnet" => {
                let address_encoder: AddressEncoder = AddressEncoder::new(NetworkPrefix::Testnet);
                let index_0_address: String = address_encoder.address_to_str(&address);
                let index_0_address: String = remove_quotes(index_0_address);
                RustKitWallet { secret_key, index_0_address, wallet, config }
            },
            _ => panic!("Invalid network"),
        }
    }

    /// Get sigma-rust address type from wallet
    pub fn get_address(&self) -> Address {
        let master_key: &SecretKey = &self.secret_key.clone();
        let master_key: SecretKey = master_key.to_owned();
        let address: Address = master_key.get_address_from_public_image();
        address
    }

    /// Get a p2pk address as a string from wallet
    pub fn get_p2pk_address(&self) -> String {
        return self.index_0_address.clone();
    }

    /// Get unspent boxes from explorer for wallet address
    pub fn get_input_boxes(&self) -> Option<Vec<ErgoBox>> {
        let bxs: Option<Vec<ErgoBox>> = ergo_rustkit_endpoints::get_unspent_boxes_for_address(self.index_0_address.as_str(), &self.config.clone().explorer_url);
        bxs
    }

    pub fn get_config(&self) -> Config {
        self.config.clone()
    }
}