Add password option to lcli deploy command

tests/eth1_test_rig/src/lib.rs

@@ -8,7 +8,7 @@
 mod ganache;
 
 use deposit_contract::{eth1_tx_data, testnet, ABI, BYTECODE, CONTRACT_DEPLOY_GAS, DEPOSIT_GAS};
-use futures::{stream, Future, IntoFuture, Stream};
+use futures::{future, stream, Future, IntoFuture, Stream};
 use ganache::GanacheInstance;
 use std::time::{Duration, Instant};
 use tokio::{runtime::Runtime, timer::Delay};
@@ -17,7 +17,7 @@ use types::{EthSpec, Hash256, Keypair, Signature};
 use web3::contract::{Contract, Options};
 use web3::transports::Http;
 use web3::types::{Address, TransactionRequest, U256};
-use web3::{Transport, Web3};
+use web3::Web3;
 
 pub const DEPLOYER_ACCOUNTS_INDEX: usize = 0;
 pub const DEPOSIT_ACCOUNTS_INDEX: usize = 0;
@@ -31,7 +31,7 @@ pub struct GanacheEth1Instance {
 impl GanacheEth1Instance {
     pub fn new() -> impl Future<Item = Self, Error = String> {
         GanacheInstance::new().into_future().and_then(|ganache| {
-            DepositContract::deploy(ganache.web3.clone(), 0).map(|deposit_contract| Self {
+            DepositContract::deploy(ganache.web3.clone(), 0, None).map(|deposit_contract| Self {
                 ganache,
                 deposit_contract,
             })
@@ -58,15 +58,23 @@ impl DepositContract {
     pub fn deploy(
         web3: Web3<Http>,
         confirmations: usize,
+        password: Option<String>,
     ) -> impl Future<Item = Self, Error = String> {
-        Self::deploy_bytecode(web3, confirmations, BYTECODE, ABI)
+        Self::deploy_bytecode(web3, confirmations, BYTECODE, ABI, password)
     }
 
     pub fn deploy_testnet(
         web3: Web3<Http>,
         confirmations: usize,
+        password: Option<String>,
     ) -> impl Future<Item = Self, Error = String> {
-        Self::deploy_bytecode(web3, confirmations, testnet::BYTECODE, testnet::ABI)
+        Self::deploy_bytecode(
+            web3,
+            confirmations,
+            testnet::BYTECODE,
+            testnet::ABI,
+            password,
+        )
     }
 
     fn deploy_bytecode(
@@ -74,21 +82,28 @@ impl DepositContract {
         confirmations: usize,
         bytecode: &[u8],
         abi: &[u8],
+        password: Option<String>,
     ) -> impl Future<Item = Self, Error = String> {
         let web3_1 = web3.clone();
 
-        deploy_deposit_contract(web3.clone(), confirmations, bytecode.to_vec(), abi.to_vec())
-            .map_err(|e| {
-                format!(
-                    "Failed to deploy contract: {}. Is scripts/ganache_tests_node.sh running?.",
-                    e
-                )
-            })
-            .and_then(move |address| {
-                Contract::from_json(web3_1.eth(), address, ABI)
-                    .map_err(|e| format!("Failed to init contract: {:?}", e))
-            })
-            .map(|contract| Self { contract, web3 })
+        deploy_deposit_contract(
+            web3.clone(),
+            confirmations,
+            bytecode.to_vec(),
+            abi.to_vec(),
+            password,
+        )
+        .map_err(|e| {
+            format!(
+                "Failed to deploy contract: {}. Is scripts/ganache_tests_node.sh running?.",
+                e
+            )
+        })
+        .and_then(move |address| {
+            Contract::from_json(web3_1.eth(), address, ABI)
+                .map_err(|e| format!("Failed to init contract: {:?}", e))
+        })
+        .map(|contract| Self { contract, web3 })
     }
 
     /// The deposit contract's address in `0x00ab...` format.
@@ -224,13 +239,15 @@ fn from_gwei(gwei: u64) -> U256 {
 
 /// Deploys the deposit contract to the given web3 instance using the account with index
 /// `DEPLOYER_ACCOUNTS_INDEX`.
-fn deploy_deposit_contract<T: Transport>(
-    web3: Web3<T>,
+fn deploy_deposit_contract(
+    web3: Web3<Http>,
     confirmations: usize,
     bytecode: Vec<u8>,
     abi: Vec<u8>,
+    password_opt: Option<String>,
 ) -> impl Future<Item = Address, Error = String> {
     let bytecode = String::from_utf8(bytecode).expect("bytecode must be valid utf8");
+    let web3_1 = web3.clone();
 
     web3.eth()
         .accounts()
@@ -241,6 +258,28 @@ fn deploy_deposit_contract<T: Transport>(
                 .cloned()
                 .ok_or_else(|| "Insufficient accounts for deployer".to_string())
         })
+        .and_then(move |from_address| {
+            let future: Box<dyn Future<Item = Address, Error = String> + Send> =
+                if let Some(password) = password_opt {
+                    // Unlock for only a single transaction.
+                    let duration = None;
+
+                    let future = web3_1
+                        .personal()
+                        .unlock_account(from_address, &password, duration)
+                        .then(move |result| match result {
+                            Ok(true) => Ok(from_address),
+                            Ok(false) => Err("Eth1 node refused to unlock account".to_string()),
+                            Err(e) => Err(format!("Eth1 unlock request failed: {:?}", e)),
+                        });
+
+                    Box::new(future)
+                } else {
+                    Box::new(future::ok(from_address))
+                };
+
+            future
+        })
         .and_then(move |deploy_address| {
             Contract::deploy(web3.eth(), &abi)
                 .map_err(|e| format!("Unable to build contract deployer: {:?}", e))?