wallet-gen 0.3.2

/*
 * wallet.rs
 *
 * Copyright 2018 Standard Mining
 *
 * Available to be used and modified under the terms of the MIT License.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
 * FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
 * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN
 * AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
 * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

//! A representation of a cryptocurrency wallet. Stores in that
//! coin's native WIF, or "wallet import format".

use super::prelude::*;
use {bitcoin, cryptonote, ethereum};
use std::collections::HashMap;

/// The actual wallet structure.
#[derive(Debug, Clone)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
pub struct Wallet {
    /// Which cryptocurrency this wallet is for
    pub coin: Coin,

    /// The wallet address as a human-readable string
    pub address: String,

    /// The wallet's public key in WIF format
    pub public_key: String,

    /// The wallet's private key in WIF format
    pub private_key: String,

    /// Extra fields, depending on the wallet
    pub other: Option<HashMap<String, String>>,
}

impl Wallet {
    /// Generates a new random wallet for the given coin.
    /// The private key will be sourced from a source of
    /// cryptographically-secure randomness. See OpenSSL
    /// documentation for more details.
    pub fn generate(coin: Coin) -> Result<Self> {
        use self::Coin::*;

        match coin {
            Ethereum | EthereumClassic => ethereum::new_wallet(coin),
            Monero | Aeon => cryptonote::new_wallet(coin),
            coin if bitcoin::wif_data(coin).is_some() => bitcoin::new_wallet(coin),
            _ => Err(Error::CoinNotSupported(coin)),
        }
    }
}

#[test]
fn gen_all_wallets() {
    use coin::COINS;

    println!("Generating wallets for all coins...");
    for coin in COINS.iter() {
        let wallet = match Wallet::generate(*coin) {
            Ok(wallet) => wallet,
            Err(Error::CoinNotSupported(_)) => continue,
            Err(e) => {
                panic!(
                    "Error generating wallet for {:?} ({}): {}",
                    coin,
                    coin.symbol(),
                    e,
                )
            },
        };

        println!("Coin: {:?} ({})", coin, coin.symbol());
        println!("Address: {}", &wallet.address);
        println!("Public key: {}", &wallet.public_key);
        println!("Private key: {}", &wallet.private_key);
        if let Some(ref other) = wallet.other {
            println!("Other: {:?}", other);
        }
        println!();
    }
}