Add `mullvad proxy use`

Allow for settings a specific Access Method to use

2 files changed, 20 insertions, 0 deletions

diff --git a/mullvad-management-interface/proto/management_interface.proto b/mullvad-management-interface/proto/management_interface.proto
index ffc3f12120..a062e0b642 100644
--- a/mullvad-management-interface/proto/management_interface.proto
+++ b/mullvad-management-interface/proto/management_interface.proto
@@ -87,6 +87,9 @@ service ManagementService {
   rpc ToggleApiAccessMethod(ApiAccessMethodToggle) returns (google.protobuf.Empty) {
     // Can I return something useful here instead of Empty?
   }
+  rpc SetApiAccessMethod(ApiAccessMethod) returns (google.protobuf.Empty) {
+    // Can I return something useful here instead of Empty?
+  }
 
   // Split tunneling (Linux)
   rpc GetSplitTunnelProcesses(google.protobuf.Empty) returns (stream google.protobuf.Int32Value) {}
diff --git a/mullvad-management-interface/src/client.rs b/mullvad-management-interface/src/client.rs
index fb2eae28c3..900775b10a 100644
--- a/mullvad-management-interface/src/client.rs
+++ b/mullvad-management-interface/src/client.rs
@@ -513,6 +513,23 @@ impl MullvadProxyClient {
             .map(drop)
     }
 
+    /// Set the [`AccessMethod`] which [`ApiConnectionModeProvider`] should
+    /// pick.
+    ///
+    /// - `access_method`: If `Some(access_method)`, [`ApiConnectionModeProvider`] will skip
+    ///     ahead and return `access_method` when asked for a new access method.
+    ///     If `None`, [`ApiConnectionModeProvider`] will pick the next access
+    ///     method "randomly"
+    ///
+    /// [`ApiConnectionModeProvider`]: mullvad_daemon::api::ApiConnectionModeProvider
+    pub async fn set_access_method(&mut self, access_method: AccessMethod) -> Result<()> {
+        self.0
+            .set_api_access_method(types::ApiAccessMethod::from(access_method))
+            .await
+            .map_err(Error::Rpc)
+            .map(drop)
+    }
+
     #[cfg(target_os = "linux")]
     pub async fn get_split_tunnel_processes(&mut self) -> Result<Vec<i32>> {
         use futures::TryStreamExt;