1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
use mullvad_management_interface::types::RelayLocation;
pub fn get_subcommand() -> clap::App<'static, 'static> {
clap::SubCommand::with_name("location")
.arg(
clap::Arg::with_name("country")
.help("The two letter country code, or 'any' for no preference.")
.required(true)
.index(1)
.validator(country_code_validator),
)
.arg(
clap::Arg::with_name("city")
.help("The three letter city code")
.index(2)
.validator(city_code_validator),
)
.arg(
clap::Arg::with_name("hostname")
.help("The hostname")
.index(3),
)
}
pub fn get_constraint(matches: &clap::ArgMatches<'_>) -> RelayLocation {
let country_original = matches.value_of("country").unwrap();
let country = country_original.to_lowercase();
let city = matches.value_of("city").map(str::to_lowercase);
let hostname = matches.value_of("hostname").map(str::to_lowercase);
match (country_original, city, hostname) {
("any", None, None) => RelayLocation::default(),
("any", ..) => clap::Error::with_description(
"City can't be given when selecting 'any' country",
clap::ErrorKind::InvalidValue,
)
.exit(),
(_, None, None) => RelayLocation {
country,
..Default::default()
},
(_, Some(city), None) => RelayLocation {
country,
city,
..Default::default()
},
(_, Some(city), Some(hostname)) => RelayLocation {
country,
city,
hostname,
},
(..) => clap::Error::with_description(
"Invalid country, city and hostname combination given",
clap::ErrorKind::InvalidValue,
)
.exit(),
}
}
pub fn format_location(location: Option<&RelayLocation>) -> String {
if let Some(location) = location {
if !location.hostname.is_empty() {
return format!(
"city {}, {}, hostname {}",
location.city, location.country, location.hostname
);
} else if !location.city.is_empty() {
return format!("city {}, {}", location.city, location.country);
} else if !location.country.is_empty() {
return format!("country {}", location.country);
}
}
"any location".to_string()
}
pub fn format_providers(providers: &Vec<String>) -> String {
if !providers.is_empty() {
format!("provider(s) {}", providers.join(", "))
} else {
"any provider".to_string()
}
}
fn country_code_validator(code: String) -> std::result::Result<(), String> {
if code.len() == 2 || code == "any" {
Ok(())
} else {
Err(String::from("Country codes must be two letters, or 'any'."))
}
}
fn city_code_validator(code: String) -> std::result::Result<(), String> {
if code.len() == 3 {
Ok(())
} else {
Err(String::from("City codes must be three letters"))
}
}
|
