Add relay override commands to wrapper client

1 files changed, 20 insertions, 1 deletions

diff --git a/mullvad-management-interface/src/client.rs b/mullvad-management-interface/src/client.rs
index 1c15a169cd..140eddc08a 100644
--- a/mullvad-management-interface/src/client.rs
+++ b/mullvad-management-interface/src/client.rs
@@ -8,7 +8,9 @@ use mullvad_types::{
     custom_list::{CustomList, Id},
     device::{Device, DeviceEvent, DeviceId, DeviceState, RemoveDeviceEvent},
     location::GeoIpLocation,
-    relay_constraints::{BridgeSettings, BridgeState, ObfuscationSettings, RelaySettings},
+    relay_constraints::{
+        BridgeSettings, BridgeState, ObfuscationSettings, RelayOverride, RelaySettings,
+    },
     relay_list::RelayList,
     settings::{DnsOptions, Settings},
     states::TunnelState,
@@ -339,6 +341,23 @@ impl MullvadProxyClient {
         Ok(())
     }
 
+    pub async fn set_relay_override(&mut self, relay_override: RelayOverride) -> Result<()> {
+        let r#override = types::RelayOverride::from(relay_override);
+        self.0
+            .set_relay_override(r#override)
+            .await
+            .map_err(Error::Rpc)?;
+        Ok(())
+    }
+
+    pub async fn clear_all_relay_overrides(&mut self) -> Result<()> {
+        self.0
+            .clear_all_relay_overrides(())
+            .await
+            .map_err(Error::Rpc)?;
+        Ok(())
+    }
+
     pub async fn create_new_account(&mut self) -> Result<AccountToken> {
         Ok(self
             .0