Add servers to a custom list by hostname

1 files changed, 7 insertions, 2 deletions

diff --git a/mullvad-cli/src/cmds/custom_lists.rs b/mullvad-cli/src/cmds/custom_lists.rs
index 04efd8a967..d30f29f62d 100644
--- a/mullvad-cli/src/cmds/custom_lists.rs
+++ b/mullvad-cli/src/cmds/custom_lists.rs
@@ -1,4 +1,7 @@
-use super::relay_constraints::LocationArgs;
+use super::{
+    relay::{find_relay_by_hostname, get_filtered_relays},
+    relay_constraints::LocationArgs,
+};
 use anyhow::Result;
 use clap::Subcommand;
 use mullvad_management_interface::MullvadProxyClient;
@@ -106,7 +109,9 @@ impl CustomList {
     }
 
     async fn add_location(name: String, location_args: LocationArgs) -> Result<()> {
-        let location = Constraint::<GeographicLocationConstraint>::from(location_args);
+        let countries = get_filtered_relays().await?;
+        let location = find_relay_by_hostname(&countries, &location_args.country)
+            .map_or(Constraint::from(location_args), Constraint::Only);
         let update = CustomListLocationUpdate::Add { name, location };
         let mut rpc = MullvadProxyClient::new().await?;
         rpc.update_custom_list_location(update).await?;