Add `mullvad proxy test`

For quickly assessing whether an api access method can reach the API or not.

1 files changed, 34 insertions, 0 deletions

diff --git a/mullvad-cli/src/cmds/proxy.rs b/mullvad-cli/src/cmds/proxy.rs
index c876a25d2e..fa05081f89 100644
--- a/mullvad-cli/src/cmds/proxy.rs
+++ b/mullvad-cli/src/cmds/proxy.rs
@@ -24,6 +24,8 @@ pub enum Proxy {
     Enable(SelectItem),
     /// Disable an API proxy
     Disable(SelectItem),
+    /// Test an API proxy
+    Test(SelectItem),
     /// Force the use of a specific API proxy.
     Use(SelectItem),
 }
@@ -59,6 +61,10 @@ impl Proxy {
                 let enabled = false;
                 Self::toggle(index, enabled).await?;
             }
+            Proxy::Test(cmd) => {
+                let index = Self::zero_to_one_based_index(cmd.index)?;
+                Self::test(index).await?;
+            }
             Proxy::Use(cmd) => {
                 let index = Self::zero_to_one_based_index(cmd.index)?;
                 Self::set(index).await?;
@@ -199,6 +205,34 @@ impl Proxy {
         Ok(())
     }
 
+    /// Test an access method to see if it successfully reaches the Mullvad API.
+    async fn test(index: usize) -> Result<()> {
+        let mut rpc = MullvadProxyClient::new().await?;
+        // TODO: Refactor this into some helper function. This code has been copy-pastead a lot already ..
+        // Step 1.
+        let access_method = rpc
+            .get_api_access_methods()
+            .await?
+            .get(index)
+            .ok_or(anyhow!(format!(
+                "Access method {} does not exist",
+                index + 1
+            )))?
+            .clone();
+
+        // Step 2.
+        rpc.set_access_method(access_method).await?;
+        // Step 3.
+        let api_call = rpc.test_api().await;
+        // Step 4.
+        match api_call {
+            Ok(_) => println!("API call succeeded!"),
+            Err(_) => println!("API call failed :-("),
+        }
+
+        Ok(())
+    }
+
     /// Force the use of a specific access method when trying to reach the
     /// Mullvad API. If this method fails, the daemon will resume the automatic
     /// roll-over behavior (which is the default).