Add functions for checking and installing hwi

Author: wszdexdrf

src/interface.rs

@@ -1,4 +1,5 @@
 use std::ops::Deref;
+use std::process::Command;
 
 use bitcoin::consensus::encode::serialize;
 use bitcoin::util::bip32::DerivationPath;
@@ -360,4 +361,35 @@ impl HWIClient {
             status.into()
         })
     }
+
+    /// Get installed version of hwilib
+    pub fn get_version() -> Result<String, Error> {
+        Python::with_gil(|py| {
+            Ok(PyModule::import(py, "hwilib")?
+                .getattr("__version__")?
+                .to_string())
+        })
+    }
+
+    /// Install hwi for the current user via pip
+    /// # Arguments
+    /// * `version` - the version of hwi to be installed. Installs default version from pip if not specified
+    pub fn install_hwilib(version: Option<&str>) -> Result<(), Error> {
+        let hwi_with_version = match version {
+            Some(ver) => "hwi==".to_owned() + ver,
+            None => "hwi".to_owned(),
+        };
+        let output = Command::new("pip")
+            .args(vec!["install", "--user", hwi_with_version.as_str()])
+            .output()?;
+        if output.status.success() {
+            Ok(())
+        } else {
+            Err(Error::HWIError(
+                std::str::from_utf8(&output.stderr)
+                    .expect("Unknown Error while installing")
+                    .to_string(),
+            ))
+        }
+    }
 }

src/lib.rs

@@ -244,6 +244,11 @@ mod tests {
             break;
         }
     }
+    #[test]
+    #[serial]
+    fn test_get_version() {
+        HWIClient::get_version().unwrap();
+    }
 
     #[test]
     #[serial]
@@ -260,4 +265,10 @@ mod tests {
             client.wipe_device().unwrap();
         }
     }
+    #[test]
+    #[serial]
+    #[ignore]
+    fn test_install_hwi() {
+        HWIClient::install_hwilib(Some("2.1.1")).unwrap();
+    }
 }