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
|
use crate::{new_rpc_client, Command, Error, Result, PRODUCT_VERSION};
use clap::value_t_or_exit;
pub struct BetaProgram;
#[mullvad_management_interface::async_trait]
impl Command for BetaProgram {
fn name(&self) -> &'static str {
"beta-program"
}
fn clap_subcommand(&self) -> clap::App<'static, 'static> {
clap::SubCommand::with_name(self.name())
.about("Receive notifications about beta updates")
.setting(clap::AppSettings::SubcommandRequiredElseHelp)
.subcommand(
clap::SubCommand::with_name("set")
.about("Change beta notifications setting")
.arg(
clap::Arg::with_name("policy")
.required(true)
.possible_values(&["on", "off"]),
),
)
.subcommand(clap::SubCommand::with_name("get").about("Get beta notifications setting"))
}
async fn run(&self, matches: &clap::ArgMatches<'_>) -> Result<()> {
match matches.subcommand() {
("get", Some(_)) => {
let mut rpc = new_rpc_client().await?;
let settings = rpc.get_settings(()).await?.into_inner();
let enabled_str = if settings.show_beta_releases {
"on"
} else {
"off"
};
println!("Beta program: {}", enabled_str);
Ok(())
}
("set", Some(matches)) => {
let enable_str = value_t_or_exit!(matches.value_of("policy"), String);
let enable = enable_str == "on";
if !enable && PRODUCT_VERSION.contains("beta") {
return Err(Error::InvalidCommand(
"The beta program must be enabled while running a beta version",
));
}
let mut rpc = new_rpc_client().await?;
rpc.set_show_beta_releases(enable).await?;
println!("Beta program: {}", enable_str);
Ok(())
}
_ => {
unreachable!("unhandled comand");
}
}
}
}
|
