main.rs

use anyhow::Result;
mod eth_wallet;
mod utils;
use std::env;
use std::io;
use std::str::FromStr;
use web3::types::Address;

#[tokio::main]
async fn main() -> Result<()> {
    dotenv::dotenv().ok();
    let (secret_key, pub_key) = eth_wallet::generate_keypair();

    println!("secret key: {}", &secret_key.to_string());
    println!("public key: {}", &pub_key.to_string());

    let pub_address = eth_wallet::public_key_address(&pub_key);
    println!("public address: {:?}", pub_address);

    // let crypto_wallet = eth_wallet::Wallet::new(&secret_key, &pub_key);
    // println!("crypto_wallet: {:?}", &crypto_wallet);

    let wallet_file_path = "crypto_wallet.json";
    // crypto_wallet.save_to_file(wallet_file_path)?;

    let loaded_wallet = eth_wallet::Wallet::from_file(wallet_file_path)?;
    println!("loaded_wallet: {:?}", loaded_wallet);

    let endpoint = env::var("INFURA_RINKEBY_WS")?;
    let web3_con = eth_wallet::establish_web3_connection(&endpoint).await?;

    let block_number = web3_con.eth().block_number().await?;
    println!("block number: {}", &block_number);

    let balance = loaded_wallet.get_balance_in_eth(&web3_con).await?;
    println!("wallet balance: {} eth", &balance);

    // Message signing
    println!("Enter a message to sign:");
    let mut message = String::new();
    io::stdin().read_line(&mut message)?;
    let signature = loaded_wallet.sign_message(message.trim())?;
    println!("Signature: {}", signature);

    // Verify signature
    let is_valid = loaded_wallet.verify_signature(message.trim(), &signature)?;
    println!("Signature valid: {}", is_valid);

    // Transaction
    println!("Enter recipient address (or 'skip' to skip):");
    let mut to_addr = String::new();
    io::stdin().read_line(&mut to_addr)?;
    let to_addr = to_addr.trim();
    if to_addr != "skip" {
        let to = Address::from_str(to_addr)?;
        println!("Enter amount in ETH:");
        let mut amount_str = String::new();
        io::stdin().read_line(&mut amount_str)?;
        let amount: f64 = amount_str.trim().parse()?;
        let transaction = eth_wallet::create_eth_transaction(to, amount);
        let gas_estimate = loaded_wallet.estimate_gas(&web3_con, &transaction).await?;
        println!("Estimated gas: {}", gas_estimate);
        let nonce = loaded_wallet.get_nonce(&web3_con).await?;
        println!("Nonce: {}", nonce);
        let transact_hash = eth_wallet::sign_and_send(&web3_con, transaction, &loaded_wallet.get_secret_key()?).await?;
        println!("Transaction hash: {:?}", transact_hash);
    }

    Ok(())
}