CreateWallet update+ListWallets+GetWalletInfo

Author: Tibo-lg

client/src/client.rs

@@ -272,9 +272,27 @@ pub trait RpcApi: Sized {
         &self,
         wallet: &str,
         disable_private_keys: Option<bool>,
+        blank: Option<bool>,
+        passphrase: Option<&str>,
+        avoid_reuse: Option<bool>,
     ) -> Result<json::LoadWalletResult> {
-        let mut args = [wallet.into(), opt_into_json(disable_private_keys)?];
-        self.call("createwallet", handle_defaults(&mut args, &[null()]))
+        let mut args = [
+            wallet.into(),
+            opt_into_json(disable_private_keys)?,
+            opt_into_json(blank)?,
+            opt_into_json(passphrase)?,
+            opt_into_json(avoid_reuse)?,
+        ];
+        self.call("createwallet", handle_defaults(
+            &mut args, &[false.into(), false.into(), into_json("")?, false.into()]))
+    }
+
+    fn list_wallets(&self) -> Result<Vec<String>> {
+        self.call("listwallets", &[])
+    }
+
+    fn get_wallet_info(&self) -> Result<json::GetWalletInfoResult> {
+        self.call("getwalletinfo", &[])
     }
 
     fn backup_wallet(&self, destination: Option<&str>) -> Result<()> {

integration_test/src/main.rs

@@ -75,15 +75,23 @@ fn sbtc<F: Into<f64>>(btc: F) -> SignedAmount {
     SignedAmount::from_btc(btc.into()).unwrap()
 }
 
-fn main() {
-    let rpc_url = std::env::var("RPC_URL").expect("RPC_URL must be set");
-    let auth = if let Ok(cookie) = std::env::var("RPC_COOKIE") {
-        Auth::CookieFile(cookie.into())
+fn get_rpc_url() -> String {
+    return std::env::var("RPC_URL").expect("RPC_URL must be set");
+}
+
+fn get_auth() -> bitcoincore_rpc::Auth {
+    if let Ok(cookie) = std::env::var("RPC_COOKIE") {
+        return Auth::CookieFile(cookie.into());
     } else if let Ok(user) = std::env::var("RPC_USER") {
-        Auth::UserPass(user, std::env::var("RPC_PASS").unwrap_or_default())
+        return Auth::UserPass(user, std::env::var("RPC_PASS").unwrap_or_default());
     } else {
         panic!("Either RPC_COOKIE or RPC_USER + RPC_PASS must be set.");
     };
+}
+
+fn main() {
+    let rpc_url = get_rpc_url();
+    let auth = get_auth();
 
     let cl = Client::new(rpc_url, auth).unwrap();
 
@@ -137,6 +145,7 @@ fn main() {
     test_ping(&cl);
     test_get_peer_info(&cl);
     test_rescan_blockchain(&cl);
+    test_create_wallet(&cl);
     //TODO import_multi(
     //TODO verify_message(
     //TODO wait_for_new_block(&self, timeout: u64) -> Result<json::BlockRef> {
@@ -148,7 +157,6 @@ fn main() {
     //TODO add_multisig_address(
     //TODO load_wallet(&self, wallet: &str) -> Result<json::LoadWalletResult> {
     //TODO unload_wallet(&self, wallet: Option<&str>) -> Result<()> {
-    //TODO create_wallet(
     //TODO backup_wallet(&self, destination: Option<&str>) -> Result<()> {
     test_stop(cl);
 }
@@ -778,6 +786,97 @@ fn test_rescan_blockchain(cl: &Client) {
     assert_eq!(stop, Some(count - 1));
 }
 
+struct WalletParams<'a> {
+    name: &'a str,
+    disable_private_keys: Option<bool>,
+    blank: Option<bool>,
+    passphrase: Option<&'a str>,
+    avoid_reuse: Option<bool>,
+}
+
+fn build_wallet_params<'a>(
+    name: &'a str,
+    disable_private_keys: Option<bool>,
+    blank: Option<bool>,
+    passphrase: Option<&'a str>,
+    avoid_reuse: Option<bool>,
+) -> WalletParams<'a> {
+    WalletParams {
+        name,
+        disable_private_keys,
+        blank,
+        passphrase,
+        avoid_reuse,
+    }
+}
+
+fn test_create_single_wallet(cl: &Client, wallet_param: WalletParams) {
+    let result = cl
+        .create_wallet(
+            wallet_param.name,
+            wallet_param.disable_private_keys,
+            wallet_param.blank,
+            wallet_param.passphrase,
+            wallet_param.avoid_reuse,
+        )
+        .unwrap();
+
+    assert_eq!(result.name, wallet_param.name);
+    let expected_warning = match (wallet_param.passphrase, wallet_param.avoid_reuse) {
+        (None, Some(true)) => Some("Empty string given as passphrase, wallet will not be encrypted.".to_string()),
+        _ => Some("".to_string()),
+    };
+    assert_eq!(result.warning, expected_warning);
+
+    let wallet_client_url = format!("{}{}{}", get_rpc_url(), "/wallet/", wallet_param.name);
+    let wallet_client = Client::new(wallet_client_url, get_auth()).unwrap();
+    let wallet_info = wallet_client.get_wallet_info().unwrap();
+
+    assert_eq!(wallet_info.wallet_name, wallet_param.name);
+
+    fn option_to_bool(option: Option<bool>) -> bool {
+        match option {
+            None | Some(false) => false,
+            Some(true) => true,
+        }
+    }
+
+    let has_private_keys = !option_to_bool(wallet_param.disable_private_keys);
+    assert_eq!(wallet_info.private_keys_enabled, has_private_keys);
+    let has_hd_seed = has_private_keys && !option_to_bool(wallet_param.blank);
+    assert_eq!(wallet_info.hd_seed_id.is_some(), has_hd_seed);
+    let has_avoid_reuse = option_to_bool(wallet_param.avoid_reuse);
+    assert_eq!(option_to_bool(wallet_info.avoid_reuse), has_avoid_reuse);
+}
+
+fn test_create_wallet(cl: &Client) {
+    let wallet_names = vec!["alice", "bob", "carol", "denise", "emily"];
+    let mut wallet_params = vec![
+        build_wallet_params(wallet_names[0], None, None, None, None),
+        build_wallet_params(wallet_names[1], Some(true), None, None, None),
+        build_wallet_params(wallet_names[2], None, Some(true), None, None),
+    ];
+
+    if version() >= 190000 {
+        wallet_params.push(build_wallet_params(wallet_names[3], None, None, Some("pass"), None));
+        wallet_params.push(build_wallet_params(wallet_names[4], None, None, None, Some(true)));
+    }
+
+    for param in wallet_params {
+        test_create_single_wallet(&cl, param);
+    }
+
+    let mut wallet_list = cl.list_wallets().unwrap();
+
+    wallet_list.sort();
+
+    // Default wallet
+    assert_eq!(wallet_list.remove(0), "");
+
+    // Created wallets
+    assert!(wallet_list.iter().zip(wallet_names).all(|(a, b)| a == b));
+}
+
 fn test_stop(cl: Client) {
     println!("Stopping: '{}'", cl.stop().unwrap());
 }

json/src/lib.rs

@@ -123,6 +123,37 @@ pub struct LoadWalletResult {
     pub warning: Option<String>,
 }
 
+#[derive(Clone, PartialEq, Eq, Debug, Deserialize, Serialize)]
+pub struct GetWalletInfoResult {
+    #[serde(rename = "walletname")]
+    pub wallet_name: String,
+    #[serde(rename = "walletversion")]
+    pub wallet_version: u32,
+    #[serde(with = "bitcoin::util::amount::serde::as_btc")]
+    pub balance: Amount,
+    #[serde(with = "bitcoin::util::amount::serde::as_btc")]
+    pub unconfirmed_balance: Amount,
+    #[serde(with = "bitcoin::util::amount::serde::as_btc")]
+    pub immature_balance: Amount,
+    #[serde(rename = "txcount")]
+    pub tx_count: usize,
+    #[serde(rename = "keypoololdest")]
+    pub keypool_oldest: usize,
+    #[serde(rename = "keypoolsize")]
+    pub keypool_size: usize,
+    #[serde(rename = "keypoolsize_hd_internal")]
+    pub keypool_size_hd_internal: usize,
+    pub unlocked_until: Option<usize>,
+    #[serde(rename = "paytxfee", with = "bitcoin::util::amount::serde::as_btc")]
+    pub pay_tx_fee: Amount,
+    #[serde(rename = "hdseedid")]
+    pub hd_seed_id: Option<String>,
+    #[serde(rename = "hdmasterkeyid")]
+    pub hd_master_key_id: Option<String>,
+    pub private_keys_enabled: bool,
+    pub avoid_reuse: Option<bool>,
+}
+
 #[derive(Clone, PartialEq, Debug, Deserialize, Serialize)]
 #[serde(rename_all = "camelCase")]
 pub struct GetBlockResult {