Add functions for checking and installing hwi

Author: wszdexdrf

requirements.txt

@@ -1,2 +1,3 @@
 hwi>=2.1.1,<3
 protobuf==3.20.1
+requests>=2.27.1

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,38 @@ impl HWIClient {
             status.into()
         })
     }
+
+    /// Get installed version of hwilib. Returns None if not found, or if version couldn't be known
+    pub fn get_version() -> Option<String> {
+        match || -> Result<String, Error> {
+            Python::with_gil(|py| {
+                Ok(PyModule::import(py, "hwilib")?
+                    .getattr("__version__")?
+                    .to_string())
+            })
+        }() {
+            Ok(s) => Some(s),
+            Err(_) => None,
+        }
+    }
+
+    /// Install hwi for the current user via pip. If no version is specified, the default version from pip will be installed.
+    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("Non UTF-8 error while installing")
+                    .to_string(),
+            ))
+        }
+    }
 }

src/lib.rs

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